from ctypes import util import os, oss2, time, redis, requests, shutil, sys, subprocess, json, platform,random from PIL import Image, ImageDraw, ImageFont from retrying import retry import atexit,platform import get_preview_image # if platform.system() == 'Windows': # sys.path.append('libs\\') sys.path.append('/home/acprint/code/libs/') import common def create_redis_connection(): """创建 Redis 连接，若连接失败则重试""" while True: try: r = redis.Redis(host="mark.api.suwa3d.com", password="kcV2000", port=6379, db=6) # 尝试进行一次操作，检查连接是否有效 r.ping() # ping 操作是一个简单的连接测试 print("Redis连接成功！") return r except ConnectionError: print("Redis连接失败，正在重试...") time.sleep(5) def main(r): while True: try: #构建打印文件 make3d4print_task(r) except Exception as e: print(f'出现异常:{e}') time.sleep(15) r = create_redis_connection() continue #根据pid获取orderId def getPidFromOrderId(orderId): getPidFromOrderId_url = 'https://mp.gray.api.suwa3d.com/api/printOrder/infoByOrderId' print(f'{getPidFromOrderId_url}?id={orderId}') res = requests.get(f'{getPidFromOrderId_url}?id={orderId}') resCode = res.json()['code'] if int(resCode) != 1000: return -1 print(res.text) return res.json()['data'] def down_obj_fromoss(pid, print_type=1, order_id=None,download_flag="print_build"): # print_type：// 打印状态 1:正常打印 2:重打 3:加打，4: 样品 print('开始下载obj文件...' , pid) if not order_id is None: path = os.path.join(workdir, 'print', f'{pid}_{order_id}') if not os.path.exists(path): os.makedirs(path) # 根据前缀获取文件列表 prefix = f'objs/print/{pid}/' #根据order_id 获取文件在 oss 上的存储路径 if order_id is not None: tempURL = "https://mp.gray.api.suwa3d.com/api/order/getOssSuffixByOrderId?order_id="+str(order_id) res = requests.get(tempURL) resCode = res.json()['code'] if int(resCode) != 1000: return -1 print(res.text) if res.json()['data'] != "": prefix = f'objs/print/{pid}/{res.json()["data"]}/{order_id}' print(f"prefix ======= {prefix}") filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix) find = False findTex1Jpg = False for file in filelist: filename = file.key.split('/')[-1] if filename == '': continue if filename.endswith(f'{pid}.obj'): find = True if download_flag == "print_build": if filename != str(pid)+".obj" and filename != str(pid)+".mtl" and filename != str(pid)+"Tex1.jpg": continue localfile = os.path.join(path, filename) res = oss_client.get_object_to_file(file.key, localfile) print(f'下载文件：{file.key}，状态：{res.status}') if not find: filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix) for file in filelist: filename = file.key.split('/')[-1] if filename == '': continue if filename.endswith(f'.obj'): find = True # if download_flag == "print_build": # if filename != str(pid)+".obj" and filename != str(pid)+".mtl" and filename != str(pid)+"Tex1.jpg": # continue localfile = os.path.join(path, filename) res = oss_client.get_object_to_file(file.key, localfile) print(f'下载文件：{file.key}，状态：{res.status}') break if find: os.rename(localfile, os.path.join(path,f"{pid}.obj")) #重命名文件 #判断是否有Tex1.jpg if not os.path.exists(os.path.join(path,f"{pid}Tex1.jpg")): filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix) for file in filelist: filename = file.key.split('/')[-1] if filename == '': continue if filename.endswith(f'{pid}.jpg'): findTex1Jpg = True localfile = os.path.join(path, filename) res = oss_client.get_object_to_file(file.key, localfile) print(f'下载文件：{file.key}，状态：{res.status}') break if findTex1Jpg: os.rename(localfile, os.path.join(path,f"{pid}Tex1.jpg")) #重命名文件 def delete_files_in_directory(directory): for file_name in os.listdir(directory): file_path = os.path.join(directory, file_name) try: if os.path.isfile(file_path): os.remove(file_path) print(f"Deleted: {file_path}") except Exception as e: print(f"Error deleting {file_path}: {e}") def make3d4print_task(r): try: if r.llen('model:noresize') == 0: print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), '暂无可处理的数据') time.sleep(5) except Exception as e: time.sleep(5) print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), 'redis连接异常，重新连接') r = redis.Redis(host='106.14.158.208', password='kcV2000', port=6379, db=6) return orderId = r.lpop('model:noresize') if orderId is None: return # arrOrder = orderIdStr.decode('utf-8').split('_') # orderId = arrOrder[0] # size = arrOrder[1] if type(orderId) != str: orderId = orderId.decode('utf-8') # orderId = 56077 res = getPidFromOrderId(orderId) if res == -1: print("查询打印订单信息失败，重新开启进程") #os.system(f'python auto_convert3d.py') return pid = str(res['pid']) if pid == "88985": return if pid == "": return down_obj_fromoss(pid, 1, orderId,download_flag="print_build") path = os.path.join(workdir, 'print', f'{pid}_{orderId}') #获取打印尺寸信息 get_printsize_url = 'https://mp.gray.api.suwa3d.com/api/printOrder/infoByOrderId' res = requests.get(f'{get_printsize_url}?id={orderId}') print('获取打印尺寸:', res.text) objName = "" for f in res.json()['data']['fileList']: if "undefined" in f: continue try: objName = f break except Exception as e: print("eeee",e) return print("obj尺寸名称",objName) # 如果指定文件夹目标文件存在，则先删除 if os.path.exists(os.path.join(workdir, f'complate/objs/{pid}/order_{orderId}/')): delete_files_in_directory(os.path.join(workdir, f'complate/objs/{pid}/order_{orderId}/')) for file in os.listdir(path): # 跳过一些不需要上传的文件 # if file in [f'{pid}.png',f'{pid}_old.jpg', f'{pid}.obj', f'{pid}_decimate.glb', f'{pid}_decimate.obj', f'{pid}_decimate.mtl', f'{pid}Tex1_decimate.jpg', f'{pid}_original.obj', f'{pid}_original.mtl']: continue # print("当前目录",os.path.join(path, file)) #将文件移动到指定目录 if not os.path.exists(os.path.join(workdir, f'complate/objs/{pid}/')): os.makedirs(os.path.join(workdir, f'complate/objs/{pid}/'),mode=0o777, exist_ok=True) if not os.path.exists(os.path.join(workdir, f'complate/objs/{pid}/order_{orderId}/')): os.makedirs(os.path.join(workdir, f'complate/objs/{pid}/order_{orderId}/'),mode=0o777, exist_ok=True) shutil.move(os.path.join(path, file), os.path.join(workdir, f'complate/objs/{pid}/order_{orderId}/')) #判断是否是 Obj， 如果是的话要进行重命名 if file.endswith(f'{pid}.obj'): os.rename(os.path.join(workdir, f'complate/objs/{pid}/order_{orderId}/{file}'), os.path.join(workdir, f'complate/objs/{pid}/order_{orderId}/{objName}')) print("重命名成功") res = requests.get(f'{update_makeprintobj_status_url}?id={orderId}') print('更新打印状态:', res.text) print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), f'orderId：{orderId} pid：{pid} 上传生成的模型 end') #生成封面图片 print('小票封面图处理中....') get_preview_image.createImage(pid) #os.system(f'python get_preview_image.py {pid}') print(f"{pid}-已处理结束") if __name__ == '__main__': # atexit.register(common.notify,"虚拟机，生成打印任务程序停止") AccessKeyId = 'LTAI5tSReWm8hz7dSYxxth8f' AccessKeySecret = '8ywTDF9upPAtvgXtLKALY2iMYHIxdS' Endpoint = 'oss-cn-shanghai.aliyuncs.com' Bucket = 'suwa3d-securedata' oss_client = oss2.Bucket(oss2.Auth(AccessKeyId, AccessKeySecret), Endpoint, Bucket) update_check_url = 'https://mp.gray.api.suwa3d.com/api/customerP3dLog/updateStatusToWaitingPlatformCheckingStatus' update_team_check_url = 'https://mp.gray.api.suwa3d.com/api/customerP3dLog/updateStatusToWaitingTeamCheckingStatus' update_status_printstatus_url = 'https://mp.gray.api.suwa3d.com/api/customerP3dLog/updateBuildPrintModelStatus' update_makeprintobj_status_url = 'https://mp.gray.api.suwa3d.com/api/printOrder/updateMakePrintObjSucceed' getRepairInfo_url = 'https://repair.api.suwa3d.com/api/modelRepairOrder/teamCheckGLBInfo' update_repair_status_url = 'https://repair.api.suwa3d.com/api/modelRepairOrder/updateStatusToWaitingTeamCheckingStatus' if platform.system() == 'Windows': workdir = 'E:\\' else: workdir = '/data/datasets/' r = create_redis_connection() #E:\\complate/objs/147852_54579/ # os.remove(os.path.join(workdir, f'complate/objs/147852_54579/')) main(r)