Add extract-postalcodes script

master
informatic 2020-03-29 14:41:39 +02:00
parent 313449e200
commit 5726dfda5b
1 changed files with 11 additions and 0 deletions

View File

@ -1,4 +1,5 @@
import csv
import re
import click
from flask.cli import AppGroup
from formity.models import FaceshieldRequest, PostalCode
@ -53,3 +54,13 @@ def import_file(input_file, model, lookup, columns=None, delimiter=',', ignore_h
db.session.commit()
db.session.commit()
@cli.command('extract-postalcodes')
def extract_postalcodes():
for r in FaceshieldRequest.query.all():
code = re.findall(r'(\d{2})\s*-\s*(\d{3})', str(r.entity_info))
if code:
r.shipping_postalcode = '-'.join(code[0])
r.shipping_city = PostalCode.query.get(r.shipping_postalcode).city
print(r.id, r.shipping_postalcode)
db.session.commit()