diff --git a/src/dashboard.py b/src/dashboard.py index 443ce4e..51659dd 100644 --- a/src/dashboard.py +++ b/src/dashboard.py @@ -19,6 +19,7 @@ from modules.Email import EmailSender from modules.Log import Log from modules.DashboardLogger import DashboardLogger from modules.PeerJobLogger import PeerJobLogger +from modules.PeerJob import PeerJob DASHBOARD_VERSION = 'v4.2.0' @@ -55,43 +56,12 @@ def ResponseObject(status=True, message=None, data=None) -> Flask.response_class response.content_type = "application/json" return response -""" -Peer Job -""" -class PeerJob: - def __init__(self, JobID: str, Configuration: str, Peer: str, - Field: str, Operator: str, Value: str, CreationDate: datetime, ExpireDate: datetime, Action: str): - self.Action = Action - self.ExpireDate = ExpireDate - self.CreationDate = CreationDate - self.Value = Value - self.Operator = Operator - self.Field = Field - self.Configuration = Configuration - self.Peer = Peer - self.JobID = JobID - def toJson(self): - return { - "JobID": self.JobID, - "Configuration": self.Configuration, - "Peer": self.Peer, - "Field": self.Field, - "Operator": self.Operator, - "Value": self.Value, - "CreationDate": self.CreationDate, - "ExpireDate": self.ExpireDate, - "Action": self.Action - } - - def __dict__(self): - return self.toJson() """ Peer Jobs """ class PeerJobs: - def __init__(self): self.Jobs: list[PeerJob] = [] self.jobdb = sqlite3.connect(os.path.join(CONFIGURATION_PATH, 'db', 'wgdashboard_job.db'), diff --git a/src/modules/PeerJob.py b/src/modules/PeerJob.py new file mode 100644 index 0000000..83d75c3 --- /dev/null +++ b/src/modules/PeerJob.py @@ -0,0 +1,32 @@ +""" +Peer Job +""" +from datetime import datetime +class PeerJob: + def __init__(self, JobID: str, Configuration: str, Peer: str, + Field: str, Operator: str, Value: str, CreationDate: datetime, ExpireDate: datetime, Action: str): + self.Action = Action + self.ExpireDate = ExpireDate + self.CreationDate = CreationDate + self.Value = Value + self.Operator = Operator + self.Field = Field + self.Configuration = Configuration + self.Peer = Peer + self.JobID = JobID + + def toJson(self): + return { + "JobID": self.JobID, + "Configuration": self.Configuration, + "Peer": self.Peer, + "Field": self.Field, + "Operator": self.Operator, + "Value": self.Value, + "CreationDate": self.CreationDate, + "ExpireDate": self.ExpireDate, + "Action": self.Action + } + + def __dict__(self): + return self.toJson() \ No newline at end of file