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 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):
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)
