程序优化

main_step1.py

@@ -1,12 +1,12 @@
 import os, sys, time, shlex, subprocess, shutil, requests, cv2, numpy as np
 from PIL import Image
-import platform
+import platform,socket,atexit
 if platform.system() == 'Windows':
     sys.path.append('e:\\libs\\')
     #sys.path.append('libs')
 else:
     sys.path.append('/data/deploy/make3d/make2/libs/')
-import config, libs, libs_db,main_service_db
+import config, libs, libs_db,main_service_db,common
 
 def filter_dark_texture_image(pid):
     start_time = time.time()
@@ -204,6 +204,7 @@ def main(pid, experience=False, makeloop=True,task_distributed_id="",isNoColorTe
 if __name__ == '__main__':    
     # 取云端redis任务，完成第一步的数据预处理后，将数据放入共享存储目录，将第二步任务塞入本地mysql队列
     # 默认循环值守，可传参数运行单一任务，以方便调试
+    atexit.register(common.notify,socket.gethostname()+"建模任务已经停止")
     pid = '0'
     isNoColorTexture = ""
     if len(sys.argv) == 2: