程序优化

main_service.py

@@ -1,6 +1,6 @@
 import sys,socket,time,platform
 from logic import logic_main_service
-import logging
+import logging,atexit
 import main_step1,main_step2,main_step3
 if platform.system() == 'Windows':
     #线上正式运行
@@ -8,8 +8,9 @@ if platform.system() == 'Windows':
     #本地测试
 else:
     sys.path.append('/data/deploy/make3d/make2/libs/')
-import main_service_db,config
+import main_service_db,config,common
 if __name__ == '__main__':
+    atexit.register(common.notify,socket.gethostname()+"分布式建模任务已经停止")
     if len(sys.argv) == 2:
         print(sys.argv[1])
         os.system(f'python main_step1.py {sys.argv[1]}')