def __init__(self, filename='db.sqlite') -> None:

self.filename = filename

global_init(filename)

self.weather = WeatherApi()

def search_city(self, name):

session = create_session()

cities = self.weather.get_city(name)

data = map(lambda x: x.title, session.query(City).all())

filtered = filter(lambda x: x['name'] not in data, cities)

return list(filtered)

def add_city(self, data:dict):

session = create_session()

city = City()

city.title = data['name']

city.link = self.weather.link + data['url'].replace('weather', 'weather-history')

session.add(city)

session.commit()

def load_cities(self) -> List[City]:

session = create_session()

cities = session.query(City).all()

return cities

def load_weather(self, date_from:datetime, date_to:datetime, city:City) -> List[str]:

session = create_session()

data = []

url = city.link

winds = session.query(Winds).filter(

and_(Winds.date >= date_from.strftime('%Y-%m-%d'), Winds.date <= date_to.strftime('%Y-%m-%d'), Winds.city_id == city.id)).all()

if len(winds) == 0:

days = date_to.__sub__(date_from).days

for i in range(1, days):

date = date_to - timedelta(days=i)

direction = self.weather.get_wind_direction(url, date)

if not direction:

continue

data.append(direction)

wind = Winds()

wind.date = date

wind.city_id = city.id

wind.wind = direction

session.add(wind)

session.commit()

else:

data = list(map(lambda x: x.wind, winds))

db_date_from = min(list(map(lambda x: x.date, winds)))

days = db_date_from.__sub__(date_from).days

for i in range(1, days):

date = db_date_from - timedelta(days=i)

direction = self.weather.get_wind_direction(url, date)

if not direction:

continue

data.append(direction)

wind = Winds()

wind.date = date

wind.city_id = city.id

wind.wind = direction

session.add(wind)

session.commit()

return data