diff --git a/formity/scripts.py b/formity/scripts.py index 9b9550f..db127f6 100644 --- a/formity/scripts.py +++ b/formity/scripts.py @@ -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()