mirror of
https://github.com/dobin/SuperMega
synced 2026-06-03 01:27:11 +00:00
Dobin
57 lines
1.6 KiB
Python
57 lines
1.6 KiB
Python
import pickle
|
|
import os
|
|
import yaml
|
|
import pickle
|
|
|
|
from typing import List, Tuple
|
|
from model.settings import Settings
|
|
from model.defs import *
|
|
|
|
|
|
class WebProject():
|
|
def __init__(self, name: str, settings: Settings):
|
|
self.name = name
|
|
self.settings: Settings = settings
|
|
self.comment: str = ""
|
|
|
|
|
|
class Storage():
|
|
def __init__(self):
|
|
pass
|
|
|
|
|
|
def get_projects(self) -> List[WebProject]:
|
|
projects: List[WebProject] = []
|
|
for project_name in os.listdir(PATH_WEB_PROJECT):
|
|
project = self.get_project(project_name)
|
|
if project is None:
|
|
continue
|
|
project.settings.prep_web(project_name)
|
|
projects.append(project)
|
|
return projects
|
|
|
|
|
|
def get_project(self, project_name: str) -> WebProject:
|
|
path = "{}/{}".format(PATH_WEB_PROJECT, project_name)
|
|
json_path = "{}/project.pickle".format(path)
|
|
if not os.path.exists(json_path):
|
|
return None
|
|
with open(json_path, "rb") as f:
|
|
project = pickle.load(f)
|
|
project.settings.prep_web(project_name)
|
|
return project
|
|
|
|
|
|
def add_project(self, project: WebProject):
|
|
# directories and contents
|
|
os.makedirs(PATH_WEB_PROJECT + project.name, exist_ok=True)
|
|
with open("{}/{}/project.pickle".format(PATH_WEB_PROJECT, project.name), "wb") as f:
|
|
pickle.dump(project, f)
|
|
|
|
|
|
def save_project(self, project: WebProject):
|
|
with open("{}/{}/project.pickle".format(PATH_WEB_PROJECT, project.name), "wb") as f:
|
|
pickle.dump(project, f)
|
|
|
|
|
|
storage = Storage() |