make2/apps/auto_convert3d_new.py

from ctypes import util
import os, oss2, time, redis, requests, shutil, sys, subprocess, json, platform
from PIL import Image, ImageDraw, ImageFont
from retrying import retry
import atexit,platform
if platform.system() == 'Windows':
    sys.path.append('e:\\libs\\')
import common
import config
def find_blender_bin_path():
    if platform.system() == 'Linux': return 'blender'
    base_path = 'C:\\Program Files\\Blender Foundation\\'
    if os.path.exists(base_path):
        for dir in os.listdir(base_path):
            if dir.startswith('Blender'):
                blender_bin_path = base_path + dir + '\\blender.exe'
                return f'"{blender_bin_path}"'
    else:
        print('未找到blender安装目录')
        exit(1)

@retry(stop_max_attempt_number=10, wait_fixed=3000)
def down_obj_fromoss(pid,order_id):
    # print_type：// 打印状态 1:正常打印 2:重打 3:加打，4: 样品
    print('开始下载obj文件...' , pid)
    #请求接口 获取 对应的 oss 地址
    tempUrl = f'https://mp.gray.api.suwa3d.com/api/order/getOssSuffixByOrderId?order_id={order_id}'
    #发起请求
    res = requests.get(tempUrl)
    ossPath = ""
    if res.json()['code'] == 1000:
        ossPath = res.json()['data']
        ossPath = ossPath.replace("\\","/")
    

    path = os.path.join(workdir, 'print', pid)
    if not os.path.exists(path): os.makedirs(path)
    
    # 根据前缀获取文件列表
    prefix = f'objs/print/{pid}/'
    if ossPath != "":
        prefix = os.path.join("objs","print",pid,ossPath,order_id)
        prefix = prefix.replace("\\","/")
   
    print(f'prefix:{prefix}')
    print(f'实际有用的地址:objs/print/253312/model/standard_size/88942')
    filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix)
    
    if len(list(filelist)) == 0:
        filelist = oss2.ObjectIteratorV2(oss_client, prefix=f"objs/print/{pid}")
    else:
        filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix)
    

    find = False
    for file in filelist:
        filename = file.key.split('/')[-1]
        if filename == '': continue
        if filename.endswith(f'.obj') and find == True:
            continue
        
        if filename.endswith(f'{pid}.obj'):
            find = True
        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 and print_type == 3:
        for file in os.listdir(path):
            if file.endswith('.obj'):
                print('找到其他obj文件，采用这个文件来生成需要的尺寸', file)
                shutil.copy(os.path.join(path, file), os.path.join(path, f'{pid}.obj'))
                find = True
                break
    if not find:
        print('找不到obj文件，异常退出')
        sys.exit(1)

    return ossPath
    # print('下载完成后静默10秒，等待文件写入完成')
    # time.sleep(10)

def getPidFromOrderId(orderId):
    getPidFromOrderId_url = 'https://mp.gray.api.suwa3d.com/api/printOrder/info'
    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 team_check(r):
    try:
        if r.llen('model:IndependentRepairTeamcheckGLBQueue') == 0:
            time.sleep(5)
            return
    except Exception as e:
        print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), 'redis连接异常，5秒后重试')
        print(e)
        time.sleep(5)
        r = create_redis_connection()
        return
    repair_id = r.lpop('model:IndependentRepairTeamcheckGLBQueue')
    if repair_id is None: return
    repair_id = repair_id.decode('utf-8')

    res = requests.get(f'{getRepairInfo_url}?id={repair_id}')
    print(f'getRepairInfo_url:{getRepairInfo_url}?id={repair_id}')
    print(res.text)
    pid = res.json()['data']['pid']
    orderId = res.json()['data']['order_id']
    pid = str(pid)
    orderId = str(orderId)

    print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), f'pid：{pid} repair_id:{repair_id} 生成团队审核模型 start model:IndependentRepairTeamcheckGLBQueue', )
    ossPath = down_obj_fromoss(pid,orderId)
    if ossPath == "":
        print("ossPath is null")
        return
    obj_filename = f'{pid}.obj'
    glb_filename = f'{orderId}.glb'
    print('开始转换obj文件为glb文件...')
    os.system(f'gltfpack -c -i {os.path.join(workdir, "print", pid, obj_filename)} -o {os.path.join(workdir, "print", pid, glb_filename)} -vt 16 -vc 16')
    
    #处理封面图
    #执行获取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}')
    
    #判断文件是否存在，存在则上传到oss， 更新数据库内容
    if os.path.exists(f'D://print/{pid}/{pid}_pic.png'):
        #获取拍照订单的信息，从中得到拍照订单的信息
        uuid = get_p3d_info(str(pid))
        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')
    
    
    


    print('上传glb文件到oss...')
    oss_client.put_object_from_file(f'glbs/print/{ossPath}/{glb_filename}', os.path.join(workdir, "print", pid, glb_filename))

    print(f'{update_repair_status_url}?id={repair_id}')
    res = requests.get(f'{update_repair_status_url}?id={repair_id}')
    print('更新团队审核状态:', res.text)
    shutil.rmtree(os.path.join(workdir, 'print', pid), ignore_errors=True)
    print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), f'pid：{pid} 生成团队审核模型 end')

def get_p3d_info(pid):
    url = "https://mp.gray.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 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):
    print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), '模型生成程序 start')
    while True:  
        try:
            team_check(r)   
        except Exception as e:
            print('出现异常:', e)
            time.sleep(5)
            r = create_redis_connection()
            continue               

if __name__ == '__main__':
    atexit.register(common.notify,"R13-修模师审核通过，生成打印任务程序停止")
    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.gray.api.suwa3d.com/api/modelRepairOrder/teamCheckGLBInfo'
    update_repair_status_url = 'https://repair.gray.api.suwa3d.com/api/modelRepairOrder/updateStatusToWaitingTeamCheckingStatus'
    if platform.system() == 'Windows':
        workdir = 'D:\\'
    else:
        workdir = '/data/datasets/'
    blenderbin = find_blender_bin_path()
    r = create_redis_connection()
    main(r)