async def _submit_new_event(self, memo_dict): answer = 'New event was successfully created 馃帀' await telegram_bot.send_message(chat_id=self.chat.user_id, text=answer) # Save to database create_event( db=next(get_db()), title=memo_dict['title'], start=memo_dict['start'], end=memo_dict['end'], content=memo_dict['content'], owner_id=self.user.id, location=memo_dict['location'], ) # Delete current session del telegram_bot.MEMORY[self.chat.user_id] return answer
def test_get_db(self): assert isinstance(next(get_db()), Session)
import os from app import create_app from app.database.database import get_db config_name = os.getenv('CONFIG') app = create_app(config_name) if __name__ == '__main__': if config_name == 'DEVELOPMENT': host = '127.0.0.1' port = 5000 else: host = '0.0.0.0' port = 80 app.run(host=host, port=port) # Connect to database at least one to avoid cold start get_db()
def test_session_db(): assert get_db() is not None
create_tables(engine, PSQL_ENVIRONMENT) app = FastAPI() app.mount("/static", StaticFiles(directory=STATIC_PATH), name="static") app.mount("/media", StaticFiles(directory=MEDIA_PATH), name="media") app.logger = logger # This MUST come before the app.routers imports. set_ui_language() from app.routers import ( # noqa: E402 agenda, calendar, categories, currency, dayview, email, event, invitation, profile, search, telegram, whatsapp) json_data_loader.load_to_db(next(get_db())) routers_to_include = [ agenda.router, calendar.router, categories.router, currency.router, dayview.router, email.router, event.router, invitation.router, profile.router, search.router, telegram.router, whatsapp.router, ]