IDI_Master/projet_INSEE
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Process departments

Browse source
This commit is contained in:
Yorick Barbanneau 2022-04-17 22:17:19 +02:00
parent 99e9fdaec0
commit fa03597765
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

18
create_db.py
View file

@ -47,7 +47,7 @@ def parse_args():
default='communes2021.csv')
parser.add_argument('--departments',
help='departments raw csv file (inside source follder)',
default='departements2021.csv')
default='departement2021.csv')
parser.add_argument('--states',
help='states raw csv file (inside source follder)',
default='region2021.csv')
@ -68,11 +68,16 @@ def create_states_csv(raw_file):
return states
def create_adepartment_csv(raw_file):
def create_department_csv(raw_file):
"""
Process department files
"""
dep = pd.read_csv(raw_file)
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
if __name__ == '__main__':
args = parse_args()
@ -93,5 +98,10 @@ if __name__ == '__main__':
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()