style: isort 格式化

This commit is contained in:
2025-12-19 15:13:42 +08:00
parent cfb1385f4d
commit 243eea864b
54 changed files with 202 additions and 203 deletions

2
.gitignore vendored
View File

@@ -11,8 +11,6 @@ electron/test.toml
build/
dist/
old/
# Project specific directories
# config/
data/cache/
data/electron/

View File

@@ -3,8 +3,9 @@
以及基准路径
"""
from contextvars import ContextVar
import pathlib
from contextvars import ContextVar
from heurams.services.config import ConfigFile
from heurams.services.logger import get_logger

View File

@@ -1,10 +1,12 @@
from textual.app import App
from textual.widgets import Button
from heurams.services.logger import get_logger
from .screens.about import AboutScreen
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__)
@@ -66,11 +68,11 @@ def is_subdir(parent, child):
return 0
# 开发模式
from heurams.context import rootdir, workdir, config_var
from pathlib import Path
from heurams.context import rootdir
import os
from pathlib import Path
# 开发模式
from heurams.context import config_var, rootdir, workdir
if is_subdir(Path(rootdir), Path(os.getcwd())):
os.chdir(Path(rootdir) / ".." / "..")

View File

@@ -1,15 +1,8 @@
#!/usr/bin/env python3
from textual.app import ComposeResult
from textual.widgets import (
Header,
Footer,
Label,
Static,
Button,
Markdown,
)
from textual.containers import ScrollableContainer, ScrollableContainer
from textual.containers import ScrollableContainer
from textual.screen import Screen
from textual.widgets import Button, Footer, Header, Label, Markdown, Static
import heurams.services.version as version
from heurams.context import *

View File

@@ -1,26 +1,20 @@
#!/usr/bin/env python3
import pathlib
from textual.app import ComposeResult
from textual.widgets import (
Header,
Footer,
Label,
ListView,
ListItem,
Button,
Static,
)
from textual.containers import ScrollableContainer
from textual.screen import Screen
from textual.widgets import (Button, Footer, Header, Label, ListItem, ListView,
Static)
from heurams.kernel.particles import *
from heurams.context import *
import heurams.services.version as version
import heurams.services.timer as timer
from .preparation import PreparationScreen
from .about import AboutScreen
import heurams.services.version as version
from heurams.context import *
from heurams.kernel.particles import *
from heurams.services.logger import get_logger
import pathlib
from .about import AboutScreen
from .preparation import PreparationScreen
logger = get_logger(__name__)
@@ -51,8 +45,8 @@ class DashboardScreen(Screen):
res = dict()
filestem = pathlib.Path(filename).stem
res[0] = f"{filename}\0"
from heurams.kernel.particles.loader import load_electron
import heurams.kernel.particles as pt
from heurams.kernel.particles.loader import load_electron
electron_file_path = pathlib.Path(config_var.get()["paths"]["electron_dir"]) / (
filestem + ".json"

View File

@@ -1,16 +1,18 @@
#!/usr/bin/env python3
from textual.app import ComposeResult
from textual.widgets import Header, Footer, Label, Static, Button
from textual.containers import Center, ScrollableContainer
from textual.screen import Screen
from textual.reactive import reactive
from enum import Enum, auto
from heurams.services.logger import get_logger
from heurams.context import config_var
from heurams.kernel.reactor import *
from textual.app import ComposeResult
from textual.containers import Center, ScrollableContainer
from textual.reactive import reactive
from textual.screen import Screen
from textual.widgets import Button, Footer, Header, Label, Static
import heurams.kernel.particles as pt
import heurams.kernel.puzzles as pz
from heurams.context import config_var
from heurams.kernel.reactor import *
from heurams.services.logger import get_logger
from .. import shim

View File

@@ -1,21 +1,15 @@
#!/usr/bin/env python3
from pathlib import Path
import toml
from textual.app import ComposeResult
from textual.widgets import (
Header,
Footer,
Label,
Input,
Select,
Button,
Markdown,
)
from textual.containers import ScrollableContainer
from textual.screen import Screen
from textual.widgets import (Button, Footer, Header, Input, Label, Markdown,
Select)
from heurams.services.version import ver
import toml
from pathlib import Path
from heurams.context import config_var
from heurams.services.version import ver
class NucleonCreatorScreen(Screen):
@@ -27,6 +21,7 @@ class NucleonCreatorScreen(Screen):
def search_templates(self):
from pathlib import Path
from heurams.context import config_var
template_dir = Path(config_var.get()["paths"]["template_dir"])

View File

@@ -1,22 +1,15 @@
#!/usr/bin/env python3
from textual.app import ComposeResult
from textual.widgets import (
Header,
Footer,
Label,
Button,
Static,
ProgressBar,
)
from textual.containers import ScrollableContainer, Horizontal
from textual.containers import ScrollableContainer
from textual.screen import Screen
import pathlib
from textual.app import ComposeResult
from textual.containers import Horizontal, ScrollableContainer
from textual.screen import Screen
from textual.widgets import Button, Footer, Header, Label, ProgressBar, Static
from textual.worker import get_current_worker
import heurams.kernel.particles as pt
import heurams.services.hasher as hasher
from heurams.context import *
from textual.worker import get_current_worker
class PrecachingScreen(Screen):
@@ -96,7 +89,7 @@ class PrecachingScreen(Screen):
def precache_by_text(self, text: str):
"""预缓存单段文本的音频"""
from heurams.context import rootdir, workdir, config_var
from heurams.context import config_var, rootdir, workdir
cache_dir = pathlib.Path(config_var.get()["paths"]["cache_dir"])
cache_dir.mkdir(parents=True, exist_ok=True)
@@ -166,7 +159,7 @@ class PrecachingScreen(Screen):
def precache_all_files(self):
"""预缓存所有文件"""
from heurams.context import rootdir, workdir, config_var
from heurams.context import config_var, rootdir, workdir
nucleon_path = pathlib.Path(config_var.get()["paths"]["nucleon_dir"])
nucleon_files = [
@@ -220,7 +213,8 @@ class PrecachingScreen(Screen):
# 清空缓存
try:
import shutil
from heurams.context import rootdir, workdir, config_var
from heurams.context import config_var, rootdir, workdir
shutil.rmtree(
f"{config_var.get()["paths"]["cache_dir"]}", ignore_errors=True

View File

@@ -1,21 +1,15 @@
#!/usr/bin/env python3
from textual.app import ComposeResult
from textual.widgets import (
Header,
Footer,
Label,
Static,
Button,
Markdown,
)
from textual.containers import ScrollableContainer
from textual.reactive import reactive
from textual.screen import Screen
from heurams.context import config_var
from textual.widget import Widget
from textual.widgets import Button, Footer, Header, Label, Markdown, Static
import heurams.kernel.particles as pt
import heurams.services.hasher as hasher
from heurams.context import *
from textual.reactive import reactive
from textual.widget import Widget
from heurams.context import config_var
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,22 +1,15 @@
#!/usr/bin/env python3
from textual.app import ComposeResult
from textual.widgets import (
Header,
Footer,
Label,
Button,
Static,
ProgressBar,
)
from textual.containers import ScrollableContainer, Horizontal
from textual.containers import ScrollableContainer
from textual.screen import Screen
import pathlib
from textual.app import ComposeResult
from textual.containers import Horizontal, ScrollableContainer
from textual.screen import Screen
from textual.widgets import Button, Footer, Header, Label, ProgressBar, Static
from textual.worker import get_current_worker
import heurams.kernel.particles as pt
import heurams.services.hasher as hasher
from heurams.context import *
from textual.worker import get_current_worker
class SyncScreen(Screen):

View File

@@ -1,10 +1,11 @@
"""Kernel 操作辅助函数库"""
import random
from typing import TypedDict
import heurams.interface.widgets as pzw
import heurams.kernel.particles as pt
import heurams.kernel.puzzles as pz
import heurams.interface.widgets as pzw
from typing import TypedDict
staging = {} # 细粒度缓存区, 是 ident -> quality 的封装

View File

@@ -1,6 +1,8 @@
from typing import Iterable
from textual.app import ComposeResult
from textual.widget import Widget
import heurams.kernel.particles as pt

View File

@@ -1,13 +1,11 @@
from textual.widgets import (
Label,
Static,
Button,
)
from textual.containers import ScrollableContainer, Horizontal
from textual.widget import Widget
import heurams.kernel.particles as pt
from .base_puzzle_widget import BasePuzzleWidget
from textual.containers import Horizontal, ScrollableContainer
from textual.message import Message
from textual.widget import Widget
from textual.widgets import Button, Label, Static
import heurams.kernel.particles as pt
from .base_puzzle_widget import BasePuzzleWidget
class BasicEvaluation(BasePuzzleWidget):

View File

@@ -1,17 +1,17 @@
from textual.widgets import (
Label,
Button,
)
from textual.widget import Widget
import heurams.kernel.particles as pt
import heurams.kernel.puzzles as pz
from .base_puzzle_widget import BasePuzzleWidget
import copy
import random
from typing import TypedDict
from textual.containers import Container
from textual.message import Message
from textual.widget import Widget
from textual.widgets import Button, Label
import heurams.kernel.particles as pt
import heurams.kernel.puzzles as pz
from heurams.services.logger import get_logger
from typing import TypedDict
from .base_puzzle_widget import BasePuzzleWidget
logger = get_logger(__name__)

View File

@@ -1,8 +1,5 @@
from textual.widgets import (
Label,
Button,
)
from textual.widget import Widget
from textual.widgets import Button, Label
class Finished(Widget):

View File

@@ -1,17 +1,17 @@
# 单项选择题
from textual.widgets import (
Label,
Button,
)
from textual.containers import ScrollableContainer, Container
from typing import TypedDict
from textual.containers import Container, ScrollableContainer
from textual.widget import Widget
from textual.widgets import Button, Label
import heurams.kernel.particles as pt
import heurams.kernel.puzzles as pz
from .base_puzzle_widget import BasePuzzleWidget
from typing import TypedDict
from heurams.services.hasher import hash
from heurams.services.logger import get_logger
from .base_puzzle_widget import BasePuzzleWidget
logger = get_logger(__name__)

View File

@@ -1,8 +1,5 @@
from textual.widgets import (
Label,
Button,
)
from textual.widget import Widget
from textual.widgets import Button, Label
class Placeholder(Widget):

View File

@@ -1,20 +1,17 @@
from textual.reactive import reactive
from textual.widgets import (
Markdown,
Label,
Static,
Button,
)
from textual.containers import Center
from textual.widget import Widget
from typing import Dict
import heurams.kernel.particles as pt
import re
from .base_puzzle_widget import BasePuzzleWidget
from typing import TypedDict, List
from typing import Dict, List, TypedDict
from textual.containers import Center
from textual.message import Message
from textual.reactive import reactive
from textual.widget import Widget
from textual.widgets import Button, Label, Markdown, Static
import heurams.kernel.particles as pt
from heurams.services.logger import get_logger
from .base_puzzle_widget import BasePuzzleWidget
logger = get_logger(__name__)

View File

@@ -1,6 +1,7 @@
from .sm2 import SM2Algorithm
from heurams.services.logger import get_logger
from .sm2 import SM2Algorithm
logger = get_logger(__name__)
__all__ = [

View File

@@ -1,5 +1,6 @@
import heurams.services.timer as timer
from typing import TypedDict
import heurams.services.timer as timer
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,8 +1,10 @@
from .base import BaseAlgorithm
import heurams.services.timer as timer
from typing import TypedDict
import heurams.services.timer as timer
from heurams.services.logger import get_logger
from .base import BaseAlgorithm
logger = get_logger(__name__)

View File

@@ -9,12 +9,12 @@ from heurams.services.logger import get_logger
logger = get_logger(__name__)
logger.debug("粒子模块已加载")
from .atom import Atom, atom_registry
from .electron import Electron
from .loader import load_electron, load_nucleon
from .nucleon import Nucleon
from .orbital import Orbital
from .atom import Atom, atom_registry
from .probe import probe_all, probe_by_filename
from .loader import load_nucleon, load_electron
__all__ = [
"Electron",

View File

@@ -1,14 +1,17 @@
import json
import pathlib
import typing
from typing import TypedDict
import bidict
import toml
from heurams.context import config_var
from heurams.services.logger import get_logger
from .electron import Electron
from .nucleon import Nucleon
from .orbital import Orbital
from typing import TypedDict
import pathlib
import typing
import toml
import json
import bidict
from heurams.context import config_var
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,12 +1,15 @@
from .nucleon import Nucleon
from .electron import Electron
import heurams.services.hasher as hasher
import pathlib
import toml
import json
import pathlib
from copy import deepcopy
import toml
import heurams.services.hasher as hasher
from heurams.services.logger import get_logger
from .electron import Electron
from .nucleon import Nucleon
logger = get_logger(__name__)

View File

@@ -1,4 +1,5 @@
from typing import TypedDict
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,5 +1,6 @@
from heurams.context import config_var
import pathlib
from heurams.context import config_var
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,7 +1,9 @@
from .base import BasePuzzle
import random
from heurams.services.logger import get_logger
from .base import BasePuzzle
logger = get_logger(__name__)

View File

@@ -1,9 +1,11 @@
# mcq.py
from .base import BasePuzzle
import random
from typing import List, Dict, Optional, Union
from typing import Dict, List, Optional, Union
from heurams.services.logger import get_logger
from .base import BasePuzzle
logger = get_logger(__name__)

View File

@@ -1,8 +1,10 @@
# mcq.py
from .base import BasePuzzle
import random
from heurams.services.logger import get_logger
from .base import BasePuzzle
logger = get_logger(__name__)

View File

@@ -1,8 +1,9 @@
from .states import PhaserState, ProcessionState
from .procession import Procession
from heurams.services.logger import get_logger
from .fission import Fission
from .phaser import Phaser
from heurams.services.logger import get_logger
from .procession import Procession
from .states import PhaserState, ProcessionState
logger = get_logger(__name__)

View File

@@ -1,9 +1,11 @@
import random
import heurams.kernel.particles as pt
import heurams.kernel.puzzles as puz
import random
from .states import PhaserState
from heurams.services.logger import get_logger
from .states import PhaserState
class Fission:
"""裂变器: 单原子调度展开器"""

View File

@@ -1,10 +1,11 @@
# 移相器类定义
import heurams.kernel.particles as pt
from .states import PhaserState, ProcessionState
from .procession import Procession
from heurams.services.logger import get_logger
from .procession import Procession
from .states import PhaserState, ProcessionState
logger = get_logger(__name__)

View File

@@ -1,7 +1,8 @@
import heurams.kernel.particles as pt
from .states import PhaserState, ProcessionState
from heurams.services.logger import get_logger
from .states import PhaserState, ProcessionState
logger = get_logger(__name__)

View File

@@ -1,4 +1,5 @@
from enum import Enum, auto
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,8 +1,8 @@
# 音频播放器, 必须基于文件操作
from . import termux_audio
from . import playsound_audio
from heurams.services.logger import get_logger
from . import playsound_audio, termux_audio
logger = get_logger(__name__)
__all__ = [

View File

@@ -5,7 +5,9 @@
import os
import pathlib
import playsound
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,5 +1,6 @@
from typing import Protocol
import pathlib
from typing import Protocol
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -5,6 +5,7 @@
import os
import pathlib
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,6 +1,7 @@
from heurams.services.logger import get_logger
from .base import BaseTTS
from .edge_tts import EdgeTTS
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,4 +1,5 @@
import pathlib
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,8 +1,11 @@
from .base import BaseTTS
import pathlib
import edge_tts
from heurams.services.logger import get_logger
from .base import BaseTTS
logger = get_logger(__name__)

View File

@@ -1,7 +1,8 @@
# 音频服务
from typing import Callable
from heurams.context import config_var
from heurams.providers.audio import providers as prov
from typing import Callable
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,7 +1,9 @@
# 配置文件服务
import pathlib
import toml
import typing
import toml
from heurams.services.logger import get_logger

View File

@@ -1,5 +1,6 @@
# 哈希服务
import hashlib
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,6 +1,7 @@
# 时间服务
from heurams.context import config_var
import time
from heurams.context import config_var
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -1,7 +1,8 @@
# 文本转语音服务
from typing import Callable
from heurams.context import config_var
from heurams.providers.tts import TTSs
from typing import Callable
from heurams.services.logger import get_logger
logger = get_logger(__name__)

View File

@@ -2,17 +2,18 @@
"""
DashboardScreen 的测试, 包括单元测试和 pilot 测试.
"""
import unittest
import tempfile
import pathlib
import tempfile
import time
from unittest.mock import patch, MagicMock
import unittest
from unittest.mock import MagicMock, patch
from textual.pilot import Pilot
from heurams.context import ConfigContext
from heurams.services.config import ConfigFile
from heurams.interface.__main__ import HeurAMSApp
from heurams.interface.screens.dashboard import DashboardScreen
from heurams.services.config import ConfigFile
class TestDashboardScreenUnit(unittest.TestCase):
@@ -66,7 +67,7 @@ class TestDashboardScreenUnit(unittest.TestCase):
result = screen.compose()
widgets = list(result)
# 检查是否包含 Header 和 Footer
from textual.widgets import Header, Footer
from textual.widgets import Footer, Header
header_present = any(isinstance(w, Header) for w in widgets)
footer_present = any(isinstance(w, Footer) for w in widgets)

View File

@@ -1,5 +1,5 @@
import unittest
from unittest.mock import patch, MagicMock
from unittest.mock import MagicMock, patch
from heurams.kernel.algorithms.sm2 import SM2Algorithm

View File

@@ -1,15 +1,16 @@
import unittest
from unittest.mock import patch, MagicMock
import json
import pathlib
import tempfile
import toml
import json
import unittest
from unittest.mock import MagicMock, patch
import toml
from heurams.context import ConfigContext
from heurams.kernel.particles.atom import Atom, atom_registry
from heurams.kernel.particles.electron import Electron
from heurams.kernel.particles.nucleon import Nucleon
from heurams.kernel.particles.orbital import Orbital
from heurams.context import ConfigContext
from heurams.services.config import ConfigFile

View File

@@ -1,9 +1,9 @@
import unittest
from unittest.mock import patch, MagicMock
import sys
import unittest
from unittest.mock import MagicMock, patch
from heurams.kernel.particles.electron import Electron
from heurams.kernel.algorithms import algorithms
from heurams.kernel.particles.electron import Electron
class TestElectron(unittest.TestCase):

View File

@@ -1,5 +1,5 @@
import unittest
from unittest.mock import patch, MagicMock
from unittest.mock import MagicMock, patch
from heurams.kernel.particles.nucleon import Nucleon

View File

@@ -1,5 +1,5 @@
import unittest
from unittest.mock import patch, MagicMock
from unittest.mock import MagicMock, patch
from heurams.kernel.puzzles.cloze import ClozePuzzle

View File

@@ -1,5 +1,5 @@
import unittest
from unittest.mock import patch, MagicMock, call
from unittest.mock import MagicMock, call, patch
from heurams.kernel.puzzles.mcq import MCQPuzzle

View File

@@ -1,10 +1,10 @@
import unittest
from unittest.mock import Mock, patch, MagicMock
from unittest.mock import MagicMock, Mock, patch
from heurams.kernel.reactor.phaser import Phaser
from heurams.kernel.reactor.states import PhaserState, ProcessionState
from heurams.kernel.particles.atom import Atom
from heurams.kernel.particles.electron import Electron
from heurams.kernel.reactor.phaser import Phaser
from heurams.kernel.reactor.states import PhaserState, ProcessionState
class TestPhaser(unittest.TestCase):