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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
import os
import shutil
from typing import Optional, Tuple
import yaml
from .. import log
from ..types import CommandList
from .path import getConfigPath, getConfigFile
CONST_rootPath: str = "rootPath"
CONST_after: str = "after"
CONST_subject: str = "subject"
class Template:
def __init__(self, subject: str, rootpath: str):
self.subject: str = subject
self.rootPath: str = rootpath
self.after: CommandList = []
def load(basepath: str) -> Tuple[Optional[Template], Optional[Exception]]:
if not os.path.isdir(getConfigPath(basepath)) or not os.path.isfile(getConfigFile(basepath)):
return Template(os.path.split(basepath)[-1], basepath), None
ret = Template("", "")
exp = None
try:
with open(getConfigFile(basepath), "r", encoding='utf-8') as f:
config = yaml.load(f.read())
ret.subject = config[CONST_subject]
ret.rootPath = os.path.join(basepath, config[CONST_rootPath])
ret.after = config[CONST_after]
except Exception as e:
log.errorWithException(f"Loading template failed from {basepath}")
exp = e
return ret, exp
def clear(basepath: str)->None:
oipath = getConfigPath(basepath)
if os.path.isdir(oipath):
log.debug(f"Clear template data at {basepath}")
shutil.rmtree(oipath)
def initialize(basepath: str)->None:
clear(basepath)
log.debug(f"Initialize template data at {basepath}")
if not os.path.isdir(basepath):
os.mkdir(basepath)
from ..core.manager import initializeCodeDirectory
initializeCodeDirectory(basepath)
oipath = getConfigPath(basepath)
os.mkdir(oipath)
config = {CONST_subject: os.path.split(basepath)[-1],
CONST_rootPath: "",
CONST_after: [], }
with open(getConfigFile(basepath), "w", encoding='utf-8') as f:
f.write(yaml.dump(config, indent=4,
default_flow_style=False))
|