66 lines
1.4 KiB
Python
66 lines
1.4 KiB
Python
import sys
|
||
from typing import Optional, Callable, Tuple, Union
|
||
|
||
if "/www/server/panel/class" not in sys.path:
|
||
sys.path.insert(0, "/www/server/panel/class")
|
||
|
||
import public
|
||
|
||
ExecShell: Callable = public.ExecShell
|
||
write_log: Callable[[str, str, Tuple], Union[int, str, type(None)]] = public.WriteLog
|
||
|
||
|
||
def write_file(filename: str, s_body: str, mode='w+') -> bool:
|
||
"""
|
||
写入文件内容
|
||
@filename 文件名
|
||
@s_body 欲写入的内容
|
||
return bool 若文件不存在则尝试自动创建
|
||
"""
|
||
try:
|
||
fp = open(filename, mode=mode)
|
||
fp.write(s_body)
|
||
fp.close()
|
||
return True
|
||
except:
|
||
try:
|
||
fp = open(filename, mode=mode, encoding="utf-8")
|
||
fp.write(s_body)
|
||
fp.close()
|
||
return True
|
||
except:
|
||
return False
|
||
|
||
|
||
def read_file(filename, mode='r') -> Optional[str]:
|
||
"""
|
||
读取文件内容
|
||
@filename 文件名
|
||
return string(bin) 若文件不存在，则返回None
|
||
"""
|
||
import os
|
||
if not os.path.exists(filename):
|
||
return None
|
||
fp = None
|
||
try:
|
||
fp = open(filename, mode=mode)
|
||
f_body = fp.read()
|
||
except:
|
||
return None
|
||
finally:
|
||
if fp and not fp.closed:
|
||
fp.close()
|
||
return f_body
|
||
|
||
|
||
class _DB:
|
||
|
||
def __call__(self, table: str):
|
||
import db
|
||
with db.Sql() as t:
|
||
t.table(table)
|
||
return t
|
||
|
||
|
||
DB = _DB()
|