Add prepopulate command
This commit is contained in:
parent
0f9fe187df
commit
cfe5e0a0e9
36
src/main.py
36
src/main.py
|
@ -44,6 +44,42 @@ def populate_boucles():
|
|||
break
|
||||
|
||||
|
||||
@app.command()
|
||||
def prepopulate_today():
|
||||
today = pendulum.today("UTC")
|
||||
|
||||
limit = 10
|
||||
start = 0
|
||||
process = True
|
||||
while process:
|
||||
r = requests.get(
|
||||
f"https://data.nantesmetropole.fr/api/records/1.0/search/?dataset=244400404_comptages-velo-nantes-metropole&q=&rows={limit}&start={start}&sort=jour&facet=boucle_num&facet=libelle&facet=jour&facet=probabilite_presence_anomalie&facet=jour_de_la_semaine&facet=boucle_libelle&facet=vacances_zone_b"
|
||||
)
|
||||
if r.status_code == requests.codes.ok:
|
||||
data = r.json()
|
||||
if data["records"] != []:
|
||||
for record in data["records"]:
|
||||
log.debug(f'record date is {record["fields"]["dateformat"]}')
|
||||
record_date = pendulum.parse(record["fields"]["dateformat"])
|
||||
if record_date == today:
|
||||
log.info("Date is good")
|
||||
parseRecord(record)
|
||||
else:
|
||||
log.info("Stop processing")
|
||||
process = False
|
||||
break
|
||||
start += 10
|
||||
else:
|
||||
log.debug("Records is empty, stop it")
|
||||
process = False
|
||||
pprint.pprint(data["records"])
|
||||
|
||||
else:
|
||||
pprint.pprint(r.status_code)
|
||||
pprint.pprint(r.text)
|
||||
process = False
|
||||
|
||||
|
||||
@app.command()
|
||||
def populate_day():
|
||||
today = pendulum.today("UTC")
|
||||
|
|
Loading…
Reference in a new issue