initial commit

.gitignore

@@ -0,0 +1,198 @@
+# Created by https://www.toptal.com/developers/gitignore/api/visualstudiocode,python
+# Edit at https://www.toptal.com/developers/gitignore?templates=visualstudiocode,python
+
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+### Python Patch ###
+# Poetry local configuration file - https://python-poetry.org/docs/configuration/#local-configuration
+poetry.toml
+
+# ruff
+.ruff_cache/
+
+# LSP config files
+pyrightconfig.json
+
+### VisualStudioCode ###
+.vscode/*
+!.vscode/settings.json
+!.vscode/tasks.json
+!.vscode/launch.json
+!.vscode/extensions.json
+!.vscode/*.code-snippets
+
+# Local History for Visual Studio Code
+.history/
+
+# Built Visual Studio Code Extensions
+*.vsix
+
+### VisualStudioCode Patch ###
+# Ignore all local history of files
+.history
+.ionide
+
+# End of https://www.toptal.com/developers/gitignore/api/visualstudiocode,python
+
+# log path
+log/*

cfg_main.toml

@@ -0,0 +1,14 @@
+[debug]
+logger_filename = "log/file_{time}.log"
+logger_format = "{time} {level} {message}"
+
+[task]
+GetBlock_enable = true # 人员施救使能
+PutBlock_enable = false # 紧急转移使能
+GetBBall_enable = true # 整装上阵使能
+UpTower_enable = true  # 通信抢修使能
+GetRBall_enable = true # 高空排险使能
+PutBBall_enable = true # 派发物资使能
+PutHanoi_enable = true # 物资盘点使能
+MoveArea_enable = true # 应急避险使能
+KickAss_enable = true  # 扫黑除暴使能

main.py

@@ -0,0 +1,42 @@
+import toml
+import queue
+import threading
+from loguru import logger
+import subtask as sb
+
+# 读取配置
+cfg_main = toml.load('cfg_main.toml')
+
+# 配置日志输出
+logger.add(cfg_main['debug'].get('logger_filename'), format=cfg_main['debug'].get('logger_format'), level="INFO")
+
+task_queue = queue.Queue()  
+
+task_queue.put(sb.task(sb.get_block, cfg_main['task'].get('GetBlock_enable'))) 
+task_queue.put(sb.task(sb.put_block, cfg_main['task'].get('PutBlock_enable'))) 
+task_queue.put(sb.task(sb.get_bball, cfg_main['task'].get('GetBBall_enable'))) 
+task_queue.put(sb.task(sb.up_tower, cfg_main['task'].get('UpTower_enable'))) 
+task_queue.put(sb.task(sb.get_rball, cfg_main['task'].get('GetRBall_enable'))) 
+task_queue.put(sb.task(sb.put_bball, cfg_main['task'].get('PutBBall_enable'))) 
+task_queue.put(sb.task(sb.put_hanoi, cfg_main['task'].get('PutHanoi_enable'))) 
+task_queue.put(sb.task(sb.move_area, cfg_main['task'].get('MoveArea_enable'))) 
+task_queue.put(sb.task(sb.kick_ass, cfg_main['task'].get('KickAss_enable'))) 
+
+task_queuem_t = sb.task_queuem(task_queue)
+
+def worker_thread():
+    while task_queuem_t.exec() is True:
+        pass
+
+worker = threading.Thread(target=worker_thread, daemon=True)  
+worker.start()  
+
+# 主线程仅在子线程搜索 (SEARCHING) 和 空闲 (IDLE) 状态下进行操作
+while task_queuem_t.busy is True:
+    if task_queuem_t.status is sb.task_queuem_status.EXECUTING:
+        pass
+    else:
+        # 模拟执行回归任务
+        logger.info("***** sim huigui task *****")
+
+logger.info("Main thread exit")

subtask.py

@@ -0,0 +1,108 @@
+from enum import Enum
+from loguru import logger
+import time
+
+# 任务类
+class task:  
+    def __init__(self, task_operation, enable=True):  
+        self.enable = enable
+        self.task_operation = task_operation  
+    def check(self):
+        # 检查该任务是否需要执行
+        # TODO 完善该接口，是否需要单独为每种 task 编写一个函数，还是设置一个通用的过滤器（从 detection 模块过滤结果）
+        ret = True
+        return ret
+    def execute(self):  
+        # 根据标志位确定是否执行该任务
+        if self.enable is True:
+            logger.info(f"[Task]#Executing task \"{self.task_operation.__name__}\"")  
+            self.task_operation()  
+            logger.info(f"[Task]#Task \"{self.task_operation.__name__}\" completed.")  
+        else:
+            logger.warning(f"[Task]#Skip task \"{self.task_operation.__name__}\"")
+
+# 任务队列状态类
+class task_queuem_status(Enum):
+    IDEL      = 0
+    SEARCHING = 1
+    EXECUTING = 2
+
+# 任务队列类 非EXECUTEING 时均执行 huigui，注意互斥操作
+class task_queuem(task):
+    # task_now = task(None, False)
+    def __init__(self, queue):
+        super(task_queuem, self)
+        self.queue = queue
+        self.status = task_queuem_status.IDEL
+        self.busy = True
+        logger.info(f"[TaskM]#Task num {self.queue.qsize()}")
+    def exec(self):
+        # 如果空闲状态则将下一个队列任务取出
+        if self.status is task_queuem_status.IDEL:
+            if self.queue.qsize() == 0:
+                self.busy = False
+                logger.info(f"[TaskM]#Task queue empty, exit")
+                return False
+            self.task_now = self.queue.get()
+
+            # 如果当前任务没有使能，则直接转入执行状态，由任务执行函数打印未执行信息
+            if self.task_now.enable is True:
+                self.status = task_queuem_status.SEARCHING
+            else:
+                self.status = task_queuem_status.EXECUTING
+
+            logger.info(f"[TaskM]#Start process task \"{self.task_now.task_operation.__name__}\" >>>>")
+        # 阻塞搜索任务标志位
+        elif self.status is task_queuem_status.SEARCHING:
+            logger.info(f"[TaskM]#Start searching task target")
+            while self.task_now.check() is False: # TODO 增加超时处理
+                break
+            self.status = task_queuem_status.EXECUTING
+        # 执行任务函数
+        elif self.status is task_queuem_status.EXECUTING:
+            logger.info(f"[TaskM]#Start execute task function")
+            self.task_now.execute()                 # 执行当前任务函数
+            self.queue.task_done()                  # 弹出已执行的任务
+            self.status = task_queuem_status.IDEL   #
+            logger.info(f"[TaskM]#finish process task {self.task_now.task_operation.__name__} <<<<")
+        return True
+
+
+
+
+# 人员施救
+def get_block():
+    pass
+
+# 紧急转移
+def put_block():
+    pass
+
+# 整装上阵
+def get_bball():
+    pass
+
+# 通信抢修
+def up_tower():
+    pass
+
+# 高空排险
+def get_rball():
+    pass
+
+# 派发物资
+def put_bball():
+    pass
+
+# 物资盘点
+def put_hanoi():
+    pass
+
+# 应急避险
+def move_area():
+    pass
+
+# 扫黑除暴
+def kick_ass():
+    pass
+