def handle(self, *args, **options): from main.utils import ( add_district_to_database, add_street_to_database, extract_district_from_tr, extract_street_from_tr, get_districts_stuttgart, get_streets_from_district, ) for tr in get_districts_stuttgart(): d = extract_district_from_tr(tr) district = add_district_to_database(d) for street in get_streets_from_district(district): data = extract_street_from_tr(street) street = add_street_to_database(data, district)
def test_extract_district_from_tr(self, mocked_district): for index, tr in enumerate(mocked_district): x = extract_district_from_tr(tr) assert isinstance(x, dict) assert 'name' in x break # one is enough
def stuttgart_districts_process(mocked_district): for index, tr in enumerate(mocked_district): x = extract_district_from_tr(tr) add_district_to_database(x)
def stuttgart_districts_process(mocked_district): from main.utils import add_district_to_database, extract_district_from_tr for index, tr in enumerate(mocked_district): x = extract_district_from_tr(tr) add_district_to_database(x)