示例#1
0
def init() -> None:
    try:
        pg = Session()
        pg.execute('SELECT 1')
        pass
    except Exception as e:
        logger.error(e)
        raise e
示例#2
0
def init() -> None:
    try:
        pg = Session()
        pg.execute('SELECT 1')
        mongo = Mongo()
        mongo.mongo_db.list_collections()
    except Exception as e:
        logger.error(e)
        raise e
logger = logging.getLogger(__name__)

max_tries = 60  # 1 minute
wait_seconds = 1


@retry(
    stop=stop_after_attempt(max_tries),
    wait=wait_fixed(wait_seconds),
    before=before_log(logger, logging.INFO),
    after=after_log(logger, logging.WARN),
)
def init() -> None:
    try:
{%- if cookiecutter.use_postgres == "Yes" %}
        pg = Session()
        pg.execute('SELECT 1')
{%- endif %}
{%- if cookiecutter.use_mongo == "Yes" %}
        mongo = Mongo()
        mongo.mongo_db.list_collections()
{%- endif %}
        pass
    except Exception as e:
        logger.error(e)
        raise e


if __name__ == '__main__':
    init()