zwf
93 lines
2.0 KiB
Python
93 lines
2.0 KiB
Python
"""
|
|
系统服务，用于读取服务配置文件，启动或停止相关的服务。
|
|
"""
|
|
import logging
|
|
import os
|
|
import sys
|
|
from typing import Optional
|
|
|
|
from dock.oa import oa_security
|
|
from paste.core.logging import echo_log, set_logger_config
|
|
from paste.service.task_service import TaskService
|
|
|
|
logger_config_name = 'logger.oa_service'
|
|
"""
|
|
配置文件中日志配置字段名称。
|
|
"""
|
|
|
|
task_serv: Optional[TaskService] = None
|
|
"""
|
|
任务服务对象。
|
|
"""
|
|
|
|
pid_file = os.path.join(os.path.curdir, 'oa_service.pid')
|
|
"""
|
|
PID 文件路径。
|
|
"""
|
|
|
|
service_name = 'OA计划任务服务'
|
|
"""
|
|
服务名称。
|
|
"""
|
|
|
|
|
|
def init_task_service():
|
|
"""
|
|
初始化服务对象并安装具体任务。
|
|
"""
|
|
global task_serv
|
|
task_serv = TaskService(service_name=service_name, pid_file=pid_file)
|
|
|
|
# 每隔 10 分钟执行，更新 OA Token
|
|
task_serv.add_task(creator=task_serv.create_delay_task, fn=renew_oa_token, delay=60 * 10)
|
|
|
|
return task_serv
|
|
|
|
|
|
async def renew_oa_token():
|
|
try:
|
|
echo_log(f"开始执行 OA Token 更新...")
|
|
await oa_security.login()
|
|
echo_log(f"完成 OA Token 更新.")
|
|
except Exception as e:
|
|
echo_log(msg=e, level=logging.ERROR, is_log_exc=True)
|
|
|
|
|
|
def start_service():
|
|
"""
|
|
控制台服务方式启动。
|
|
"""
|
|
set_logger_config(logger_config_name)
|
|
_ts = init_task_service()
|
|
_ts.start_service(env_check=False)
|
|
|
|
|
|
def start():
|
|
"""
|
|
驻内存服务方式启动。
|
|
"""
|
|
set_logger_config(logger_config_name)
|
|
_ts = init_task_service()
|
|
_ts.start()
|
|
|
|
|
|
def stop():
|
|
"""
|
|
驻内存服务方式停止。
|
|
"""
|
|
set_logger_config(logger_config_name)
|
|
_ts = init_task_service()
|
|
_ts.stop()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 1:
|
|
if sys.argv[1] == "start":
|
|
start_service()
|
|
elif sys.argv[1] == "stop":
|
|
stop()
|
|
else:
|
|
print("用法: python service/task_service.py start")
|
|
else:
|
|
start_service()
|