78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
from textual.app import App
|
|
from textual.widgets import Button
|
|
from .screens.dashboard import DashboardScreen
|
|
from .screens.nucreator import NucleonCreatorScreen
|
|
from .screens.precache import PrecachingScreen
|
|
from .screens.about import AboutScreen
|
|
from heurams.services.logger import get_logger
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
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('nucleon_creator')", "创建新单元"),
|
|
("0", "app.push_screen('about')", "版本信息"),
|
|
]
|
|
SCREENS = {
|
|
"dashboard": DashboardScreen,
|
|
"nucleon_creator": NucleonCreatorScreen,
|
|
"precache_all": PrecachingScreen,
|
|
"about": AboutScreen,
|
|
}
|
|
|
|
def on_mount(self) -> None:
|
|
self.push_screen("dashboard")
|
|
|
|
def on_button_pressed(self, event: Button.Pressed) -> None:
|
|
self.exit(event.button.id)
|
|
|
|
|
|
def environment_check():
|
|
from pathlib import Path
|
|
logger.debug("检查环境路径")
|
|
|
|
for i in config_var.get()["paths"].values():
|
|
i = Path(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("环境检查完成")
|
|
|
|
|
|
def is_subdir(parent, child):
|
|
try:
|
|
child.relative_to(parent)
|
|
logger.debug("is_subdir: %s 是 %s 的子目录", child, parent)
|
|
return 1
|
|
except:
|
|
logger.debug("is_subdir: %s 不是 %s 的子目录", child, parent)
|
|
return 0
|
|
|
|
|
|
# 开发模式
|
|
from heurams.context import rootdir, workdir, config_var
|
|
from pathlib import Path
|
|
from heurams.context import rootdir
|
|
import os
|
|
|
|
if is_subdir(Path(rootdir), Path(os.getcwd())):
|
|
os.chdir(Path(rootdir) / ".." / "..")
|
|
print(f'转入开发数据目录: {Path(rootdir)/".."/".."}')
|
|
|
|
environment_check()
|
|
|
|
app = HeurAMSApp()
|
|
app.run()
|