chore: add .pylint and clean up code

This commit is contained in:
agatha 2024-09-07 16:27:49 -04:00
parent 4d398654cb
commit a3329c0a7f
4 changed files with 12 additions and 8 deletions

5
server/.pylintrc Normal file
View File

@ -0,0 +1,5 @@
[MASTER]
init-hook='import sys; sys.path.append("src")'
[MESSAGES CONTROL]
disable=R0903,C0114,C0115,C0116

View File

@ -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()

View File

@ -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())

View File

@ -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