From a3329c0a7f6c8b0b45a5ab93678f48610b5f6a63 Mon Sep 17 00:00:00 2001
From: agatha <agatha@juggalol.com>
Date: Sat, 7 Sep 2024 16:27:49 -0400
Subject: [PATCH] chore: add .pylint and clean up code

---
 server/.pylintrc              | 5 +++++
 server/src/database/models.py | 2 +-
 server/src/main.py            | 8 +++-----
 server/src/utils/config.py    | 5 +++--
 4 files changed, 12 insertions(+), 8 deletions(-)
 create mode 100644 server/.pylintrc

diff --git a/server/.pylintrc b/server/.pylintrc
new file mode 100644
index 0000000..8ad4571
--- /dev/null
+++ b/server/.pylintrc
@@ -0,0 +1,5 @@
+[MASTER]
+init-hook='import sys; sys.path.append("src")'
+
+[MESSAGES CONTROL]
+disable=R0903,C0114,C0115,C0116
diff --git a/server/src/database/models.py b/server/src/database/models.py
index aa71039..44ccc84 100644
--- a/server/src/database/models.py
+++ b/server/src/database/models.py
@@ -1,6 +1,6 @@
+from datetime import datetime
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy import Column, Integer, String, DateTime
-from datetime import datetime
 
 Base = declarative_base()
 
diff --git a/server/src/main.py b/server/src/main.py
index cf22472..2d3c36d 100644
--- a/server/src/main.py
+++ b/server/src/main.py
@@ -1,11 +1,12 @@
 import json
 import socket
 import threading
+from datetime import datetime
+from loguru import logger
+
 import database
 from database.models import Agent
 from utils.config import load_yaml_config
-from loguru import logger
-from datetime import datetime
 
 
 class Server:
@@ -61,9 +62,6 @@ class Server:
             conn (socket.socket): The socket object for the connection to the
                 agent.
             addr (tuple): The address of the agent.
-
-        Raises:
-            Exception: If there is an error handling the agent.
         """
         try:
             agent_info = json.loads(conn.recv(1024).decode())
diff --git a/server/src/utils/config.py b/server/src/utils/config.py
index 6d961bd..bf7669e 100644
--- a/server/src/utils/config.py
+++ b/server/src/utils/config.py
@@ -17,9 +17,10 @@ def load_yaml_config(file_path):
         dict: A dictionary containing the configuration data.
     """
     try:
-        with open(file_path, "r") as f:
+        with open(file_path, "r", encoding="utf-8") as f:
             config = yaml.safe_load(f)
-    except Exception as e:
+    except Exception:
+        # TODO: Log error
         config = {
             "host": "0.0.0.0",
             "port": 9999