Spaces:
No application file
No application file
File size: 1,221 Bytes
2875866 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
PROJECTS_DIR = "projects"
DATASET_DIR = "dataset"
import json
import os
class Project:
def __init__(self):
pass
def list_projects(self):
return [f.replace(".json", "") for f in os.listdir(PROJECTS_DIR) if f.endswith(".json")]
def load_project(self,project_name):
file_path = os.path.join(PROJECTS_DIR, f"{project_name}.json")
if os.path.exists(file_path):
with open(file_path, "r") as f:
return json.load(f)
return None
def save_project(self,project_name, data):
file_path = os.path.join(PROJECTS_DIR, f"{project_name}.json")
with open(file_path, "w") as f:
json.dump(data, f, indent=4)
def initialize_project(self,project_name):
data = {
"project_name": project_name,
"assertions": {"deterministic": [], "misc": [], "factual": "", "sql-only": False, "json-only": False},
"log_history": [],
"accuracy_history": {
"DSA" : [],
"Technical" : [],
"Behaviour": []
},
}
self.save_project(project_name, data)
return data |