gameui/drawing
2
0
Fork 0
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

优雅退出2

Browse Source
This commit is contained in:
桜華
2023-03-01 02:04:56 +08:00
parent 164c65c8f9
commit 90f5f92abb
1 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
server.py
View File

@@ -207,6 +207,13 @@ def parse_args():
return opt return opt
import sys
import signal
def quit(signum, frame):
print('stop fusion')
sys.exit()
def put_watermark(img, wm_encoder=None): def put_watermark(img, wm_encoder=None):
if wm_encoder is not None: if wm_encoder is not None:
img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR) img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
@@ -214,7 +221,6 @@ def put_watermark(img, wm_encoder=None):
img = Image.fromarray(img[:, :, ::-1]) img = Image.fromarray(img[:, :, ::-1])
return img return img
# 对任务状态的修改 # 对任务状态的修改
def update_task_status(task: dict, status: str, progress: int, data: list = []): def update_task_status(task: dict, status: str, progress: int, data: list = []):
task["status"] = status task["status"] = status
@@ -224,6 +230,8 @@ def update_task_status(task: dict, status: str, progress: int, data: list = []):
# 从局域网中获取一组任务(如果列表为空，等待2s后重新获取) # 从局域网中获取一组任务(如果列表为空，等待2s后重新获取)
def get_tasks(tasks:list=[]): def get_tasks(tasks:list=[]):
signal.signal(signal.SIGINT, quit)
signal.signal(signal.SIGTERM, quit)
while len(tasks) == 0: while len(tasks) == 0:
try: try:
tasks = requests.get("http://localhost:3000/api/drawing?status=waiting").json() tasks = requests.get("http://localhost:3000/api/drawing?status=waiting").json()
@@ -234,12 +242,6 @@ def get_tasks(tasks:list=[]):
time.sleep(2) time.sleep(2)
return tasks return tasks
import sys
import signal
def quit(signum, frame):
print('stop fusion')
sys.exit()
def main_dev(opt): def main_dev(opt):
signal.signal(signal.SIGINT, quit) signal.signal(signal.SIGINT, quit)