Add support for ASREVIEW_LAB_SQLALCHEMY_DATABASE_URI in auth-tool

asreview · Mar 20, 2024 · 7f52f1c · 7f52f1c
1 parent 5f3c398
commit 7f52f1c
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/asreview/webapp/entry_points/auth_tool.py b/asreview/webapp/entry_points/auth_tool.py
@@ -48,8 +48,8 @@ def auth_parser():
         type=str,
         default=None,
         help=("URI of the database. By default, the value is given by the environment "
-              "variable SQLALCHEMY_DATABASE_URI. If not set, the default is "
-              "'asreview.production.sqlite' in the ASReview folder."),
+              "variable ASREVIEW_LAB_SQLALCHEMY_DATABASE_URI. If not set, the default "
+              "is 'asreview.production.sqlite' in the ASReview folder."),
     )
 
     # ADD USERS
@@ -64,8 +64,8 @@ def auth_parser():
         type=str,
         default=None,
         help=("URI of the database. By default, the value is given by the environment "
-              "variable SQLALCHEMY_DATABASE_URI. If not set, the default is "
-              "'asreview.production.sqlite' in the ASReview folder."),
+              "variable ASREVIEW_LAB_SQLALCHEMY_DATABASE_URI. If not set, the default "
+              "is 'asreview.production.sqlite' in the ASReview folder."),
     )
 
     user_par.add_argument(
@@ -87,8 +87,8 @@ def auth_parser():
         type=str,
         default=None,
         help=("URI of the database. By default, the value is given by the environment "
-              "variable SQLALCHEMY_DATABASE_URI. If not set, the default is "
-              "'asreview.production.sqlite' in the ASReview folder."),
+              "variable ASREVIEW_LAB_SQLALCHEMY_DATABASE_URI. If not set, the default "
+              "is 'asreview.production.sqlite' in the ASReview folder."),
     )
 
     # LIST PROJECTS
@@ -122,8 +122,8 @@ def auth_parser():
         type=str,
         default=None,
         help=("URI of the database. By default, the value is given by the environment "
-              "variable SQLALCHEMY_DATABASE_URI. If not set, the default is "
-              "'asreview.production.sqlite' in the ASReview folder."),
+              "variable ASREVIEW_LAB_SQLALCHEMY_DATABASE_URI. If not set, the default "
+              "is 'asreview.production.sqlite' in the ASReview folder."),
     )
 
     return parser
@@ -200,6 +200,7 @@ def execute(self, argv):
         if self.argv != ["list-projects"]:
             self.uri = getattr(self.args, "db_uri", False) or \
                 os.environ.get("SQLALCHEMY_DATABASE_URI", False) or \
+                os.environ.get("ASREVIEW_LAB_SQLALCHEMY_DATABASE_URI", False) or \
                 DEFAULT_DATABASE_URI
             Session = sessionmaker()
             engine = create_engine(self.uri)