You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
208 lines
9.1 KiB
208 lines
9.1 KiB
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) |