ksaa 2025.5.16.1__py3-none-any.whl → 2025.5.23.1__py3-none-any.whl

kotonebot/kaa/main/cli.py

@@ -1,3 +1,4 @@
+import os
 import sys
 import runpy
 import logging
@@ -16,6 +17,8 @@ psr.add_argument('-v', '--version', action='version', version='kaa v' + version)
 psr.add_argument('-c', '--config', default='./config.json', help='Path to the configuration file. Default: ./config.json')
 psr.add_argument('-lp', '--log-path', default=None, help='Path to the log file. Does not log to file if not specified. Default: None')
 psr.add_argument('-ll', '--log-level', default='DEBUG', help='Log level. Default: DEBUG')
+psr.add_argument('--kill-dmm', action='store_true', default=False, help='Kill DMM Game Player when tasks are completed. Overrides config().end_game.kill_dmm')
+psr.add_argument('--kill-game', action='store_true', default=False, help='Kill gakumasu.exe when tasks are completed. Overrides config().end_game.kill_game')
 
 # 子命令
 subparsers = psr.add_subparsers(dest='subcommands', title='Subcommands')
@@ -65,6 +68,10 @@ def task_invoke() -> int:
         kaa().run_all()
     else:
         kaa().run(tasks_from_id(tasks_args))
+    if psr.parse_args().kill_dmm:
+        os.system('taskkill /f /im DMMGamePlayer.exe')
+    if psr.parse_args().kill_game:
+        os.system('taskkill /f /im gakumas.exe')
     return 0
 
 def task_list() -> int:

kotonebot/kaa/main/dmm_host.py

@@ -0,0 +1,53 @@
+from typing_extensions import override
+
+from kotonebot.client import Device, create_device
+from kotonebot.client import DeviceImpl, Device
+from kotonebot.client.host import HostProtocol, Instance
+
+# TODO: 可能应该把 start_game 和 end_game 里对启停的操作移动到这里来
+class DmmInstance(Instance):
+    def __init__(self):
+        super().__init__('dmm', 'gakumas')
+
+    @override
+    def refresh(self):
+        raise NotImplementedError()
+
+    @override
+    def start(self):
+        raise NotImplementedError()
+
+    @override
+    def stop(self):
+        raise NotImplementedError()
+
+    @override
+    def running(self) -> bool:
+        raise NotImplementedError()
+
+    @override
+    def wait_available(self, timeout: float = 180):
+        raise NotImplementedError()
+
+    @override
+    def create_device(self, impl: DeviceImpl, *, timeout: float = 180) -> Device:
+        if impl not in ['windows', 'remote_windows']:
+            raise ValueError(f'Unsupported device implementation: {impl}')
+        return create_device('', impl, timeout=timeout)
+
+class DmmHost(HostProtocol):
+    instance = DmmInstance()
+    """DmmInstance 单例。"""
+
+    @staticmethod
+    def installed() -> bool:
+        # TODO: 应该检查 DMM 和 gamkumas 的安装情况
+        raise NotImplementedError()
+
+    @staticmethod
+    def list() -> list[Instance]:
+        raise NotImplementedError()
+
+    @staticmethod
+    def query(*, id: str) -> Instance | None:
+        raise NotImplementedError()