dongchangxi
3 weeks ago
3 changed files with 234 additions and 38 deletions
@ -0,0 +1,134 @@ |
|||||||
|
import os,shutil |
||||||
|
import redis |
||||||
|
import oss2,time,sys |
||||||
|
import requests |
||||||
|
import argparse,json |
||||||
|
|
||||||
|
# 将当前脚本所在目录添加到 Python 路径，以便导入 utils 模块 |
||||||
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) |
||||||
|
|
||||||
|
# from download_print_out import download_transform_save_by_json |
||||||
|
from utils.oss_redis import redisClient |
||||||
|
from utils.funcs import downloadJsonAndJpgFileAndMoveToCorrectDir, downloadDataByOssAndTransformSave |
||||||
|
# 默认使用脚本所在目录 |
||||||
|
currentDir = os.path.dirname(os.path.abspath(__file__)) |
||||||
|
|
||||||
|
ENV = 'prod' |
||||||
|
url = 'https://mp.api.suwa3d.com' |
||||||
|
if ENV == 'dev': |
||||||
|
url = 'http://mp.api.dev.com' |
||||||
|
elif ENV == 'prod': |
||||||
|
url = 'https://mp.api.suwa3d.com' |
||||||
|
|
||||||
|
|
||||||
|
#判断是否上传了 JSON 文件 |
||||||
|
def step1(versionId): |
||||||
|
|
||||||
|
# 下载json 文件 和 图片 |
||||||
|
dirName,machineInfo = downloadJsonAndJpgFileAndMoveToCorrectDir(versionId,currentDir) |
||||||
|
if not dirName: |
||||||
|
return False |
||||||
|
|
||||||
|
#判断是否是小机台 |
||||||
|
isSmallMachine = False |
||||||
|
if str(machineInfo["machine_type"]) == '1': |
||||||
|
isSmallMachine = True |
||||||
|
|
||||||
|
|
||||||
|
#下载数据，转换数据 |
||||||
|
res = downloadDataByOssAndTransformSave(dirName,isSmallMachine) |
||||||
|
if not res: |
||||||
|
return False |
||||||
|
|
||||||
|
#判断下载的obj文件数量和json里的是否一致,排除arrange文件夹 |
||||||
|
objFilePath = os.path.join(dirName, 'data') |
||||||
|
objCounts = 0 |
||||||
|
for file in os.listdir(objFilePath): |
||||||
|
if file == 'arrange': |
||||||
|
continue |
||||||
|
if file.endswith('.obj'): |
||||||
|
objCounts += 1 |
||||||
|
|
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 下载处理完成的obj文件数量: {objCounts}') |
||||||
|
# requestApiToUpdateSliceStatus(versionId,objCounts) |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# 读取 队列中一个数据出来 |
||||||
|
def main(work_dir=None): |
||||||
|
|
||||||
|
|
||||||
|
# 如果指定了工作目录，使用指定的目录 |
||||||
|
if work_dir: |
||||||
|
work_dir = os.path.abspath(work_dir) |
||||||
|
if not os.path.exists(work_dir): |
||||||
|
print(f'指定的工作目录不存在: {work_dir}，将创建该目录') |
||||||
|
os.makedirs(work_dir, exist_ok=True) |
||||||
|
currentDir = work_dir |
||||||
|
print(f'使用指定的工作目录: {currentDir}') |
||||||
|
else: |
||||||
|
print(f'没有指定工作目录，退出') |
||||||
|
exit(0) |
||||||
|
|
||||||
|
# 循环处理，直到队列为空 |
||||||
|
while True: |
||||||
|
r = redisClient() |
||||||
|
#检测队列是否有值 |
||||||
|
if r.scard('pb:sliceing') == 0: |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 队列为空，等待10秒') |
||||||
|
time.sleep(10) |
||||||
|
continue |
||||||
|
#获取队列中的值 |
||||||
|
data = r.spop('pb:sliceing') |
||||||
|
if data is None: |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 取出的数据为空，等待10秒') |
||||||
|
time.sleep(10) |
||||||
|
continue |
||||||
|
data = data.decode('utf-8') |
||||||
|
#判断是否是数字 |
||||||
|
if not data.isdigit(): |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 取出的数据不是数字，等待10秒') |
||||||
|
time.sleep(10) |
||||||
|
continue |
||||||
|
|
||||||
|
versionId = str(data) |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 正在处理版次ID={versionId}') |
||||||
|
res = step1(versionId) |
||||||
|
if res == False: |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} JSON文件下载数据失败，等待10秒') |
||||||
|
time.sleep(10) |
||||||
|
continue |
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# 在长时间操作后，确保 Redis 连接仍然有效 |
||||||
|
# 通过重新获取客户端来触发连接检查 |
||||||
|
try: |
||||||
|
r = redisClient() |
||||||
|
r.ping() # 测试连接 |
||||||
|
except Exception as e: |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} Redis连接检查失败: {str(e)}，将在下次循环时自动重连') |
||||||
|
|
||||||
|
#time.sleep(10) |
||||||
|
|
||||||
|
|
||||||
|
def testMain(): |
||||||
|
global currentDir |
||||||
|
currentDir = "/Users/dcx/code/make2/script/factory_sliceing_v2/tempData" |
||||||
|
versionId = '10153' #'10153 10158' |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 正在处理版次ID={versionId}') |
||||||
|
res = step1(versionId) |
||||||
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 处理完成,res={res}') |
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
testMain() |
||||||
|
# parser = argparse.ArgumentParser(description='排版打印订单处理程序') |
||||||
|
# parser.add_argument( |
||||||
|
# '--work-dir', |
||||||
|
# type=str, |
||||||
|
# default=None, |
||||||
|
# help='指定工作目录（磁盘路径），例如: D:/work 或 /Users/username/work。如果不指定，则使用脚本所在目录' |
||||||
|
# ) |
||||||
|
# args = parser.parse_args() |
||||||
|
# main(work_dir=args.work_dir) |
||||||
Loading…
Reference in new issue