mirror of
https://github.com/shadoll/scatcher.git
synced 2025-12-20 03:26:59 +00:00
Add schemas
This commit is contained in:
58
manager/history.py
Normal file
58
manager/history.py
Normal file
@@ -0,0 +1,58 @@
|
||||
import json
|
||||
import os
|
||||
from schema.methods import Methods
|
||||
|
||||
class History:
|
||||
HISTORY_LIMIT = 10
|
||||
HISTORY_STORAGE = "storage"
|
||||
|
||||
def __init__(self, namespace: str = "requests") -> None:
|
||||
self.namespace = namespace
|
||||
self.filename = f"{self.HISTORY_STORAGE}/{namespace}.json"
|
||||
self.data: list = []
|
||||
self.check_directory()
|
||||
|
||||
def check_directory(self) -> None:
|
||||
if not os.path.exists(self.HISTORY_STORAGE):
|
||||
os.makedirs(self.HISTORY_STORAGE)
|
||||
|
||||
def load(self) -> None:
|
||||
try:
|
||||
with open(self.filename, "r") as f:
|
||||
self.data = json.load(f)
|
||||
except (FileNotFoundError, json.JSONDecodeError):
|
||||
self.data = []
|
||||
|
||||
def save(self) -> None:
|
||||
with open(self.filename, "w") as f:
|
||||
json.dump(self.data, f, indent=4)
|
||||
|
||||
def clear(self) -> None:
|
||||
self.data = []
|
||||
self.save()
|
||||
|
||||
def all(self) -> list | None:
|
||||
self.load()
|
||||
if len(self.data) == 0:
|
||||
return None
|
||||
for item in self.data:
|
||||
if "method" in item:
|
||||
item["method"] = Methods(item["method"])
|
||||
return self.data
|
||||
|
||||
def get(self, index: int) -> dict | None:
|
||||
self.all()
|
||||
|
||||
return self.data[index] if index < len(self.data) else None
|
||||
|
||||
def last(self) -> dict | None:
|
||||
self.all()
|
||||
|
||||
return self.data[-1] if len(self.data) > 0 else None
|
||||
|
||||
def add(self, request_data: dict) -> None:
|
||||
self.load()
|
||||
self.data.append(request_data)
|
||||
if len(self.data) > self.HISTORY_LIMIT:
|
||||
self.data.pop(0)
|
||||
self.save()
|
||||
Reference in New Issue
Block a user