64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from textual.app import App
|
|
from textual.widgets import Button
|
|
|
|
from heurams.context import config_var
|
|
from heurams.services.logger import get_logger
|
|
|
|
from .screens.about import AboutScreen
|
|
from .screens.dashboard import DashboardScreen
|
|
from .screens.precache import PrecachingScreen
|
|
from .screens.repocreator import RepoCreatorScreen
|
|
from .screens.synctool import SyncScreen
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
def environment_check():
|
|
from pathlib import Path
|
|
|
|
logger.debug("检查环境路径")
|
|
subdir = ['cache/voice', 'repo', 'global', 'config']
|
|
for i in subdir:
|
|
i = Path(config_var.get()['paths']['data']) / i
|
|
if not i.exists():
|
|
logger.info("创建目录: %s", i)
|
|
print(f"创建 {i}")
|
|
i.mkdir(exist_ok=True, parents=True)
|
|
else:
|
|
logger.debug("目录已存在: %s", i)
|
|
print(f"找到 {i}")
|
|
logger.debug("环境检查完成")
|
|
|
|
|
|
class HeurAMSApp(App):
|
|
TITLE = "潜进"
|
|
CSS_PATH = "css/main.tcss"
|
|
SUB_TITLE = "启发式辅助记忆调度器"
|
|
BINDINGS = [
|
|
("q", "quit", "退出"),
|
|
("d", "toggle_dark", "切换色调"),
|
|
("1", "app.push_screen('dashboard')", "仪表盘"),
|
|
("2", "app.push_screen('precache_all')", "缓存管理器"),
|
|
("3", "app.push_screen('repo_creator')", "创建新仓库"),
|
|
# ("4", "app.push_screen('synctool')", "同步工具"),
|
|
("0", "app.push_screen('about')", "版本信息"),
|
|
]
|
|
SCREENS = {
|
|
"dashboard": DashboardScreen,
|
|
"repo_creator": RepoCreatorScreen,
|
|
"precache_all": PrecachingScreen,
|
|
"synctool": SyncScreen,
|
|
"about": AboutScreen,
|
|
}
|
|
|
|
def on_mount(self) -> None:
|
|
environment_check()
|
|
self.push_screen("dashboard")
|
|
|
|
def on_button_pressed(self, event: Button.Pressed) -> None:
|
|
self.exit(event.button.id)
|
|
|
|
def action_do_nothing(self):
|
|
print("DO NOTHING")
|
|
self.refresh()
|