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