Attempt to fix db connection with pre and post request hooks

Author: CalPinSW

backend/src/app.py

@@ -8,6 +8,7 @@
 from src.musicbrainz import MusicbrainzClient
 from src.spotify import SpotifyClient
 from src.controllers.auth import auth_controller
+from src.database.models import database
 
 
 def create_app():
@@ -47,6 +48,15 @@ def handle_unauthorized_exception(_):
         resp.delete_cookie("user_id")
         return resp
 
+    @app.before_request
+    def connect_db():
+        database.connect(reuse_if_open=True)
+
+    @app.teardown_request
+    def _db_close(exc):
+        if not database.is_closed():
+            database.close()
+
     app.register_blueprint(auth_controller(spotify=spotify))
     app.register_blueprint(spotify_controller(spotify=spotify))
     app.register_blueprint(music_controller(spotify=spotify))

backend/src/controllers/database.py

@@ -16,7 +16,6 @@
 from src.database.crud.user import get_or_create_user
 from src.musicbrainz import MusicbrainzClient
 from src.spotify import SpotifyClient
-from time import sleep
 
 
 def database_controller(spotify: SpotifyClient, musicbrainz: MusicbrainzClient):

backend/src/database/crud/playlist.py

@@ -16,8 +16,6 @@
 from src.dataclasses.album import Album
 from src.dataclasses.playlist import Playlist
 from peewee import fn, prefetch
-from playhouse.shortcuts import model_to_dict
-import time
 
 
 def get_playlist_by_id_or_none(id: str):