import os,sys,time,shlex,subprocess,shutil,requests,json
import platform
if platform.system() == 'Windows':
    sys.path.append('e:\\libs\\')
    #sys.path.append('libs')
else:
    sys.path.append('/data/deploy/make3d/make2/libs/')
import libs,config
def main(pid = 0):
    print(f'开始时间-{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}')
    if pid == 0:
        #获取要处理PID的url
        url = "https://repair.api.suwa3d.com/api/modelRepairOrder/getRepairOrderListByPriority"
        #发起请求
        res = requests.get(url)
        print('res:', res.text)
        if res.status_code != 200:
            print('获取失败，程序退出')
            return
        res = json.loads(res.text)
        pid = 0
        try:
            pid = res['data']['pid']
        except Exception as e:
            return 0

    if pid == 0:
        return 0
    
    #调用oss下载功能，下载对应的文件
    pathStr = os.path.join(config.workdir,str(pid)+"_ai")
    libs.down_obj_from_oss_auto(str(pid),"auto",pathStr)
    #检查是否存在产生的对应的文件
    if not os.path.exists(os.path.join(pathStr,f'{pid}.obj')):
        print(f"找不到{pid}obj文件")
        return 
    
    if not os.path.exists(os.path.join(pathStr,f'{pid}.mtl')):
        print(f"找不到{pid}mtl文件")
        return 
    #下载完成后调用
    os.system(f"python optimize_model.py -ip {pathStr}/{pid}.obj -op {pathStr}/{pid}_ai.obj")
    #检查是否存在产生的对应的文件
    if not os.path.exists(os.path.join(pathStr,f'{pid}_ai.obj')):
        print(f"找不到{pid}ai_obj文件")
        return 
    
    if not os.path.exists(os.path.join(pathStr,f'{pid}_ai.mtl')):
        print(f"找不到{pid}ai_mtl文件")
        return 
    
    #处理完成完后上传到oss指定目录
    config.oss_bucket.put_object_from_file(f'objs/auto/{pid}/ai/fix_mesh/{pid}_ai.obj', os.path.join(pathStr,f'{pid}_ai.obj'))
    config.oss_bucket.put_object_from_file(f'objs/auto/{pid}/ai/fix_mesh/{pid}_ai.mtl', os.path.join(pathStr,f'{pid}_ai.mtl'))
    config.oss_bucket.put_object_from_file(f'objs/auto/{pid}/ai/fix_mesh/{pid}.jpg', os.path.join(pathStr,f'{pid}.jpg'))
    #更新处理结果
    url = "https://repair.api.suwa3d.com/api/modelRepairOrder/updateShootOrderBuildModelType?pid="+str(pid)
    #发起请求
    requests.get(url)
    #移除文件夹
    shutil.rmtree(pathStr)
    print(f'结束时间-{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}')



if __name__ == '__main__':

    if len(sys.argv) == 2:
        pid = sys.argv[1]
        main(pid)
        print(f"{pid}-处理结束")
    else:
        while True:
            res = main()
            if res == 0:
                print("休眠60s")
                time.sleep(60)