Added proper support for all databases

spoolman/main.py

@@ -1,20 +1,77 @@
 """Main entrypoint to the server."""
 
+import logging
+from pathlib import Path
 
 import uvicorn
 from fastapi import FastAPI
+from platformdirs import user_data_dir
+from sqlalchemy import URL
 
+from spoolman import env
 from spoolman.api.v1.router import app as v1_app
 from spoolman.database.database import setup_db
 
+# Get the data directory
+data_dir = Path(user_data_dir("spoolman"))
+data_dir.mkdir(parents=True, exist_ok=True)
+
+# Setup file logger
+logging.basicConfig(
+    filename=data_dir.joinpath("spoolman.log"),
+    filemode="w",
+    level=logging.INFO,
+    format="%(asctime)s:%(levelname)s:%(message)s",
+    datefmt="%Y-%m-%d %I:%M:%S%p",
+)
+
+# Setup console logger
+formatter = logging.Formatter("%(name)-12s: %(levelname)-8s %(message)s")
+console = logging.StreamHandler()
+console.setLevel(logging.INFO)
+console.setFormatter(formatter)
+logging.getLogger("").addHandler(console)
+
+# Get logger instance
+logger = logging.getLogger(__name__)
+
+# Setup FastAPI
 app = FastAPI(debug=True)
 app.mount("/api/v1", v1_app)
 
 
+def get_connection_url() -> URL:
+    """Construct the connection URL for the database based on environment variables."""
+    db_type = env.get_database_type()
+    host = env.get_host()
+    port = env.get_port()
+    database = env.get_database()
+    query = env.get_query()
+    username = env.get_username()
+    password = env.get_password()
+
+    if db_type is None:
+        db_type = env.DatabaseType.SQLITE
+        database = str(data_dir.joinpath("spoolman.db"))
+        logger.info('No database type specified, using a default SQLite database located at "%s"', database)
+    else:
+        logger.info('Connecting to database of type "%s" at "%s:%s"', db_type, host, port)
+
+    return URL.create(
+        drivername=db_type.to_drivername(),
+        host=host,
+        port=port,
+        database=database,
+        query=query or {},
+        username=username,
+        password=password,
+    )
+
+
 @app.on_event("startup")
 async def startup() -> None:
     """Run the service's startup sequence."""
-    await setup_db("sqlite+aiosqlite:///./sql_app.db")
+    await setup_db(get_connection_url())
 
 
 if __name__ == "__main__":