Rebase against distributed-system-analysis#1937

This includes more change than it should, because I'd originally
segregated the database files into a `db` directory while Nikhil
went with `database`.

Author: dbutenhof

lib/pbench/cli/server/shell.py

@@ -8,7 +8,7 @@
 
 from pbench.common.exceptions import BadConfig, ConfigFileNotSpecified
 from pbench.server.api import create_app, get_server_config
-from pbench.server.db.database import Database
+from pbench.server.database.database import Database
 from pbench.common.logger import get_pbench_logger
 
 

lib/pbench/server/api/resources/upload_api.py

@@ -9,7 +9,7 @@
 from flask import request, jsonify
 from werkzeug.utils import secure_filename
 from pbench.server.utils import filesize_bytes
-from pbench.server.db.models.tracker import Dataset, States
+from pbench.server.database.models.tracker import Dataset, States
 
 
 ALLOWED_EXTENSIONS = {"xz"}

lib/pbench/server/api/resources/users_api.py

@@ -92,13 +92,13 @@ def post(self):
 
         first_name = user_data.get("first_name")
         if not first_name:
-            self.logger.warning("Missing firstName field")
-            abort(400, message="Missing firstName field")
+            self.logger.warning("Missing first_name field")
+            abort(400, message="Missing first_name field")
 
         last_name = user_data.get("last_name")
         if not last_name:
-            self.logger.warning("Missing lastName field")
-            abort(400, message="Missing lastName field")
+            self.logger.warning("Missing last_name field")
+            abort(400, message="Missing last_name field")
 
         try:
             user = User(
@@ -325,8 +325,8 @@ def get(self, username):
                     "message": "Success"/"failure message",
                     "data": {
                         "username": <username>,
-                        "firstName": <firstName>,
-                        "lastName": <lastName>,
+                        "first_name": <firstName>,
+                        "last_name": <lastName>,
                         "registered_on": registered_on,
                     }
                 }

lib/pbench/server/database/database.py

@@ -21,24 +21,18 @@ def get_engine_uri(config, logger):
             )
             return None
 
-        # return f"postgresql+{psql_driver}://{psql_username}:{psql_password}@{psql_host}:{psql_port}/{psql_db}"
-
-    @staticmethod
-    def init_engine(server_config, logger):
-        try:
-            return create_engine(Database.get_engine_uri(server_config, logger))
-        except Exception:
-            logger.exception("Exception while creating a sqlalchemy engine")
-            return None
-
     @staticmethod
     def init_db(server_config, logger):
+        # Attach the logger to the base class for models to find
+        if not hasattr(Database.Base, "logger"):
+            Database.Base.logger = logger
+
         # Make sure all the models are imported before this function gets called
         # so that they will be registered properly on the metadata. Otherwise
         # metadata will not have any tables and create_all functionality will do nothing
 
         Database.Base.query = Database.db_session.query_property()
-
+        Database.db_session.remove()  # Remove the placeholder static session
         engine = create_engine(Database.get_engine_uri(server_config, logger))
         Database.Base.metadata.create_all(bind=engine)
         Database.db_session.configure(bind=engine)

lib/pbench/server/db/models/tracker.py renamed to lib/pbench/server/database/models/tracker.py

@@ -16,7 +16,7 @@
 from sqlalchemy.orm import relationship, validates
 from sqlalchemy.exc import SQLAlchemyError
 
-from pbench.server.db.database import Database
+from pbench.server.database.database import Database
 
 
 class DatasetError(Exception):

lib/pbench/server/db/__init__.py

@@ -21,7 +21,7 @@
     es_index,
     VERSION,
 )
-from pbench.server.db.models.tracker import (
+from pbench.server.database.models.tracker import (
     Dataset,
     States,
     Metadata,

lib/pbench/server/db/database.py

@@ -2,7 +2,7 @@
 import sys
 import shutil
 
-from pbench.server.db.models.tracker import Dataset, States, DatasetNotFound
+from pbench.server.database.models.tracker import Dataset, States, DatasetNotFound
 
 
 def rename_tb_link(tb, dest, logger):