style: run python linter & prettifier on backend code

server/core/auth.py

@@ -5,6 +5,7 @@ from flask_bcrypt import Bcrypt
 
 EMAIL_REGEX = re.compile(r"[^@]+@[^@]+\.[^@]+")
 
+
 class AuthManager:
     def __init__(self, db: PostgresConnector, bcrypt: Bcrypt):
         self.db = db
@@ -24,13 +25,13 @@ class AuthManager:
 
         if len(username) < 3:
             raise ValueError("Username must be longer than 3 characters")
-        
+
         if not EMAIL_REGEX.match(email):
             raise ValueError("Please enter a valid email address")
 
         if self.get_user_by_email(email):
             raise ValueError("Email already registered")
-        
+
         if self.get_user_by_username(username):
             raise ValueError("Username already taken")
 
@@ -38,20 +39,22 @@ class AuthManager:
 
     def authenticate_user(self, username, password):
         user = self.get_user_by_username(username)
-        if user and self.bcrypt.check_password_hash(user['password_hash'], password):
+        if user and self.bcrypt.check_password_hash(user["password_hash"], password):
             return user
         return None
-    
+
     def get_user_by_id(self, user_id):
         query = "SELECT id, username, email FROM users WHERE id = %s"
         result = self.db.execute(query, (user_id,), fetch=True)
         return result[0] if result else None
-    
+
     def get_user_by_username(self, username) -> dict:
-        query = "SELECT id, username, email, password_hash FROM users WHERE username = %s"
+        query = (
+            "SELECT id, username, email, password_hash FROM users WHERE username = %s"
+        )
         result = self.db.execute(query, (username,), fetch=True)
         return result[0] if result else None
-    
+
     def get_user_by_email(self, email) -> dict:
         query = "SELECT id, username, email, password_hash FROM users WHERE email = %s"
         result = self.db.execute(query, (email,), fetch=True)

server/core/datasets.py

@@ -3,6 +3,7 @@ from server.db.database import PostgresConnector
 from psycopg2.extras import Json
 from server.exceptions import NonExistentDatasetException
 
+
 class DatasetManager:
     def __init__(self, db: PostgresConnector):
         self.db = db
@@ -15,18 +16,18 @@ class DatasetManager:
 
         if dataset_info.get("user_id") != user_id:
             return False
-        
+
         return True
-    
+
     def get_user_datasets(self, user_id: int) -> list[dict]:
         query = "SELECT * FROM datasets WHERE user_id = %s"
-        return self.db.execute(query, (user_id, ), fetch=True)
+        return self.db.execute(query, (user_id,), fetch=True)
 
     def get_dataset_content(self, dataset_id: int) -> pd.DataFrame:
         query = "SELECT * FROM events WHERE dataset_id = %s"
         result = self.db.execute(query, (dataset_id,), fetch=True)
         return pd.DataFrame(result)
-    
+
     def get_dataset_info(self, dataset_id: int) -> dict:
         query = "SELECT * FROM datasets WHERE id = %s"
         result = self.db.execute(query, (dataset_id,), fetch=True)
@@ -35,14 +36,16 @@ class DatasetManager:
             raise NonExistentDatasetException(f"Dataset {dataset_id} does not exist")
 
         return result[0]
-    
+
     def save_dataset_info(self, user_id: int, dataset_name: str, topics: dict) -> int:
         query = """
             INSERT INTO datasets (user_id, name, topics)
             VALUES (%s, %s, %s)
             RETURNING id
         """
-        result = self.db.execute(query, (user_id, dataset_name, Json(topics)), fetch=True)
+        result = self.db.execute(
+            query, (user_id, dataset_name, Json(topics)), fetch=True
+        )
         return result[0]["id"] if result else None
 
     def save_dataset_content(self, dataset_id: int, event_data: pd.DataFrame):
@@ -113,7 +116,9 @@ class DatasetManager:
 
         self.db.execute_batch(query, values)
 
-    def set_dataset_status(self, dataset_id: int, status: str, status_message: str | None = None):
+    def set_dataset_status(
+        self, dataset_id: int, status: str, status_message: str | None = None
+    ):
         if status not in ["fetching", "processing", "complete", "error"]:
             raise ValueError("Invalid status")
 
@@ -137,24 +142,24 @@ class DatasetManager:
             WHERE id = %s
         """
 
-        result = self.db.execute(query, (dataset_id, ), fetch=True)
-        
+        result = self.db.execute(query, (dataset_id,), fetch=True)
+
         if not result:
             print(result)
             raise NonExistentDatasetException(f"Dataset {dataset_id} does not exist")
-        
+
         return result[0]
-    
+
     def update_dataset_name(self, dataset_id: int, new_name: str):
         query = "UPDATE datasets SET name = %s WHERE id = %s"
         self.db.execute(query, (new_name, dataset_id))
-    
+
     def delete_dataset_info(self, dataset_id: int):
         query = "DELETE FROM datasets WHERE id = %s"
 
-        self.db.execute(query, (dataset_id, ))
+        self.db.execute(query, (dataset_id,))
 
     def delete_dataset_content(self, dataset_id: int):
         query = "DELETE FROM events WHERE dataset_id = %s"
 
-        self.db.execute(query, (dataset_id, ))
+        self.db.execute(query, (dataset_id,))