diff --git a/timer/external_order_glbV3.py b/timer/external_order_glbV3.py new file mode 100644 index 0000000..9f8d83b --- /dev/null +++ b/timer/external_order_glbV3.py @@ -0,0 +1,402 @@ +import os, sys, time, bpy, math, requests, bmesh, json, shutil,oss2,chardet +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): + print(bpy.data.objects) + for obj in bpy.data.objects: + print(obj) + 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)}已存在,跳过') + #删除目录 + shutil.rmtree(os.path.join(workdir, action, pid)) + + os.makedirs(os.path.join(workdir, action, pid)) + + jpgsTemp = f'objs/print/{pid}/base/model/{pid}Tex1.jpg' + mtlTemp = f'objs/print/{pid}/base/model/{pid}.mtl' + objTemp = f'objs/print/{pid}/base/model/{pid}.obj' + + # 根据前缀获取文件列表 + 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 file.key != jpgsTemp and file.key != mtlTemp and file.key != objTemp: + continue + + if filename.endswith('.obj'): + obj_filename = filename + print('正在下载:', file.key,filename) + localfile = os.path.join(workdir, action, pid, filename) + config.oss_bucket.get_object_to_file(file.key, localfile) + + return obj_filename + +def checkDownloadFile(pid): + mtlfilePath = os.path.join(config.workdir, "print", pid, f"{pid}.mtl") + # objfilePath = os.path.join(config.workdir, "print", pid, f"{pid}.obj") + + # 读取文件内容(自动检测编码) + def read_file_auto_encoding(path): + with open(path, 'rb') as f: + raw_data = f.read() + encoding = chardet.detect(raw_data)['encoding'] or 'utf-8' + return raw_data.decode(encoding, errors='ignore') + + mtlContent = read_file_auto_encoding(mtlfilePath) + + if mtlContent == "": + print(f'{pid}的mtl文件检测出现问题了, 内容为空') + return False + + newmtlCount = mtlContent.count('newmtl') + if newmtlCount > 1: + print(f'{pid}的mtl文件检测出现问题了, 包含多个newmtl') + return False + + if f'{pid}Tex1.jpg' not in mtlContent: + print(f'{pid}的mtl文件检测出现问题了, 没有找到{pid}Tex1.jpg') + return False + + # # 检测obj文件 + # isContain = False + + # objContent = read_file_auto_encoding(objfilePath) + + + # for line in objContent.splitlines(): + # if line.startswith(f'o {pid}'): + # isContain = True + # print(f'{pid}的obj文件检测没问题') + # break + + # if not isContain: + # print(f'{pid}的obj文件检测出现问题了, 没有找到o {pid} 的行') + # return False + + return True + +#加载obj文件 +def reload_obj(pid,order_ids): + for order_id in order_ids: + 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','') + key = float(key)*10 + 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}") + #计算克重信息 + + + 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: + return res["data"]["guid"] + else: + return 0 + + +def base_fix(pid,order_ids): + # 统一blender环境 + print("BBBBBBBBBBBBB") + reload_obj(pid,order_ids) + # 统一模型方向、位置、大小... + pid_objname = find_pid_objname(pid) + print("AAAAAAAAAAAAA",pid_objname) + obj = bpy.context.selected_objects[0] #bpy.data.objects[pid_objname] + obj.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,order_ids): + 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.context.selected_objects[0] + 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 / obj.dimensions.y + obj.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(obj.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(obj.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}-3d.glb')) + config.oss_bucket.put_object_from_file(f'glbs/print/{pid}.glb', os.path.join(config.workdir, 'print',pid, f'{pid}-3d.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 + # order_ids 逗号隔开 + print(order_ids) + order_ids = ",".join(map(str,order_ids)) + strRequest = f"https://shop.api.suwa3d.com/api/printOrder/updateExternalOrderStatusV2?pid={pid}&order_ids={order_ids}" + print(strRequest) + res = requests.post("https://shop.api.suwa3d.com/api/printOrder/updateExternalOrderStatusV2", data={'pid': pid,"order_ids":order_ids}) + print(res.text) + # # 记录日志文件 + # logPath = os.path.join(config.workdir,"log",'external_order_glbV2',pid,f'{order_ids}.log') + # os.makedirs(logPath, exist_ok=True) + # # 创建并且写入日志 + # with open(logPath, 'w+') as f: + # f.write(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 订单状态更新完成,共费时{libs.diff_time(start_time)}\n') + # f.write(f'{res.text}\n') + # f.write(f'{strRequest}\n') + + + + #执行获取obj缩略图 + print("执行获取obj全身缩略图脚本") + os.system(f'python d:\\make2\\tools\pic_for_obj\image_rander_small.py -pid {pid} -i D://print/{pid} -o D://print/{pid}') + print("判断获取封面图是否存在0") + #判断文件是否存在,存在则上传到oss, 更新数据库内容 + if os.path.exists(f'D://print/{pid}/{pid}_pic.png'): + #获取拍照订单的信息,从中得到拍照订单的信息 + uuid = get_p3d_info(str(pid)) + print("判断获取封面图是否存在1") + if uuid != 0 and uuid != None and uuid != "": + print("uuid",uuid) + config.oss_bucket_3d_view.put_object_from_file(f'{uuid}/3d_view.png', f'D://print/{pid}/{pid}_pic.png') + + # 读取队列的信息,获取内容快照, 按时间戳命名 文件 + r = create_redis_connection() + info = r.lrange('model:printOrder', 0, -1) + print(info) + # 将info写入到文件中 + with open(os.path.join(config.workdir,'log','external_order_glbV2',f'{pid}_{time.time()}.log'), 'w') as f: + f.write(str(info)) + +def createGlb(pid,order_ids): + + #下载obj文件 + down_obj_from_oss(config.workdir, pid, "print") + print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} 下载obj文件完成...') + #检测mtl 文件是否正常, + isNormal = checkDownloadFile(pid) + if isNormal == False: + print(f'{pid}的mtl文件检测出现问题,不执行该笔任务,调用取消的接口') + for order_id in order_ids: + url = f"https://mp.api.suwa3d.com/api/order/cancelExternalOrder?order_id={order_id}&typename=system" + response = requests.get(url) + response = response.json() + print(response) + if response['code'] != 1000: + continue + print(f'取消订单{order_id}成功') + return + + base_fix(pid,order_ids) + export_and_update_glbs(pid,order_ids) + #移除文件夹 + # 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__': + # pid = "286049" + # order_ids = [863038] + # createGlb(str(pid),order_ids) + 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:3dglbV2') == 0: + print('队列为空,等待10秒') + time.sleep(10) + continue + + info = r.lpop('model:3dglbV2') + if info is None: + print('队列为空,等待10秒') + time.sleep(10) + + + info = json.loads(info) + print(info) + pid = info['pid'] + order_ids = info['order_id'] + + # if not pid.isdigit(): + # continue + + createGlb(str(pid),order_ids) + except Exception as e: + print(f'错误:{e}') + time.sleep(10) + r = create_redis_connection() + continue + +