def test_extract_street_from_tr(self, mocked_streets_hausen):
for index, tr in enumerate(mocked_streets_hausen):
x = extract_street_from_tr(tr)
assert x['name'] == 'Gerlinger Straße'
assert x['zipcode'] == '70499'
assert x['city'] == 'Stuttgart-Hausen'
break # one is enough
def test_extract_street_from_tr(self, mocked_streets_hausen):
for index, tr in enumerate(mocked_streets_hausen):
x = extract_street_from_tr(tr)
assert x['name'] == 'Gerlinger Straße'
assert x['zipcode'] == '70499'
assert x['city'] == 'Stuttgart-Hausen'
break # one is enough
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 stuttgart_streets_hausen_process(mocked_streets_hausen):
from main.models import District
for index, tr in enumerate(mocked_streets_hausen):
x = extract_street_from_tr(tr)
add_street_to_database(x, District.objects.first())
def stuttgart_streets_hausen_process(mocked_streets_hausen):
from main.models import District
for index, tr in enumerate(mocked_streets_hausen):
x = extract_street_from_tr(tr)
add_street_to_database(x, District.objects.first())
