新影棚的对齐程序

old_to_new_build/get_reg_info.py

@@ -41,7 +41,7 @@ def get_xmps(psid,pid):
     print('下载xmps完成')
 
 #获取photo2 的对齐信息
-def get_photo2_align_info(pid):
+def get_photo2_align_info(pid,flagDoXmps=False):
     pid = str(pid)
     psid = libs.getPSid(pid)
     
@@ -59,7 +59,7 @@ def get_photo2_align_info(pid):
     print(f"影棚号:{psid} , {config.oss_bucket.object_exists(f'xmps_rate/{psid}/')}")
     filename = f'xmps_rate/{psid}/{psid}.rcbox'    
     isRate = False
-    if not config.oss_bucket.object_exists(filename):
+    if not config.oss_bucket.object_exists(filename) or flagDoXmps == True:
         #提示是否进行 xmps_rate 文件夹的创建
         yes_or_not = input('当前该影棚没有对应的对齐坐标,是否创建？(y/n)')
         if yes_or_not == 'y':
@@ -127,10 +127,13 @@ def create_redis_connection():
 
 if __name__ == '__main__':
     start = time.time()
-    if len(sys.argv) == 2:
+    if len(sys.argv) >= 2:
         pids = sys.argv[1]
+        flagDoXmps = False
+        if len(sys.argv) == 3:
+            flagDoXmps = True
         for pid in pids.split(','):
-            get_photo2_align_info(pid)
+            get_photo2_align_info(pid,flagDoXmps)
             print(f'{pid}共费时{time.time() - start}秒')
     else:
         r = create_redis_connection()  # 使用重连函数来创建连接