Using APP_ENV to define ENV_LOG_FLAG (partially #107)

phenopolis · Sep 4, 2020 · a52e4cf · a52e4cf
1 parent 26515a4
commit a52e4cf
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/views/__init__.py b/views/__init__.py
@@ -13,6 +13,13 @@
 from logging.config import dictConfig
 from flask.logging import default_handler
 
+# Options are: prod, dev, debug (default), test (for coverage)
+APP_ENV = os.getenv("APP_ENV", "debug")
+
+ENV_LOG_FLAG = True
+if APP_ENV in ["test", "prod"]:
+    ENV_LOG_FLAG = False
+
 
 def _configure_logs():
     dictConfig(

diff --git a/views/postgres.py b/views/postgres.py
@@ -5,7 +5,7 @@
 from flask import g
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, Session
-from views import application
+from views import application, ENV_LOG_FLAG
 
 
 def get_db():
@@ -25,7 +25,7 @@ def get_db_session() -> Session:
     current application context.
     """
     if not hasattr(g, "dbsession"):
-        engine = create_engine(application.config["SQLALCHEMY_DATABASE_URI"], echo=True)
+        engine = create_engine(application.config["SQLALCHEMY_DATABASE_URI"], echo=ENV_LOG_FLAG)
         engine.connect()
         DbSession = sessionmaker(bind=engine)
         DbSession.configure(bind=engine)