208 lines
9.1 KiB
Python
208 lines
9.1 KiB
Python
from ctypes import util
|
||
import os, oss2, time, redis, requests, shutil, sys, subprocess, json, qrcode
|
||
from PIL import Image, ImageDraw, ImageFont
|
||
|
||
|
||
def gen_qrcode(pid):
|
||
fontHeightMax = 40
|
||
fontsize = 1
|
||
qr = qrcode.QRCode()
|
||
qr.border = 2
|
||
qr.add_data(pid)
|
||
img = qr.make_image(fit=True)
|
||
img = img.transform((250, 294), Image.Transform.EXTENT, (0, 0, 250, 294), fillcolor='white')
|
||
|
||
cwd = os.path.dirname(os.path.abspath(__file__))
|
||
fontfile = os.path.join(cwd, 'fonts', 'Helvetica.ttf')
|
||
font = ImageFont.truetype(fontfile, fontsize)
|
||
while font.getsize(pid)[1] <= fontHeightMax and font.getsize(pid)[0] <= 240:
|
||
fontsize += 1
|
||
font = ImageFont.truetype(fontfile, fontsize)
|
||
fontsize -= 1
|
||
|
||
captionx = (250 - font.getsize(pid)[0]) / 2
|
||
draw = ImageDraw.Draw(img)
|
||
draw.text((captionx, 242), pid, font=font)
|
||
# img.show()
|
||
path = os.path.join(workdir, 'print', pid)
|
||
img.save(f'{path}/{pid}.png')
|
||
|
||
def down_obj_fromoss(pid, print_type=1):
|
||
# print_type：// 打印状态 1:正常打印 2:重打 3:加打，4: 样品
|
||
print('开始下载obj文件...' , pid)
|
||
if not os.path.exists(os.path.join(workdir, 'print')):
|
||
os.mkdir(os.path.join(workdir, 'print'))
|
||
path = os.path.join(workdir, 'print', pid)
|
||
if not os.path.exists(path): os.mkdir(path)
|
||
|
||
# 根据前缀获取文件列表
|
||
prefix = f'objs/print/{pid}/'
|
||
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'{pid}.obj'):
|
||
find = True
|
||
print('正在下载:', file.key)
|
||
localfile = os.path.join(path, filename)
|
||
oss_client.get_object_to_file(file.key, localfile)
|
||
|
||
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)
|
||
|
||
def getPidFromOrderId(orderId):
|
||
getPidFromOrderId_url = 'https://mp.api.suwa3d.com/api/printOrder/info'
|
||
print(f'{getPidFromOrderId_url}?id={orderId}')
|
||
res = requests.get(f'{getPidFromOrderId_url}?id={orderId}')
|
||
print(res.text)
|
||
return res.json()['data']
|
||
|
||
def detect_obj4print(pid):
|
||
for file in os.listdir(os.path.join(workdir, 'print', pid)):
|
||
if file.endswith('.obj') and 'x' in file:
|
||
return True
|
||
|
||
def make3d4print_task(r):
|
||
try:
|
||
if r.llen('model:printOrder') == 0:
|
||
time.sleep(5)
|
||
return
|
||
except Exception as e:
|
||
print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), 'redis连接异常，重新连接')
|
||
print(e)
|
||
time.sleep(5)
|
||
r = redis.Redis(host='106.14.158.208', password='kcV2000', port=6379, db=6)
|
||
return
|
||
orderId = r.lpop('model:printOrder')
|
||
if orderId is None: return
|
||
orderId = orderId.decode('utf-8')
|
||
res = getPidFromOrderId(orderId)
|
||
pid = str(res['pid'])
|
||
print_type = res['print_type']
|
||
digital_type = res['digital_type'] # 0: 只有手办 1: 只有数字模型 2: 手办+数字模型
|
||
|
||
print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), f'orderId：{orderId} pid：{pid} 生成待打印模型 start', )
|
||
down_obj_fromoss(pid, print_type)
|
||
gen_qrcode(pid)
|
||
os.system(f'{blenderbin} -b -P d:\\apps\\blender\\resize_model.py -- {orderId}')
|
||
if not detect_obj4print(pid):
|
||
print('obj文件生成异常，退出')
|
||
sys.exit(1)
|
||
print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), f'orderId：{orderId} pid：{pid} 生成待打印模型 end')
|
||
print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), f'orderId：{orderId} pid：{pid} 上传生成的模型 start', )
|
||
for file in os.listdir(os.path.join(workdir, 'print', pid)):
|
||
if file.endswith('.obj') and (not file.endswith(f'{pid}.obj')):
|
||
oss_client.put_object_from_file(f'objs/print/{pid}/{file}', os.path.join(workdir, 'print', pid, file))
|
||
if file.endswith('.glb'):
|
||
oss_client.put_object_from_file(f'glbs/3d/{pid}.glb', os.path.join(workdir, 'print', pid, file))
|
||
|
||
oss_client.delete_object(f'objs/print/{pid}/{pid}.obj')
|
||
shutil.rmtree(os.path.join(workdir, 'print', pid), ignore_errors=True)
|
||
|
||
print(f'{update_makeprintobj_status_url}?id={orderId}')
|
||
if digital_type == 1:
|
||
print('只有数字模型，不需要推送手办打印任务')
|
||
else:
|
||
res = requests.get(f'{update_makeprintobj_status_url}?id={orderId}')
|
||
print('更新打印状态:', res.text)
|
||
if res.json()['code'] != 1000:
|
||
print('更新打印状态失败')
|
||
sys.exit(1)
|
||
print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), f'orderId：{orderId} pid：{pid} 上传生成的模型 end')
|
||
|
||
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 = redis.Redis(host='106.14.158.208', password='kcV2000', port=6379, db=6)
|
||
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(res.text)
|
||
pid = res.json()['data']['pid']
|
||
pid = str(pid)
|
||
|
||
print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), f'pid：{pid} 生成团队审核模型 start', )
|
||
down_obj_fromoss(pid)
|
||
obj_filename = f'{pid}.obj'
|
||
glb_filename = f'{pid}.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)}')
|
||
print('上传glb文件到oss...')
|
||
oss_client.put_object_from_file(f'glbs/print/{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 platform_check(r):
|
||
try:
|
||
if r.llen('model:modelRepairGLBQueue') == 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 = redis.Redis(host='106.14.158.208', password='kcV2000', port=6379, db=6)
|
||
return
|
||
pid = r.lpop('model:modelRepairGLBQueue')
|
||
if pid is None: return
|
||
print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), f'pid：{pid} 生成平台审核模型 start', )
|
||
|
||
pid = pid.decode('utf-8')
|
||
down_obj_fromoss(pid)
|
||
obj_filename = f'{pid}.obj'
|
||
glb_filename = f'{pid}.glb'
|
||
os.system(f'gltfpack -c -i {os.path.join(workdir, "print", pid, obj_filename)} -o {os.path.join(workdir, "print", pid, glb_filename)}')
|
||
oss_client.put_object_from_file(f'glbs/print/{glb_filename}', os.path.join(workdir, "print", pid, glb_filename))
|
||
|
||
print(f'{update_check_url}?id={pid}')
|
||
res = requests.get(f'{update_check_url}?id={pid}')
|
||
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 main(r):
|
||
print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime()), '模型生成程序 start')
|
||
while True:
|
||
# team_check(r)
|
||
# platform_check(r)
|
||
make3d4print_task(r)
|
||
|
||
if __name__ == '__main__':
|
||
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.api.suwa3d.com/api/customerP3dLog/updateStatusToWaitingPlatformCheckingStatus'
|
||
update_team_check_url = 'https://mp.api.suwa3d.com/api/customerP3dLog/updateStatusToWaitingTeamCheckingStatus'
|
||
update_status_printstatus_url = 'https://mp.api.suwa3d.com/api/customerP3dLog/updateBuildPrintModelStatus'
|
||
update_makeprintobj_status_url = 'https://mp.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'
|
||
workdir = 'D:\\'
|
||
blenderbin = '"C:\\Program Files\\Blender Foundation\\Blender 3.3\\blender.exe"'
|
||
|
||
r = redis.Redis(host='106.14.158.208', password='kcV2000', port=6379, db=6)
|
||
main(r) |