From 551080dffd072c439e535298f70deaf8fa733b64 Mon Sep 17 00:00:00 2001
From: dongchangxi <458593490@qq.com>
Date: Tue, 13 May 2025 17:45:11 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AE=A1=E7=AE=97=E5=85=8B=E9=87=8D=E6=96=B0?=
 =?UTF-8?q?=E6=96=B9=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 timer/external_order_glbV2.py | 271 ++++++++++++++++++++++++++++++++++
 timer/test.py                 |  34 +++++
 2 files changed, 305 insertions(+)
 create mode 100644 timer/external_order_glbV2.py
 create mode 100644 timer/test.py

diff --git a/timer/external_order_glbV2.py b/timer/external_order_glbV2.py
new file mode 100644
index 0000000..5ea4b77
--- /dev/null
+++ b/timer/external_order_glbV2.py
@@ -0,0 +1,271 @@
+import os, sys, time, bpy, math, requests, bmesh, json, shutil,oss2
+from PIL import Image
+import platform,redis
+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,common,foot_mark_seam,libs_db_gpu
+
+def bmesh_copy_from_object(obj, transform=True, triangulate=True, apply_modifiers=False):
+    """Returns a transformed, triangulated copy of the mesh"""
+    assert obj.type == 'MESH'
+    if apply_modifiers and obj.modifiers:
+        import bpy
+        depsgraph = bpy.context.evaluated_depsgraph_get()
+        obj_eval = obj.evaluated_get(depsgraph)
+        me = obj_eval.to_mesh()
+        bm = bmesh.new()
+        bm.from_mesh(me)
+        obj_eval.to_mesh_clear()
+    else:
+        me = obj.data
+        if obj.mode == 'EDIT':
+            bm_orig = bmesh.from_edit_mesh(me)
+            bm = bm_orig.copy()
+        else:
+            bm = bmesh.new()
+            bm.from_mesh(me)
+    if transform:
+        matrix = obj.matrix_world.copy()
+        if not matrix.is_identity:
+            bm.transform(matrix)
+            matrix.translation.zero()
+            if not matrix.is_identity:
+                bm.normal_update()
+    if triangulate:
+        bmesh.ops.triangulate(bm, faces=bm.faces)
+    return bm
+
+def find_pid_objname(pid):
+    for obj in bpy.data.objects:
+        if obj.name.startswith(str(pid)):
+            return obj.name
+
+
+def down_obj_from_oss(workdir, pid, action):
+    if os.path.exists(os.path.join(workdir, action, pid)):
+        print(f'目录{os.path.join(workdir, action, pid)}已存在，跳过')
+        return
+    else:
+        os.makedirs(os.path.join(workdir, action, pid))
+
+    # 根据前缀获取文件列表
+    prefix = f'objs/print/{pid}/base/model/'
+    filelist = oss2.ObjectIteratorV2(config.oss_bucket, prefix=prefix)
+    print('正在下载:', prefix)
+    obj_filename = ""
+    for file in filelist:
+        
+
+        filename = file.key.split('/')[-1]
+        if filename.endswith(".exr"):
+            continue
+
+        if filename.endswith("_1.jpg") or filename.endswith("_8.jpg"):
+            continue
+
+        if filename.endswith('.obj'): 
+            obj_filename = filename
+        # print('正在下载:', file.key)
+        localfile = os.path.join(workdir, action, pid, filename)
+        config.oss_bucket.get_object_to_file(file.key, localfile)
+
+    return obj_filename
+
+#加载obj文件
+def reload_obj(pid):
+    
+    #下载obj文件
+    down_obj_from_oss(config.workdir, pid, "print")
+    obj_filename = os.path.join(config.workdir,'print',pid,f'{pid}.obj')
+    bpy.ops.wm.read_homefile()
+    bpy.ops.object.delete(use_global=False, confirm=False)
+    bpy.ops.import_scene.obj(filepath=obj_filename)
+    bpy.context.scene.unit_settings.scale_length = 1
+    bpy.context.scene.unit_settings.length_unit = 'CENTIMETERS'
+    bpy.context.scene.unit_settings.mass_unit = 'GRAMS'
+
+    obj = bpy.context.selected_objects[0]
+    bpy.context.view_layer.objects.active = obj
+    obj.select_set(True)
+
+def get_p3d_info(pid):
+    url = "https://mp.api.suwa3d.com/api/customerP3dLog/info?id="+pid
+    res = requests.get(url)
+    res = res.json()
+    if res["code"] == 1000:
+        
+        #https://www.suwa3d.com/external_order_cover/2025/01/09/1x4pzvxolk6d6xflb8p8r779e0oo2myn.png
+        imagePath = res["data"]["texture_cover_img"]
+        #字符串替换
+        imagePath = imagePath.replace("https://www.suwa3d.com/","")
+        imagePath = imagePath.replace(".jpg",".png")
+        
+        return imagePath
+    else:
+        return 0
+
+
+def base_fix(pid):
+    # 统一blender环境
+    reload_obj(pid)
+    # 统一模型方向、位置、大小...
+    pid_objname = find_pid_objname(pid)
+    obj = bpy.data.objects[pid_objname]
+    bpy.data.objects[pid_objname].rotation_euler = (0, 0, 0)
+    # 检查当前模型是否正确使用 z 轴作为高度轴
+    if obj.dimensions.z < obj.dimensions.y:
+        # 如果 z 轴比 y 轴小，说明高度错误，应该旋转调整模型
+        print(f"警告：模型的高度轴（z）似乎不正确，使用 y 轴作为高度。正在修正模型方向...")
+        bpy.ops.object.transform_apply(location=True, rotation=True, scale=True)  # 应用当前变换
+        obj.rotation_euler[0] = math.radians(90)  # 旋转模型 90 度，交换 z 和 y 轴
+        bpy.ops.object.transform_apply(rotation=True)  # 应用旋转
+    
+    
+
+    bpy.ops.object.origin_set(type='ORIGIN_CENTER_OF_VOLUME', center='MEDIAN')
+    bpy.context.object.location[0] = 0
+    bpy.context.object.location[1] = 0
+    bpy.ops.object.transform_apply(location=True, rotation=True, scale=True)
+    print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 模型基础校正完成')
+
+
+def export_and_update_glbs(pid):
+    print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 开始导出并上传审核模型和3D相册模型glb文件...')
+    start_time = time.time()
+    #headcount = libs.getHeadCount(pid)
+    headcount = 1
+    pid_objname = find_pid_objname(pid)
+    obj = bpy.data.objects[pid_objname]
+    obj.select_set(True)
+
+    model_info = {}
+    model_info['headcount'] = headcount
+    model_info['faces'] = round(len(obj.data.polygons) / 10000)
+    model_info['height'] = round(obj.dimensions.y * 100)
+
+    # bpy.ops.wm.save_as_mainfile(filepath=os.path.join(config.workdir, pid, f'{pid}.blend'))
+
+    # 统一缩放到9cm标准尺寸
+    scale = 90 / bpy.data.objects[pid_objname].dimensions.y
+    bpy.data.objects[pid_objname].scale = (scale, scale, scale)
+    bpy.ops.object.transform_apply(scale=True)
+
+    # bpy.ops.wm.save_as_mainfile(filepath=os.path.join(config.workdir, pid, f'{pid}-9cm.blend'))
+
+    bm = bmesh_copy_from_object(obj)
+    model_info['volume'] = round(bm.calc_volume(), 2)
+    model_info['weight'] = round(model_info['volume'] * 1.226, 2)
+    print(f'{pid}的模型数据：{model_info}')
+    
+    res = requests.get(f'{config.urls["upload_model_info_url"]}?pid={pid}&headcount={headcount}&faces={model_info["faces"]}&volume={model_info["volume"]}&weight={model_info["weight"]}&height={model_info["height"]}')
+    print('上传模型数据：', res.text)
+    # with open(os.path.join(config.sharedir, 'model_info', f'{pid}.json'), 'w') as f:
+    #     json.dump(model_info, f)
+    # f.close()
+
+    # 先生成审核模型
+    faces_dest = 500000 * headcount
+    # 减面
+    faces_current = len(bpy.data.objects[pid_objname].data.polygons)
+    print(f'当前面数：{faces_current}，目标面数：{faces_dest}')
+
+    bpy.ops.object.modifier_add(type='DECIMATE')
+    bpy.context.object.modifiers["Decimate"].ratio = faces_dest / faces_current
+    bpy.ops.object.modifier_apply(modifier="Decimate")
+
+    glb_filename = os.path.join(config.workdir,  'print',pid, f'{pid}.glb')
+    bpy.ops.export_scene.gltf(filepath=glb_filename, export_format='GLB', export_apply=True, export_jpeg_quality=75, export_draco_mesh_compression_enable=False)
+
+    # glb_filename = os.path.join(config.workdir, pid, 'output', f'{pid}.glb')
+    # bpy.ops.export_scene.gltf(filepath=glb_filename, export_format='GLB', export_apply=True, export_jpeg_quality=75, export_draco_mesh_compression_enable=False)
+
+    # 再生成数字模型
+    faces_dest = 120000 * headcount
+
+    # 减面
+    faces_current = len(bpy.data.objects[pid_objname].data.polygons)
+    print(f'当前面数：{faces_current}，目标面数：{faces_dest}')
+
+    bpy.ops.object.modifier_add(type='DECIMATE')
+    bpy.context.object.modifiers["Decimate"].ratio = faces_dest / faces_current
+    bpy.ops.object.modifier_apply(modifier="Decimate")
+
+    glb_filename = os.path.join(config.workdir,'print',pid,f'{pid}-3d.glb')
+    bpy.ops.export_scene.gltf(filepath=glb_filename, export_format='GLB', export_apply=True, export_jpeg_quality=75, export_draco_mesh_compression_enable=False)
+
+    os.system(f'gltfpack -c -i {os.path.join(config.workdir,  "print",pid, f"{pid}.glb")} -o {os.path.join(config.workdir, "print",pid,  f"{pid}-pack.glb")}')
+    config.oss_bucket.put_object_from_file(f'glbs/auto/{pid}.glb', os.path.join(config.workdir,  'print',pid, f'{pid}-pack.glb'))
+    config.oss_bucket.put_object_from_file(f'glbs/print/{pid}.glb', os.path.join(config.workdir,  'print',pid, f'{pid}-pack.glb'))
+    config.oss_bucket.put_object_from_file(f'glbs/3d/{pid}.glb', os.path.join(config.workdir,'print',pid, f'{pid}-3d.glb'))
+
+    print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} glb文件导出并上传完成，共费时{libs.diff_time(start_time)}')
+    
+    #移除文件夹
+    #shutil.rmtree(os.path.join(config.workdir,'print',pid))
+    
+    #更新该笔订单的状态为 3000
+    requests.post("https://shop.api.suwa3d.com/api/printOrder/updateExternalOrderStatus", data={'pid': pid})
+
+def createGlb(pid):
+
+    
+    base_fix(pid)
+    export_and_update_glbs(pid)
+    #移除文件夹
+    # try:
+    #     shutil.rmtree(os.path.join(config.workdir,'print',pid))
+    # except:
+    #     pass
+
+
+def create_redis_connection():
+    """创建 Redis 连接，若连接失败则重试"""
+    while True:
+        try:
+            r = redis.Redis(host="106.14.158.208",password="kcV2000",port=6379,db=6)
+            # 尝试进行一次操作，检查连接是否有效
+            r.ping()  # ping 操作是一个简单的连接测试
+            print("Redis连接成功！")
+            return r
+        except ConnectionError:
+            print("Redis连接失败，正在重试...")
+            time.sleep(5) 
+
+if __name__ == '__main__':
+    arrArgs = sys.argv
+    if len(arrArgs) == 2:
+        pid = arrArgs[1]
+        #判断是否是数字
+        if pid.isdigit():
+            print('请输入正确的pid')
+            createGlb(str(pid))
+    
+    else:
+        r = create_redis_connection()
+        while True:
+            try:
+                if r.llen('model:3dglb') == 0:
+                    print('队列为空，等待10秒')
+                    time.sleep(10)
+                    continue
+                    
+                info = r.lpop('model:3dglb')
+                if info is None:
+                    print('队列为空，等待10秒')
+                    time.sleep(10)
+                pid = info.decode('utf-8')
+                
+                if not pid.isdigit():
+                    continue
+                
+                createGlb(pid)
+            except Exception as e:
+                print(f'错误：{e}')
+                time.sleep(10)
+                r = create_redis_connection()
+                continue
+
+
diff --git a/timer/test.py b/timer/test.py
new file mode 100644
index 0000000..6c0ea27
--- /dev/null
+++ b/timer/test.py
@@ -0,0 +1,34 @@
+
+import redis,requests,json,os
+r = redis.Redis(host="172.31.1.253",password="",port=6379,db=6)
+
+# info 是json字符串   {"order_id":[858344,858345],"pid":271916}
+info = r.lpop('model:3dglb')
+info = json.loads('{"order_id":[858344,858345],"pid":271916}')
+print(info)
+#获取 order_id 列表
+order_id_list = info['order_id']
+#获取 pid
+pid = info['pid']
+
+# print(pid)
+# 获取订单信息
+for order_id in order_id_list:
+    print(order_id)
+    #url 请求
+    url = f"https://mp.api.suwa3d.com/api/printOrder/infoByOrderId?id={order_id}"
+    response = requests.get(url)
+    response = response.json()
+    if response['code'] != 1000:
+        continue
+
+    #获取打印尺寸
+    printdata = response['data']['data']
+    for key,value in printdata.items():
+        #去掉cm
+        key = key.replace('cm','')
+        print(key)
+        print(f'执行脚本--- "python D:/make2/tools/cal_weight.py print {pid} {key} {order_id}"')
+        #发起计算请求
+        os.system(f"python D:/make2/tools/cal_weight.py {print} {pid} {key} {order_id}")
+