diff --git a/create_db.py b/create_db.py index 625fbb8..da6b5b6 100755 --- a/create_db.py +++ b/create_db.py @@ -47,7 +47,7 @@ def parse_args(): default='communes2021.csv') parser.add_argument('--departments', help='departments raw csv file (inside source follder)', - default='departement2021.csv') + default='departements2021.csv') parser.add_argument('--states', help='states raw csv file (inside source follder)', default='region2021.csv') @@ -68,16 +68,11 @@ def create_states_csv(raw_file): return states -def create_department_csv(raw_file): +def create_adepartment_csv(raw_file): """ Process department files """ - reg_convert= lambda x: x if len(str(x)) == 2 else f'0{x}' - dep = pd.read_csv(raw_file, - usecols=["DEP","NCC","LIBELLE","REG","CHEFLIEU"], - converters={'REG':reg_convert}) - return dep - + dep = pd.read_csv(raw_file) if __name__ == '__main__': args = parse_args() @@ -85,7 +80,7 @@ if __name__ == '__main__': logger = logging.getLogger() if args.verbose is True: logger.setLevel(logging.DEBUG) - logger.debug('Debug mode activated') + logger.debug('Debug mode actiuvbated') tty_handler = logging.StreamHandler() @@ -93,15 +88,11 @@ if __name__ == '__main__': tty_handler.setFormatter(CustomFormatter()) logger.addHandler(tty_handler) + logging.debug("Ceci est une test") if not os.path.exists(args.source + '/' + args.states): logger.critical('can\'t find source file for states') states = create_states_csv(args.source + '/' + args.states) print(states) - - if not os.path.exists(args.source + '/' + args.states): - logger.critical('can\'t find source file for departments') - departments = create_department_csv(args.source + '/' + args.departments) - print(departments) - + sys.exit()