def check_db():
if engine.dialect.has_table(engine.connect(), ConfigModel.__tablename__):
session = session_factory()
conf = session.query(ConfigModel).first()
if conf is not None:
return
raise ValueError("Arbalet backend database has not been initialized. "
"Please run 'docker-compose run --rm app init' before starting the scheduler. "
"See INSTALL instructions for more information.")
def get_places(dishes):
c = PlacePicker()
dish_to_places = {}
with engine.connect() as conn:
# выбираем заведения по всем вариациям из ввода
place_count = 0
for dish in dishes:
tmp_places = c.select_place(conn, dish)
dish_to_places[dish] = tmp_places
# если нашли достаточное количество мест, повторно в базу не идем
place_count += len(tmp_places)
if place_count > 9:
break
return dish_to_places