53 lines
2.6 KiB
Python
53 lines
2.6 KiB
Python
import os, sys, oss2, time
|
||
|
||
def main():
|
||
pids = [pid for pid in os.listdir(workdir) if os.path.isdir(os.path.join(workdir, pid))]
|
||
for pid in pids:
|
||
if not os.path.exists(os.path.join(workdir, pid, f'{pid}.obj')):
|
||
print(f'文件不存在: {pid}.obj， 跳过处理{pid}')
|
||
continue
|
||
if not os.path.exists(os.path.join(workdir, pid, f'{pid}.mtl')):
|
||
print(f'文件不存在: {pid}.mtl， 跳过处理{pid}')
|
||
continue
|
||
if not os.path.exists(os.path.join(workdir, pid, f'{pid}Tex1.jpg')):
|
||
print(f'文件不存在: {pid}Tex1.jpg， 跳过处理{pid}')
|
||
continue
|
||
print(f'开始处理{pid}，先生成glb文件，删除oss上的文件，再上传模型文件和glb文件，最后删除本地文件')
|
||
# 生成glb文件
|
||
print(f'开始生成glb文件')
|
||
os.system(f'gltfpack -c -i {os.path.join(workdir, pid, pid + ".obj")} -o {os.path.join(workdir, pid, pid + ".glb")}')
|
||
# 根据前缀获取文件列表
|
||
prefix = f'objs/print/{pid}/'
|
||
filelist = oss2.ObjectIteratorV2(oss_client, prefix=prefix)
|
||
for file in filelist:
|
||
filename = file.key.split('/')[-1]
|
||
if filename == '': continue
|
||
print('正在删除:', file.key)
|
||
oss_client.delete_object(file.key)
|
||
|
||
print(f'开始上传{pid}的文件')
|
||
oss_client.put_object_from_file(f'objs/print/{pid}/{pid}.obj', os.path.join(workdir, pid, f'{pid}.obj'))
|
||
oss_client.put_object_from_file(f'objs/print/{pid}/{pid}.mtl', os.path.join(workdir, pid, f'{pid}.mtl'))
|
||
oss_client.put_object_from_file(f'objs/print/{pid}/{pid}Tex1.jpg', os.path.join(workdir, pid, f'{pid}Tex1.jpg'))
|
||
oss_client.put_object_from_file(f'glbs/print/{pid}.glb', os.path.join(workdir, pid, f'{pid}.glb'))
|
||
|
||
print(f'删除本地文件')
|
||
for file in os.listdir(os.path.join(workdir, pid)):
|
||
os.remove(os.path.join(workdir, pid, file))
|
||
os.rmdir(os.path.join(workdir, pid))
|
||
|
||
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid}处理完成')
|
||
|
||
str_pids = '、'.join(pids)
|
||
print(f'模型重新生成已完成，可以下载打印了。共处理{len(pids)}个模型：{str_pids}')
|
||
|
||
|
||
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)
|
||
|
||
workdir = '/home/water/Downloads/objs/temp/'
|
||
main() |