missionpanel 1.2.4__tar.gz → 1.2.6__tar.gz

{missionpanel-1.2.4 → missionpanel-1.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: missionpanel
-Version: 1.2.4
+Version: 1.2.6
 Summary: A mission panel
 Home-page: https://github.com/yindaheng98/missionpanel
 Author: yindaheng98

{missionpanel-1.2.4 → missionpanel-1.2.6}/missionpanel/example/__init__.py

@@ -1,2 +1,3 @@
 from .rsshub import RSSHubSubmitter, RSSHubRootSubmitter, RSSHubSubitemSubmitter
 from .ttrss import TTRSSClient, TTRSSSubmitter, TTRSSHubSubmitter, TTRSSHubRootSubmitter, TTRSSHubSubitemSubmitter
+from .subprocess import SubprocessAsyncHandler

missionpanel-1.2.6/missionpanel/example/subprocess.py

@@ -0,0 +1,34 @@
+import abc
+import asyncio
+import logging
+from typing import List, Callable
+from missionpanel.handler import AsyncHandler
+from missionpanel.orm.core import Mission
+from missionpanel.orm.handler import Attempt
+
+
+class SubprocessAsyncHandler(AsyncHandler, abc.ABC):
+    def getLogger(self) -> logging.Logger:
+        return logging.getLogger("SubprocessAsyncHandler")
+
+    async def __readline_info(self, f):
+        async for line in f:
+            self.getLogger().info(
+                'stdout | %s' % line.strip())
+
+    async def __readline_debug(self, f):
+        async for line in f:
+            self.getLogger().info(
+                'stderr | %s' % line.strip())
+
+    @abc.abstractmethod
+    async def construct_command(self, mission: Mission, attempt: Attempt) -> str:
+        raise NotImplementedError("Subclasses must implement construct_command")
+
+    async def execute_mission(self, mission: Mission, attempt: Attempt) -> bool:
+        proc = await asyncio.create_subprocess_shell(
+            await self.construct_command(mission, attempt),
+            stdout=asyncio.subprocess.PIPE,
+            stderr=asyncio.subprocess.PIPE)
+        await asyncio.gather(self.__readline_info(proc.stdout), self.__readline_debug(proc.stderr))
+        return (await proc.wait()) == 0

{missionpanel-1.2.4 → missionpanel-1.2.6}/missionpanel/handler/handler.py

@@ -96,14 +96,24 @@ class AsyncHandler(HandlerInterface, abc.ABC):
     async def execute_mission(self, mission: Mission, attempt: Attempt) -> bool:
         pass
 
-    async def run_once(self, tags: List[str]):
+    async def get_mission(self, tags: List[str]) -> Optional[Mission]:
         missions = (await self.session.execute(HandlerInterface.query_todo_missions(tags))).scalars().all()
-        mission = await self.select_mission(missions)
-        if mission is None:
-            return
+        return await self.select_mission(missions)
+
+    async def run_mission(self, mission: Mission):
         attempt = HandlerInterface.create_attempt(self.session, mission, self.name, self.max_time_interval)
         await self.report_attempt(mission, attempt)
         if await self.execute_mission(mission, attempt):
             attempt.success = True
         await self.report_attempt(mission, attempt)
         return attempt
+
+    async def run_once(self, tags: List[str]):
+        mission = await self.get_mission(tags)
+        if mission is None:
+            return
+        return await self.run_mission(mission)
+
+    async def run_all(self, tags: List[str]):
+        while await self.run_once(tags):
+            pass

{missionpanel-1.2.4 → missionpanel-1.2.6}/missionpanel.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: missionpanel
-Version: 1.2.4
+Version: 1.2.6
 Summary: A mission panel
 Home-page: https://github.com/yindaheng98/missionpanel
 Author: yindaheng98

{missionpanel-1.2.4 → missionpanel-1.2.6}/missionpanel.egg-info/SOURCES.txt

@@ -9,6 +9,7 @@ missionpanel.egg-info/requires.txt
 missionpanel.egg-info/top_level.txt
 missionpanel/example/__init__.py
 missionpanel/example/rsshub.py
+missionpanel/example/subprocess.py
 missionpanel/example/ttrss.py
 missionpanel/handler/__init__.py
 missionpanel/handler/handler.py

{missionpanel-1.2.4 → missionpanel-1.2.6}/setup.py

@@ -12,7 +12,7 @@ package_dir = {
 
 setup(
     name='missionpanel',
-    version='1.2.4',
+    version='1.2.6',
     author='yindaheng98',
     author_email='yindaheng98@gmail.com',
     url='https://github.com/yindaheng98/missionpanel',