base_deal.py 62 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535
  1. import json
  2. from .module_generate_goods_art_no_table import GenerateGoodsArtNoTable
  3. from func_timeout import FunctionTimedOut
  4. from .grenerate_main_image_test import GeneratePic
  5. from threading import Lock
  6. from middleware import UnicornException
  7. import settings
  8. from collections import defaultdict
  9. from .remove_bg_ali import RemoveBgALi, Picture
  10. from .deal_cutout import DealCutout
  11. import time
  12. from .image_pic_deal import OnePicDeal
  13. from natsort import natsorted, ns
  14. import os
  15. import shutil
  16. import exifread
  17. import datetime
  18. from databases import DeviceConfig, SqlQuery, CRUD
  19. from model.photo_record import PhotoRecord
  20. import requests
  21. import copy, asyncio
  22. from settings import sendSocketMessage
  23. from utils.common import message_queue
  24. from logger import logger
  25. def sendAsyncMessage(msg="", goods_arts=[], status="",progress={}):
  26. """异步发送消息"""
  27. data = {
  28. "code": 0,
  29. "msg": msg,
  30. "status": 2,
  31. "data": {
  32. "status": status,
  33. "goods_art_nos": goods_arts,
  34. },
  35. "progress":{
  36. "msg_type":"segment_progress",
  37. "name":"抠图",
  38. "goods_art_no":progress.get("goods_art_no",""),
  39. "status":progress.get("status"),
  40. "current":progress.get("current",0),
  41. "total":progress.get("total",0),
  42. "error":progress.get("error",0)
  43. },
  44. "msg_type": "segment_progress",
  45. }
  46. message_queue.put_nowait(data)
  47. """
  48. 照片自动货号匹配 将图片放置在指定文件夹下,并自动对应不同的货号进行整理
  49. """
  50. _Type = [".png", ".PNG", ".jpg", ".JPG", ".gif", ".GIF", ".jpge", ".JPGE"]
  51. class BaseDealImage(object):
  52. def __init__(self, image_dir=None, token=None):
  53. self.token = token
  54. self.goods_images_count_dict = defaultdict(int)
  55. # 数据模型
  56. # self.data_mode_auto_deal_pics = DataModeAutoDealPics()
  57. self.image_dir = image_dir
  58. pass
  59. def run_main(
  60. self,
  61. all_goods_art_no_folder_data,
  62. callback_func=None,
  63. cutout_mode=None,
  64. resize_image_view=None,
  65. windows=None,
  66. logo_path=None,
  67. image_order_list=None,
  68. ):
  69. # 对所有缺失已抠图的进行抠图处理
  70. self.run_cutout_image(
  71. all_goods_art_no_folder_data=all_goods_art_no_folder_data,
  72. callback_func=callback_func,
  73. cutout_mode=cutout_mode,
  74. windows=windows,
  75. )
  76. error_num = 0
  77. successful_num = 0
  78. for goods_art_no_folder_data in all_goods_art_no_folder_data:
  79. if goods_art_no_folder_data["label"] != "待处理":
  80. continue
  81. if windows:
  82. if windows.state != 1:
  83. break
  84. folder_name = goods_art_no_folder_data["folder_name"]
  85. callback_func("开始处理文件夹========== {} ".format(folder_name))
  86. if settings.IS_TEST:
  87. flag = self.shoes_run_one_folder_to_deal(
  88. goods_art_no_folder_data=goods_art_no_folder_data,
  89. resize_image_view=resize_image_view,
  90. logo_path=logo_path,
  91. image_order_list=image_order_list,
  92. callback_func=callback_func,
  93. windows=windows,
  94. )
  95. if flag is None:
  96. callback_func("货号:{} 数据异常".format(folder_name))
  97. else:
  98. if flag:
  99. successful_num += 1
  100. callback_func("货号:{} 图片生成处理成功".format(folder_name))
  101. else:
  102. error_num += 1
  103. callback_func("货号:{} 图片生成处理失败".format(folder_name))
  104. else:
  105. try:
  106. flag = self.shoes_run_one_folder_to_deal(
  107. goods_art_no_folder_data=goods_art_no_folder_data,
  108. resize_image_view=resize_image_view,
  109. logo_path=logo_path,
  110. image_order_list=image_order_list,
  111. callback_func=callback_func,
  112. windows=windows,
  113. )
  114. if flag is None:
  115. callback_func("货号:{} 数据异常".format(folder_name))
  116. else:
  117. if flag:
  118. successful_num += 1
  119. callback_func(
  120. "货号:{} 图片生成处理成功".format(folder_name)
  121. )
  122. else:
  123. error_num += 1
  124. callback_func(
  125. "货号:{} 图片生成处理失败".format(folder_name)
  126. )
  127. except BaseException as e:
  128. error_num += 1
  129. import traceback
  130. traceback.print_exc()
  131. callback_func(
  132. "货号:{} 图片生成处理异常,原因:{}".format(folder_name, e)
  133. )
  134. callback_func("处理成功:{}个,失败:{}".format(successful_num, error_num))
  135. def checkImageAmount(
  136. self, image_dir: str, amount: int, todo_goods_art_no_folder_name_list=None
  137. ) -> dict:
  138. result = {"code": 0, "msg": "", "data": {}}
  139. for goods_art_no_folder in self.list_dir(image_dir):
  140. # 指定内容检查
  141. if todo_goods_art_no_folder_name_list is not None:
  142. if goods_art_no_folder not in todo_goods_art_no_folder_name_list:
  143. continue
  144. if not os.path.isdir("{}/{}".format(image_dir, goods_art_no_folder)):
  145. continue
  146. if "软件" in goods_art_no_folder:
  147. continue
  148. if "无法" in goods_art_no_folder:
  149. continue
  150. if "原始图" not in self.list_dir(
  151. "{}/{}".format(image_dir, goods_art_no_folder)
  152. ):
  153. result["data"][goods_art_no_folder] = "文件夹下,没有 原始图 文件夹\n"
  154. continue
  155. # 计算单个文件夹原图数量
  156. images = [
  157. x
  158. for x in self.list_dir(
  159. "{}/{}/原始图".format(image_dir, goods_art_no_folder)
  160. )
  161. ]
  162. image_num = 0
  163. for pic_file_name in images:
  164. _, e = os.path.splitext(pic_file_name)
  165. if e in _Type:
  166. image_num += 1
  167. # if self.number_pictures != 0:
  168. if image_num > amount:
  169. result["data"][goods_art_no_folder] = "货号图片大于{}张~\n".format(
  170. amount
  171. )
  172. if image_num < 2:
  173. result["data"][goods_art_no_folder] = "货号图片小于于2张~\n"
  174. if result["data"]:
  175. result["code"] = 1
  176. return result
  177. def check_folders_image_amount(
  178. self, all_goods_art_no_folder_data, image_order_list
  179. ):
  180. print("*****************check_folders_image_amount************************")
  181. amount = len(image_order_list)
  182. # print("图片数量检查===>", amount)
  183. message = ""
  184. for goods_art_no_folder_data in all_goods_art_no_folder_data:
  185. if goods_art_no_folder_data["label"] != "待处理":
  186. continue
  187. folder_path = goods_art_no_folder_data["folder_path"]
  188. folder_name = goods_art_no_folder_data["folder_name"]
  189. images = [x for x in self.list_dir("{}/原始图".format(folder_path))]
  190. image_num = 0
  191. for pic_file_name in images:
  192. _, e = os.path.splitext(pic_file_name)
  193. if e in _Type:
  194. image_num += 1
  195. # print("图片数量判断===>", image_num)
  196. # if image_num > amount:
  197. # goods_art_no_folder_data["label"] = "错误"
  198. # message += '货号{}:图片大于{}张~\n'.format(folder_name, amount)
  199. # if image_num < 2:
  200. # message += '货号{}:图片小于于2张~\n'.format(folder_name)
  201. return all_goods_art_no_folder_data, message
  202. def check_one_folder_image_amount(self, folder_data, amount: int):
  203. print("*****************check_one_folder_image_amount************************")
  204. # 计算单个文件夹原图数量
  205. images = [
  206. x for x in self.list_dir("{}/原始图".format(folder_data["folder_path"]))
  207. ]
  208. image_num = 0
  209. for pic_file_name in images:
  210. _, e = os.path.splitext(pic_file_name)
  211. if e in _Type:
  212. image_num += 1
  213. if image_num > amount:
  214. return False, "货号{}:图片大于{}张~\n".format(
  215. folder_data["folder_name"], amount
  216. )
  217. if image_num < 2:
  218. return False, "货号{}:图片小于于2张~\n".format(folder_data["folder_name"])
  219. return True, ""
  220. # 指定的图片顺序
  221. def getImageOrder(self, image_order: str, resize_image_view: str):
  222. imageOrderList = (
  223. image_order.replace(",", ",").replace(" ", "").replace("图", "").split(",")
  224. )
  225. if len(set(imageOrderList)) != len(imageOrderList):
  226. return {"code": 1, "msg": "图片位置与顺序重复,请检查您的输入"}
  227. for val in imageOrderList:
  228. if val not in [
  229. "俯视",
  230. "侧视",
  231. "后跟",
  232. "鞋底",
  233. "内里",
  234. "组合",
  235. "组合2",
  236. "组合3",
  237. "组合4",
  238. "组合5",
  239. ]:
  240. return {
  241. "code": 1,
  242. "msg": "可选项为:俯视,侧视,后跟,鞋底,内里,组合,组合2,组合3,组合4,组合5",
  243. }
  244. if resize_image_view not in imageOrderList:
  245. return {"code": 1, "msg": "缩小的步骤必须是你填写的图片顺序中"}
  246. return {"code": 0, "msg": "sucess", "imageOrderList": imageOrderList}
  247. def shoes_run_one_folder_to_deal(
  248. self,
  249. goods_art_no_folder_data,
  250. image_order_list: list,
  251. resize_image_view: str,
  252. logo_path="",
  253. windows=None,
  254. callback_func=None,
  255. ):
  256. """
  257. 操作步骤:
  258. 1、查询每个图片的角度
  259. 2、
  260. """
  261. is_successful = True
  262. folder_path = goods_art_no_folder_data["folder_path"]
  263. folder_name = goods_art_no_folder_data["folder_name"]
  264. all_original_images = self.get_images("{}/原始图".format(folder_path))
  265. self.check_path("{}/800x800".format(folder_path))
  266. self.crate_all_folders(folder_path)
  267. print("all_original_images====>", all_original_images)
  268. if not all_original_images:
  269. return None
  270. # _ = ["俯视", "侧视", "后跟", "鞋底", "内里"]
  271. for index, image_dict in enumerate(all_original_images):
  272. if index < len(image_order_list):
  273. image_dict["image_view"] = image_order_list[index]
  274. else:
  275. image_dict["image_view"] = "其他{}".format(
  276. len(image_order_list) - index + 1
  277. )
  278. # ====================处理所有图片的顺序====================
  279. # ["俯视", "侧视", "后跟", "鞋底", "内里"]
  280. _config = {
  281. "俯视": 1,
  282. "侧视": 2,
  283. "后跟": 3,
  284. "鞋底": 4,
  285. "内里": 5,
  286. }
  287. r = time.time()
  288. n = 0
  289. _index = 0
  290. # 检查是否有重复的角度
  291. _d_views = []
  292. f = True
  293. for image_dict in all_original_images:
  294. n += 1
  295. _index += 1
  296. image_dict["old_file_name"] = image_dict["file_name"]
  297. image_dict["random_name"] = "{}-{}".format(r, n)
  298. if image_dict["image_view"] in _config:
  299. if image_dict["image_view"] not in _d_views:
  300. _d_views.append(image_dict["image_view"])
  301. else:
  302. callback_func("货号:{} 处理失败".format(folder_name))
  303. # self.show_progress_detail("货号图{} 存在多个{} 角度~".format(goods_art_no_folder, image_dict["image_view"]))
  304. return None
  305. image_dict["index"] = _config[image_dict["image_view"]]
  306. else:
  307. image_dict["index"] = _index
  308. all_original_images.sort(key=lambda x: x["index"])
  309. # ==========直接进行处理=============
  310. i_n = 0
  311. image_index = 0 # 图片顺序
  312. is_image_deal_mode = 0
  313. # 删除目录再新建
  314. try:
  315. if os.path.exists("{}/阴影图处理".format(folder_path)):
  316. shutil.rmtree("{}/阴影图处理".format(folder_path),onerror=settings.handle_remove_readonly)
  317. except Exception as e:
  318. print('An exception occurred')
  319. logger.info(f"base deal 抠图前目录删除出现问题:{str(e)}")
  320. self.crate_all_folders(folder_path)
  321. print(
  322. "***************all_original_images*********************",
  323. all_original_images,
  324. )
  325. for image_dict in all_original_images:
  326. if windows:
  327. if windows.state != 1:
  328. return None
  329. i_n += 1
  330. image_index += 1
  331. original_image_path = "{}/原始图/{}{}".format(
  332. folder_path, image_dict["file_name"], image_dict["e"]
  333. )
  334. file_name = image_dict["file_name"]
  335. print("正在处理,货号:{}".format(folder_path))
  336. # self.show_progress_detail("正在处理,货号:{}".format(file_name))
  337. # 该文件在800images下没有时,则进行生成新的抠图
  338. # 检查是否存在已抠图文件,如没有再去抠图
  339. original_move_bg_image_path = "{}/原始图_已抠图/{}{}".format(
  340. folder_path, image_dict["file_name"], ".png"
  341. )
  342. print(
  343. f"*****************此处判断鞋子是否为左右脚====>{image_index}<======={is_image_deal_mode}=======>**********************"
  344. )
  345. # 此处判断鞋子是否为左右脚
  346. if image_index == 1:
  347. is_image_deal_mode = 0
  348. print("开始识别左右脚=========>")
  349. if OnePicDeal(self.token).check_shoe_is_right(
  350. image_path=original_move_bg_image_path
  351. ):
  352. is_image_deal_mode = 1 # 1表示要镜像,0表示不做镜像
  353. print(
  354. "*************************进行800image 生成********************************************"
  355. )
  356. """进行800image 生成"""
  357. generate_pic = GeneratePic()
  358. print(
  359. "*************************进行800image 结束====>>>>********************************************"
  360. )
  361. out_pci_mode = "." + settings.getSysConfigs(
  362. "basic_configs", "image_out_format", "png"
  363. )
  364. if out_pci_mode == ".jpg":
  365. out_path = "{}/800x800/{}{}".format(folder_path, file_name, ".jpg")
  366. elif out_pci_mode == ".png":
  367. out_path = "{}/800x800/{}{}".format(folder_path, file_name, ".png")
  368. else:
  369. out_path = "{}/800x800/{}{}".format(
  370. folder_path, file_name, out_pci_mode
  371. )
  372. out_process_path_1 = "{}/阴影图处理/{}_{}_阴影{}".format(
  373. folder_path, file_name, image_dict["image_view"], ".png"
  374. )
  375. out_process_path_2 = "{}/阴影图处理/{}_{}_抠图{}".format(
  376. folder_path, file_name, image_dict["image_view"], ".png"
  377. )
  378. resize_mode = 1
  379. max_box = None
  380. print("------------1", image_dict["image_view"], resize_image_view)
  381. if image_dict["image_view"] == resize_image_view:
  382. print(image_dict["image_view"], resize_image_view)
  383. resize_mode = 2
  384. if (
  385. settings.getSysConfigs("other_configs", "product_type", "鞋类")
  386. == "皮具"
  387. ):
  388. max_box = (1000, 1200)
  389. out_pic_size = (
  390. [1600]
  391. if settings.getSysConfigs("basic_configs", "main_image_size", [1600])
  392. == ""
  393. else settings.getSysConfigs("basic_configs", "main_image_size", [1600])
  394. ) # 主图大小
  395. if resize_mode == 2:
  396. print(
  397. "参数打印================》",
  398. is_image_deal_mode,
  399. resize_mode,
  400. out_pic_size,
  401. True if i_n == 1 else False,
  402. max_box,
  403. )
  404. print("**********123456********************")
  405. curve_mask = True if "俯视" in image_order_list else False
  406. if not generate_pic.run(
  407. image_path=original_image_path,
  408. cut_image_path=original_move_bg_image_path,
  409. out_path=out_path,
  410. image_deal_mode=is_image_deal_mode,
  411. # resize_mode=resize_mode,
  412. resize_mode=1,#将这里得缩放模式改为强制不缩放 2025-10-22
  413. out_pic_size=out_pic_size,
  414. is_logo=True if i_n == 1 else False,
  415. out_process_path_1=out_process_path_1,
  416. out_process_path_2=out_process_path_2,
  417. max_box=max_box,
  418. logo_path=logo_path,
  419. curve_mask=curve_mask,
  420. ):
  421. print("**********222222222222222222222222222********************")
  422. is_successful = False
  423. if is_successful:
  424. return True
  425. else:
  426. return False
  427. def to_upload_pic(self, file_path, is_resize=True):
  428. file_name = os.path.split(file_path)[1]
  429. e = os.path.splitext(file_name)[1][1:]
  430. im = Picture(file_path)
  431. if im.x > 500:
  432. im.resize(width=500)
  433. _ = {
  434. "jpg": "JPEG",
  435. "JPG": "JPEG",
  436. "JPEG": "JPEG",
  437. "jpeg": "JPEG",
  438. "png": "PNG",
  439. "PNG": "PNG",
  440. }
  441. e = _[e]
  442. image_io = im.save_to_io(e)
  443. goods_data = {
  444. "file_path": os.path.split(file_path)[1],
  445. "image_io": image_io,
  446. "e": e,
  447. }
  448. url = self.data_mode_image_cut.get_online_data.upload_pic(goods_data=goods_data)
  449. return url
  450. def get_images(self, path):
  451. image_list = [] # 过滤非图片数据
  452. for _file in self.list_dir(path):
  453. file_name, e = os.path.splitext(_file)
  454. if e in _Type:
  455. image_list.append(
  456. {
  457. "file_path": "{}/{}".format(path, _file),
  458. "file_name": file_name,
  459. "e": e,
  460. }
  461. )
  462. return image_list
  463. def crate_all_folders(self, root_path):
  464. path_list = ["800x800", "原始图_已抠图", "阴影图处理"]
  465. for i in path_list:
  466. path = "{}/{}".format(root_path, i)
  467. self.check_path(path)
  468. def run_cutout_image(
  469. self,
  470. all_goods_art_no_folder_data,
  471. callback_func=None,
  472. cutout_mode=1,
  473. windows=None,
  474. ):
  475. """
  476. 处理所有的抠图
  477. """
  478. callback_func("开始处理抠图~")
  479. goods_art_nos = [
  480. goods_art_item["folder_name"]
  481. for goods_art_item in all_goods_art_no_folder_data
  482. ]
  483. total_progress = len(all_goods_art_no_folder_data)
  484. finish_progress = 0
  485. error_progress = 0
  486. progress = {"status":"正在处理",
  487. "current":finish_progress,
  488. "total":total_progress,
  489. "error":error_progress}
  490. sendAsyncMessage(
  491. msg="开始处理抠图", goods_arts=goods_art_nos, status="开始处理",progress=progress
  492. )
  493. error_goods_art_no_folder = []
  494. for goods_art_no_folder_data in all_goods_art_no_folder_data:
  495. if goods_art_no_folder_data["label"] != "待处理":
  496. continue
  497. folder_path = goods_art_no_folder_data["folder_path"]
  498. self.crate_all_folders(folder_path)
  499. # 检查是否存在已抠图文件,如没有再去抠图
  500. images = [x for x in self.list_dir("{}/原始图".format(folder_path))]
  501. cutImageList = []
  502. goods_art_floder_name = goods_art_no_folder_data["folder_name"]
  503. progress = {"status":"正在处理",
  504. "current":finish_progress,
  505. "total":total_progress,
  506. "error":error_progress,
  507. "goods_art_no":goods_art_floder_name
  508. }
  509. sendAsyncMessage(
  510. msg="正在抠图",
  511. goods_arts=[goods_art_floder_name],
  512. status="处理中",
  513. progress=progress
  514. )
  515. for pic_file_name in images:
  516. if windows:
  517. if windows.state != 1:
  518. break
  519. # 根据名称判断,没有抠图过的,进行统计
  520. file_name, suffix = os.path.splitext(pic_file_name)
  521. if suffix in _Type:
  522. original_image_path = "{}/原始图/{}".format(
  523. folder_path, pic_file_name
  524. )
  525. original_move_bg_image_path = "{}/原始图_已抠图/{}{}".format(
  526. folder_path, file_name, ".png"
  527. )
  528. if not os.path.exists(original_move_bg_image_path):
  529. # 没有抠图文件,进行抠图生成
  530. callback_func("正在抠图 货号:{}".format(file_name))
  531. if cutout_mode == "2":
  532. cutImageList.append(
  533. {
  534. "file_name": file_name, # 文件名
  535. "file_e": suffix, # 后缀,.jpg
  536. "file_path": original_image_path, # 完整路径
  537. "file": "{}{}".format(
  538. file_name, suffix
  539. ), # 图片文件名,带后缀
  540. "need_cutout": True, # 必须,需要抠图
  541. "out_path": original_move_bg_image_path,
  542. }
  543. )
  544. else:
  545. remove_pic_ins = RemoveBgALi()
  546. if settings.IS_TEST:
  547. im = remove_pic_ins.get_image_cut(
  548. file_path=original_image_path,
  549. out_file_path=original_move_bg_image_path,
  550. )
  551. else:
  552. try:
  553. im = remove_pic_ins.get_image_cut(
  554. file_path=original_image_path,
  555. out_file_path=original_move_bg_image_path,
  556. )
  557. except FunctionTimedOut as f:
  558. callback_func(
  559. "货号图{} 抠图处理超时~".format(file_name)
  560. )
  561. error_goods_art_no_folder.append(folder_path)
  562. im = None
  563. except BaseException as e:
  564. callback_func(
  565. "货号图{} 抠图处理失败,原因{}".format(
  566. file_name, e
  567. )
  568. )
  569. error_goods_art_no_folder.append(folder_path)
  570. im = None
  571. if not im:
  572. callback_func(
  573. "货号图{} 抠图处理失败~".format(file_name)
  574. )
  575. error_goods_art_no_folder.append(folder_path)
  576. continue
  577. else:
  578. callback_func("货号图{} 抠图完成~".format(file_name))
  579. progress = {
  580. "status":"正在处理",
  581. "current":finish_progress,
  582. "total":total_progress,
  583. "error":error_progress,
  584. "goods_art_no":goods_art_floder_name
  585. }
  586. if goods_art_floder_name not in error_goods_art_no_folder:
  587. finish_progress+=1
  588. sendAsyncMessage(
  589. msg="正在处理",
  590. goods_arts=[goods_art_floder_name],
  591. status="正在处理",
  592. progress=progress
  593. )
  594. else:
  595. progress["status"] = "处理失败"
  596. sendAsyncMessage(
  597. msg="处理失败",
  598. goods_arts=[goods_art_floder_name],
  599. status="处理失败",
  600. progress=progress
  601. )
  602. if cutout_mode == "2":
  603. dealCutout = DealCutout(windows=None, token=self.token)
  604. dealCutout.need_cutout_images = cutImageList
  605. dealCutout.run()
  606. while True:
  607. time.sleep(0.5)
  608. if dealCutout.state == 3:
  609. if len(dealCutout.resultData) != len(cutImageList):
  610. error_goods_art_no_folder.append(folder_path)
  611. break
  612. error_progress = len(error_goods_art_no_folder)
  613. progress = {
  614. "status":"处理完成",
  615. "current":finish_progress,
  616. "total":total_progress,
  617. "error":error_progress
  618. }
  619. if error_goods_art_no_folder:
  620. print("以下货号抠图失败~\n {}".format(error_goods_art_no_folder))
  621. callback_func("以下货号抠图失败~\n {}".format(error_goods_art_no_folder))
  622. sendAsyncMessage(
  623. msg=f"抠图处理失败",
  624. goods_arts=error_goods_art_no_folder,
  625. status="抠图处理失败",
  626. # progress=progress
  627. )
  628. else:
  629. pass
  630. progress["status"] = "处理失败" if error_progress == total_progress else "处理完成"
  631. sendAsyncMessage(
  632. msg="抠图完成",
  633. goods_arts=[],
  634. status="抠图完成",
  635. progress=progress
  636. )
  637. def checkCutoutImage(self, image_dir: str, todo_goods_art_no_folder_name_list=None):
  638. """
  639. 进行图片检查,不合规的直接提示
  640. """
  641. error_goods_art_no_folder = []
  642. self.check_path("{}/软件-处理失败".format(image_dir))
  643. for goods_art_no_folder in self.list_dir(image_dir):
  644. # 指定内容检查
  645. if todo_goods_art_no_folder_name_list is not None:
  646. if goods_art_no_folder not in todo_goods_art_no_folder_name_list:
  647. continue
  648. if not os.path.isdir("{}/{}".format(image_dir, goods_art_no_folder)):
  649. continue
  650. if "软件" in goods_art_no_folder:
  651. continue
  652. if "无法" in goods_art_no_folder:
  653. continue
  654. if "原始图" not in self.list_dir(
  655. "{}/{}".format(image_dir, goods_art_no_folder)
  656. ):
  657. error_goods_art_no_folder.append(goods_art_no_folder)
  658. continue
  659. self.check_path(
  660. "{}/{}/原始图_已抠图".format(image_dir, goods_art_no_folder)
  661. )
  662. self.check_path("{}/{}/800x800".format(image_dir, goods_art_no_folder))
  663. self.check_path("{}/{}/阴影图处理".format(image_dir, goods_art_no_folder))
  664. if error_goods_art_no_folder:
  665. self.move_folders(
  666. path_list=[
  667. "{}/{}".format(self.image_dir, x) for x in error_goods_art_no_folder
  668. ],
  669. target_folder="{}/软件-处理失败".format(self.image_dir),
  670. )
  671. return False
  672. def move_folders(self, path_list, target_folder):
  673. for source_folder in path_list:
  674. shutil.move(source_folder, target_folder)
  675. def rename_folder_for_hqt(self, all_goods_art_no_folder_data):
  676. """
  677. 步骤:
  678. 规整红蜻蜓的文件名
  679. 重新按文件名进行命名
  680. """
  681. goods_art_no_list = []
  682. for goods_art_no_folder_data in all_goods_art_no_folder_data:
  683. if "@" not in goods_art_no_folder_data["folder_name"]:
  684. goods_art_no_folder_data["label"] = "待处理"
  685. goods_art_no_list.append(goods_art_no_folder_data["folder_name"])
  686. else:
  687. goods_art_no_folder_data["label"] = "不处理"
  688. if goods_art_no_list:
  689. # goods_art_no_dict 文件夹与货号的字典
  690. goods_art_no_dict = self.get_data_from_hqt_with_goods_art_no(
  691. goods_art_no_list=goods_art_no_list
  692. )
  693. for goods_art_no_folder_data in all_goods_art_no_folder_data:
  694. if goods_art_no_folder_data["label"] != "待处理":
  695. continue
  696. goods_art_no_folder = goods_art_no_folder_data["folder_name"]
  697. if goods_art_no_folder in goods_art_no_dict:
  698. print(goods_art_no_folder)
  699. old_folder_path = goods_art_no_folder_data["folder_path"]
  700. new_folder_name = "{}@NUM{}".format(
  701. goods_art_no_folder,
  702. goods_art_no_dict[goods_art_no_folder]["编号"],
  703. )
  704. new_folder_path = "{}/{}".format(
  705. goods_art_no_folder_data["root_path"], new_folder_name
  706. )
  707. try:
  708. os.rename(old_folder_path, new_folder_path)
  709. goods_art_no_folder_data["folder_path"] = new_folder_path
  710. goods_art_no_folder_data["folder_name"] = new_folder_path
  711. goods_art_no_folder_data["label"] = "待处理"
  712. except BaseException as e:
  713. goods_art_no_folder_data["label"] = "不处理"
  714. print("521 文件夹重名命失败:{}".format(e))
  715. # 重新规整修改图片名称
  716. for goods_art_no_folder_data in all_goods_art_no_folder_data:
  717. if goods_art_no_folder_data["label"] != "待处理":
  718. continue
  719. goods_art_no_folder = goods_art_no_folder_data["folder_name"]
  720. _img_all = self.list_dir(
  721. "{}/原始图".format(goods_art_no_folder_data["folder_path"])
  722. )
  723. index = 0
  724. for _file in _img_all:
  725. file_name, suffix = os.path.splitext(_file)
  726. if suffix in _Type:
  727. index += 1
  728. folder_path = goods_art_no_folder_data["folder_path"]
  729. new_file_name = "{}({}){}".format(
  730. goods_art_no_folder, index, suffix
  731. )
  732. new_path = "{}/原始图/{}".format(folder_path, new_file_name)
  733. old_path = "{}/原始图/{}".format(folder_path, _file)
  734. crop_new_path = "{}/原始图_已抠图/{}".format(
  735. folder_path, "{}({}).png".format(goods_art_no_folder, index)
  736. )
  737. crop_old_path = "{}/原始图_已抠图/{}".format(
  738. folder_path, "{}.png".format(file_name)
  739. )
  740. if old_path != new_path:
  741. # 存在货号命名错误的,进行修正
  742. try:
  743. if os.path.exists(old_path):
  744. os.rename(old_path, new_path)
  745. if os.path.exists(crop_old_path):
  746. os.rename(crop_old_path, crop_new_path)
  747. except BaseException as e:
  748. goods_art_no_folder_data["label"] = "不处理"
  749. print("550 文件夹重名命失败:{}".format(e))
  750. def cutImagePiju(
  751. self,
  752. image_dir: str,
  753. image_order="",
  754. is_check_number=True,
  755. is_filter=True,
  756. resize_image_view="后跟",
  757. callback_func=None,
  758. event=None,
  759. todo_goods_art_no_folder_name_list=None,
  760. ):
  761. """
  762. 1、遍历文件夹,基于生成的结果图看哪些需要进行抠图等处理
  763. 2、压缩并上传平台获取抠图
  764. 3、抠图处理成白底图
  765. 4、做成800*800/200*200
  766. :return:
  767. """
  768. logo_path = ""
  769. res = self.getImageOrder(
  770. image_order=image_order, resize_image_view=resize_image_view
  771. )
  772. if res["code"] != 0:
  773. callback_func(res["msg"])
  774. return {"code": 1, "msg": res["msg"]}
  775. imageOrderList = res["imageOrderList"]
  776. """扫描文档,检查有哪些需要进行抠图等处理"""
  777. self.lock = Lock()
  778. to_do_images_total = 0
  779. error_goods_art_no_folder = []
  780. for goods_art_no_folder in self.list_dir(image_dir):
  781. # 指定内容检查
  782. if todo_goods_art_no_folder_name_list is not None:
  783. if goods_art_no_folder not in todo_goods_art_no_folder_name_list:
  784. continue
  785. if not os.path.isdir("{}/{}".format(image_dir, goods_art_no_folder)):
  786. continue
  787. self.check_path("{}/{}/原始图".format(image_dir, goods_art_no_folder))
  788. self.check_path(
  789. "{}/{}/原始图_已抠图".format(image_dir, goods_art_no_folder)
  790. )
  791. self.check_path("{}/{}/800x800".format(image_dir, goods_art_no_folder))
  792. self.check_path("{}/{}/阴影图处理".format(image_dir, goods_art_no_folder))
  793. # 遍历原始图片文件夹
  794. all_original_images = [
  795. x
  796. for x in self.list_dir(
  797. "{}/{}/原始图".format(image_dir, goods_art_no_folder)
  798. )
  799. ]
  800. # 检查已抠图文件夹
  801. all_moved_images = [
  802. os.path.splitext(x)[0]
  803. for x in self.list_dir(
  804. "{}/{}/原始图_已抠图".format(image_dir, goods_art_no_folder)
  805. )
  806. ]
  807. all_800images = [
  808. os.path.splitext(x)[0]
  809. for x in self.list_dir(
  810. "{}/{}/800x800".format(image_dir, goods_art_no_folder)
  811. )
  812. ]
  813. if is_check_number and len(imageOrderList) != len(all_original_images):
  814. callback_func(
  815. "{} 文件夹下图片数量与订单数量不一致,请检查!".format(
  816. goods_art_no_folder
  817. )
  818. )
  819. return {
  820. "code": 1,
  821. "msg": "{} 文件夹下图片数量与订单数量不一致,请检查!".format(
  822. goods_art_no_folder
  823. ),
  824. }
  825. all_800images = []
  826. image_num = 0
  827. for pic_file_name in all_original_images:
  828. if pic_file_name not in all_800images:
  829. # 根据名称判断,没有抠图过的,进行统计
  830. _, e = os.path.splitext(pic_file_name)
  831. print(e)
  832. if e in _Type:
  833. image_num += 1
  834. print("----------》", goods_art_no_folder, pic_file_name)
  835. to_do_images_total += 1
  836. # if image_num > 5:
  837. # error_goods_art_no_folder.append(goods_art_no_folder)
  838. # if error_goods_art_no_folder:
  839. # self.show_progress_detail("以下货号图片张数超过5张~\n {}".format(error_goods_art_no_folder))
  840. # self.set_state(state_value=2)
  841. # return
  842. if to_do_images_total > 0:
  843. # self.progress_sign.emit({"type": "处理图片 抠图、加工等", "progress_bar_value": 0})
  844. for goods_art_no_folder in self.list_dir(image_dir):
  845. # 指定内容检查
  846. if todo_goods_art_no_folder_name_list is not None:
  847. if goods_art_no_folder not in todo_goods_art_no_folder_name_list:
  848. continue
  849. if not os.path.isdir("{}/{}".format(image_dir, goods_art_no_folder)):
  850. continue
  851. self.run_one_folder_to_deal(
  852. goods_art_no_folder=goods_art_no_folder,
  853. image_dir=image_dir,
  854. image_order=image_order,
  855. resize_image_view=resize_image_view,
  856. callback_func=callback_func,
  857. logo_path=logo_path,
  858. )
  859. else:
  860. # self.show_progress_detail("没有需要处理的图片~")
  861. callback_func("没有需要处理的图片~")
  862. # self.set_state(state_value=2)
  863. return {"code": 0, "msg": "ok"}
  864. def run_one_folder_to_deal(
  865. self,
  866. goods_art_no_folder,
  867. image_dir,
  868. image_order,
  869. resize_image_view,
  870. callback_func=None,
  871. logo_path="",
  872. ):
  873. _img_all = self.list_dir("{}/{}/原始图".format(image_dir, goods_art_no_folder))
  874. all_original_images = [] # 过滤非图片数据
  875. index = 0
  876. for _file in _img_all:
  877. file_name, e = os.path.splitext(_file)
  878. if e in _Type:
  879. index += 1
  880. new_file_name = "{}({}){}".format(goods_art_no_folder, index, e)
  881. new_path = "{}/{}/原始图/{}".format(
  882. image_dir, goods_art_no_folder, new_file_name
  883. )
  884. old_path = "{}/{}/原始图/{}".format(
  885. image_dir, goods_art_no_folder, _file
  886. )
  887. if old_path != new_path:
  888. # 存在货号命名错误的,进行修正
  889. try:
  890. os.rename(old_path, new_path)
  891. except:
  892. pass
  893. all_original_images.append(new_file_name)
  894. if os.path.exists(
  895. "{}/{}/原始图/镜像.txt".format(image_dir, goods_art_no_folder)
  896. ):
  897. file_mirror_mark = True
  898. else:
  899. file_mirror_mark = None
  900. # if goods_art_no_folder == "AC51016112":
  901. # print(file_mirror_mark)
  902. # raise 111
  903. all_moved_images = [
  904. os.path.splitext(x)[0]
  905. for x in self.list_dir(
  906. "{}/{}/原始图_已抠图".format(image_dir, goods_art_no_folder)
  907. )
  908. ]
  909. all_800images = [
  910. os.path.splitext(x)[0]
  911. for x in self.list_dir(
  912. "{}/{}/800x800".format(image_dir, goods_art_no_folder)
  913. )
  914. ]
  915. all_800images = []
  916. # 检查哪些图片没有做过抠图处理
  917. i_n = 0
  918. _name_list = [
  919. "视角{}".format(x) for x in range(1, len(all_original_images) + 1)
  920. ]
  921. # if goods_art_no_folder == "AC51028001":
  922. # _name_list = ["正视", "45度", "侧视", "后视", "底视", "其他1", "其他2", "其他3"]
  923. image_index = 0 # 图片顺序
  924. is_image_deal_mode = 0
  925. max_box = None
  926. for file in all_original_images:
  927. i_n += 1
  928. image_index += 1
  929. original_image_path = "{}/{}/原始图/{}".format(
  930. image_dir, goods_art_no_folder, file
  931. )
  932. file_name = os.path.splitext(file)[0]
  933. """
  934. 当第三张就是为后跟
  935. """
  936. if file_name not in all_800images: # 所有都重新生成
  937. # if goods_art_no_folder != "AC51016112":
  938. # continue
  939. goods_art_no_folder_path = "{}/{}".format(
  940. image_dir, goods_art_no_folder
  941. )
  942. print("正在处理,货号:{}".format(goods_art_no_folder_path))
  943. # self.show_progress_detail("正在处理,货号:{}".format(file_name))
  944. callback_func("正在处理,货号:{}".format(file_name))
  945. # 该文件在800images下没有时,则进行生成新的抠图
  946. # 检查是否存在已抠图文件,如没有再去抠图
  947. original_move_bg_image_path = "{}/原始图_已抠图/{}{}".format(
  948. goods_art_no_folder_path, file_name, ".png"
  949. )
  950. image_deal_mode = 0 # 默认图片不做镜像处理
  951. if not os.path.exists(original_move_bg_image_path):
  952. # 没有抠图文件,进行抠图生成
  953. # self.show_progress_detail("正在抠图 货号:{}".format(file_name))
  954. callback_func("正在抠图 货号:{}".format(file_name))
  955. remove_pic_ins = RemoveBgALi()
  956. im = remove_pic_ins.get_image_cut(
  957. file_path=original_image_path,
  958. out_file_path=original_move_bg_image_path,
  959. )
  960. if not im:
  961. # self.show_progress_detail("货号图{} 抠图处理失败~".format(file_name))
  962. callback_func("货号图{} 抠图处理失败~".format(file_name))
  963. continue
  964. if image_index == 1:
  965. is_image_deal_mode = 0
  966. if settings.Mode == "鞋类":
  967. goods_class = "鞋"
  968. # 如果图片已存在,则需要通过加载图片判断是否为左右脚
  969. if OnePicDeal().check_shoe_is_right(
  970. image_path=original_move_bg_image_path
  971. ):
  972. image_deal_mode = 1 # 1表示要镜像,0表示不做镜像
  973. is_image_deal_mode = 1
  974. if settings.Mode == "皮具":
  975. # 图片对应的商品类型
  976. # goods_class = self.get_goods_class(goods_art_no_folder, original_image_path)
  977. max_box = (1000, 1200)
  978. # _ = {"AC51028001": "女包",
  979. # "AC51028002": "男包",
  980. # "AC51028003": "皮带",
  981. # "AC51028004": "女包"}
  982. # if goods_class in _:
  983. # goods_class = _[goods_class]
  984. # else:
  985. # goods_class = "女包"
  986. #
  987. # _ = {"女包": (1000, 1200),
  988. # "男包": (1000, 1200),
  989. # "皮带": (1000, 1000), }
  990. # max_box = _[goods_class]
  991. # 获取图片信息非必要程序,用于处理图片模式
  992. date_time_original = self.get_date_time_original(
  993. original_image_path
  994. ) # 获取照片拍照时间
  995. if date_time_original:
  996. # 基于照片的时间,与数据库匹配goods_art_no
  997. self.lock.acquire()
  998. _data = self.dataModeMatchPhoto.get_goods_art_no(date_time_original)
  999. self.lock.release()
  1000. if _data:
  1001. # 能匹配上数据库
  1002. goods_art_no, _image_index, _image_deal_mode = _data
  1003. if _image_index < 10:
  1004. image_index = _image_index
  1005. if _image_deal_mode == 1:
  1006. image_deal_mode = 1
  1007. # print(goods_art_no, image_index, image_deal_mode)
  1008. if file_mirror_mark:
  1009. image_deal_mode = 1
  1010. """进行800image 生成"""
  1011. generate_pic = GeneratePic()
  1012. out_pci_mode = "." + settings.getSysConfigs(
  1013. "basic_configs", "image_out_format", "png"
  1014. )
  1015. if out_pci_mode == ".jpg":
  1016. out_path = "{}/800x800/{}{}".format(
  1017. goods_art_no_folder_path, file_name, ".jpg"
  1018. )
  1019. elif out_pci_mode == ".png":
  1020. out_path = "{}/800x800/{}{}".format(
  1021. goods_art_no_folder_path, file_name, ".png"
  1022. )
  1023. else:
  1024. out_path = "{}/800x800/{}{}".format(
  1025. goods_art_no_folder_path, file_name, out_pci_mode
  1026. )
  1027. out_process_path_1 = "{}/阴影图处理/{}_{}_阴影{}".format(
  1028. goods_art_no_folder_path, file_name, _name_list[i_n - 1], ".png"
  1029. )
  1030. out_process_path_2 = "{}/阴影图处理/{}_{}_抠图{}".format(
  1031. goods_art_no_folder_path, file_name, _name_list[i_n - 1], ".png"
  1032. )
  1033. print("image_index", image_index)
  1034. image_index = 99
  1035. curve_mask = True if "俯视" in image_order["image_view"] else False
  1036. out_pic_size = (
  1037. [1600]
  1038. if settings.getSysConfigs(
  1039. "basic_configs", "main_image_size", [1600]
  1040. )
  1041. == ""
  1042. else settings.getSysConfigs(
  1043. "basic_configs", "main_image_size", [1600]
  1044. )
  1045. ) # 主图大小
  1046. if generate_pic.run(
  1047. image_path=original_image_path,
  1048. cut_image_path=original_move_bg_image_path,
  1049. out_path=out_path,
  1050. image_deal_mode=is_image_deal_mode,
  1051. image_index=image_index,
  1052. out_pic_size=out_pic_size,
  1053. is_logo=True if i_n == 1 else False,
  1054. out_process_path_1=out_process_path_1,
  1055. out_process_path_2=out_process_path_2,
  1056. max_box=max_box,
  1057. logo_path=logo_path,
  1058. curve_mask=curve_mask,
  1059. ):
  1060. # self.show_progress_detail("货号图{} _{} 已完成800*800图片制作~".format(image_index, file_name))
  1061. callback_func(
  1062. "货号图{} _{} 已完成800*800图片制作~".format(
  1063. image_index, file_name
  1064. )
  1065. )
  1066. else:
  1067. # self.show_progress_detail("货号图{} _{} 图片生成处理失败~".format(image_index, file_name))
  1068. callback_func(
  1069. "货号图{} _{} 图片生成处理失败~".format(image_index, file_name)
  1070. )
  1071. # 完成处理的图片进度
  1072. self.lock.acquire()
  1073. # self.set_progress()
  1074. self.lock.release()
  1075. def get_goods_art_no(self, date_time_original):
  1076. time_array = time.strptime(date_time_original, "%Y:%m:%d %H:%M:%S")
  1077. time_array = time.mktime(time_array)
  1078. datetime_obj = datetime.fromtimestamp(time_array)
  1079. session = SqlQuery()
  1080. configModel = CRUD(DeviceConfig)
  1081. result = configModel.read(
  1082. session,
  1083. conditions={"photo_create_time": datetime_obj},
  1084. order_by="id",
  1085. ascending=True,
  1086. )
  1087. session.close()
  1088. if result:
  1089. return result.goods_art_no, result.image_index, result.image_deal_mode
  1090. else:
  1091. return None
  1092. def get_goods_art_no_info(
  1093. self, numbers_list=None, goods_art_list=None, headers=None
  1094. ):
  1095. # 获取商品基础信息,入参为商品的编号
  1096. url = "{domain}/api/backend/goods_client/goods_query".format(
  1097. domain=settings.APP_HOST
  1098. )
  1099. data = {"goods_art_list": goods_art_list}
  1100. data = json.dumps(data)
  1101. _s = requests.session().post(url=url, data=data, headers=headers)
  1102. response_data = _s.json()
  1103. goods_number_data = {}
  1104. # ["", "", "", "", "", "", "", "", "", "", "", ]
  1105. if "data" not in response_data:
  1106. return {}
  1107. for data in response_data["data"]:
  1108. goods_number_data[data["goods_art_no"]] = {}
  1109. goods_number_data[data["goods_art_no"]]["商品货号"] = data[
  1110. "goods_art_no"
  1111. ].upper()
  1112. goods_number_data[data["goods_art_no"]]["款号"] = data[
  1113. "goods_number"
  1114. ].upper()
  1115. goods_number_data[data["goods_art_no"]]["商品面料"] = data["fabric"]
  1116. goods_number_data[data["goods_art_no"]]["商品内里"] = data["lining"]
  1117. goods_number_data[data["goods_art_no"]]["商品鞋底"] = data["sole"]
  1118. goods_number_data[data["goods_art_no"]]["鞋垫"] = data["insole"]
  1119. goods_number_data[data["goods_art_no"]]["颜色名称"] = data["color"]
  1120. return goods_number_data
  1121. def get_data_from_hqt_with_goods_art_no(self, goods_art_no_list):
  1122. _goods_art_no_list = copy.deepcopy(goods_art_no_list)
  1123. _list = []
  1124. # 单次请求数少于20个
  1125. goods_art_no_dict = {}
  1126. while _goods_art_no_list:
  1127. goods_art_no = _goods_art_no_list.pop()
  1128. _list.append(goods_art_no)
  1129. if len(_list) == 20 or len(_goods_art_no_list) == 0:
  1130. online_goods_art_data = self.get_goods_art_no_info(goods_art_list=_list)
  1131. if online_goods_art_data:
  1132. for _goods_art_no in online_goods_art_data:
  1133. goods_art_no_dict[_goods_art_no] = online_goods_art_data[
  1134. _goods_art_no
  1135. ]
  1136. _list = []
  1137. return goods_art_no_dict
  1138. def get_goods_art_no_info(
  1139. self, numbers_list=None, goods_art_list=None, headers=None
  1140. ):
  1141. # 获取商品基础信息,入参为商品的编号
  1142. url = "{domain}/api/backend/goods_client/goods_query".format(
  1143. domain=settings.APP_HOST
  1144. )
  1145. data = {"goods_art_list": goods_art_list}
  1146. data = json.dumps(data)
  1147. _s = requests.session().post(url=url, data=data, headers=headers)
  1148. # _s = self.s.get(url=url, params=params, headers=settings.Headers)
  1149. response_data = _s.json()
  1150. goods_number_data = {}
  1151. # ["", "", "", "", "", "", "", "", "", "", "", ]
  1152. if "data" not in response_data:
  1153. return {}
  1154. for data in response_data["data"]:
  1155. goods_number_data[data["goods_art_no"]] = {}
  1156. goods_number_data[data["goods_art_no"]]["商品货号"] = data[
  1157. "goods_art_no"
  1158. ].upper()
  1159. goods_number_data[data["goods_art_no"]]["款号"] = data[
  1160. "goods_number"
  1161. ].upper()
  1162. goods_number_data[data["goods_art_no"]]["商品面料"] = data["fabric"]
  1163. goods_number_data[data["goods_art_no"]]["商品内里"] = data["lining"]
  1164. goods_number_data[data["goods_art_no"]]["商品鞋底"] = data["sole"]
  1165. goods_number_data[data["goods_art_no"]]["鞋垫"] = data["insole"]
  1166. goods_number_data[data["goods_art_no"]]["颜色名称"] = data["color"]
  1167. return goods_number_data
  1168. def get_data_from_hqt(self, goods_number_list):
  1169. _goods_number_list = copy.deepcopy(goods_number_list)
  1170. _list = []
  1171. # 单次请求数少于20个
  1172. goods_number_dict = {}
  1173. while _goods_number_list:
  1174. goods_art_no = _goods_number_list.pop()
  1175. if "NUM" in goods_art_no:
  1176. goods_art_no = goods_art_no.replace("NUM", "")
  1177. _list.append(goods_art_no)
  1178. if len(_list) == 20 or len(_goods_number_list) == 0:
  1179. online_goods_art_data = self.get_goods_art_no_info(numbers_list=_list)
  1180. if online_goods_art_data:
  1181. for number in online_goods_art_data:
  1182. goods_number_dict["NUM" + number] = online_goods_art_data[
  1183. number
  1184. ]
  1185. _list = []
  1186. return goods_number_dict
  1187. def dealMoveImage(self, image_dir: str, callback_func=None) -> dict:
  1188. if not self.check_path(image_dir=image_dir + "/历史"):
  1189. return {"code": 1, "msg": "文件夹创建失败", "data": {}}
  1190. # 遍历目标文件夹,获取有拍摄信息的图片,并按拍摄时间排序
  1191. files = self.list_dir(image_dir)
  1192. original_photo_list = [] # 原始图片列表
  1193. for file in files:
  1194. # -----图片清洗
  1195. file_path = image_dir + "/" + file
  1196. if os.path.isdir(file_path): # 忽略文件夹
  1197. continue
  1198. file_name, suffix = os.path.splitext(file)
  1199. if suffix not in _Type: # 非图片进行移除
  1200. shutil.move(file_path, image_dir + "/历史/" + file)
  1201. continue
  1202. date_time_original = self.get_date_time_original(
  1203. file_path
  1204. ) # 获取照片拍照时间
  1205. if date_time_original:
  1206. # 基于照片的时间,与数据库匹配goods_art_no
  1207. _data = self.get_goods_art_no(date_time_original)
  1208. if _data:
  1209. # 能匹配上数据库
  1210. goods_art_no, image_index, image_deal_mode = _data
  1211. print(
  1212. "832 与数据库匹配goods_art_no",
  1213. file_name,
  1214. date_time_original,
  1215. goods_art_no,
  1216. )
  1217. original_photo_list.append(
  1218. {
  1219. "file_path": file_path,
  1220. "file": file,
  1221. "date_time_original": date_time_original,
  1222. "goods_art_no": goods_art_no,
  1223. "image_index": image_index,
  1224. "real_goods_art_no": "",
  1225. "real_goods_number": "",
  1226. }
  1227. )
  1228. else:
  1229. # 匹配不上报错
  1230. # self.show_progress_detail("图片:{} 无法对应货号,不做处理".format(file))
  1231. if callback_func:
  1232. callback_func("图片:{} 无对应货号".format(file))
  1233. # shutil.move(photo_dict["file_path"], self.image_dir + "/历史/" + photo_dict["file"])
  1234. continue
  1235. else:
  1236. shutil.move(file_path, image_dir + "/历史/" + file)
  1237. if not original_photo_list:
  1238. return {"code": 1, "msg": "没有任何匹配的图片", "data": {}}
  1239. if settings.PROJECT == "红蜻蜓":
  1240. # 批量请求货号图信息
  1241. goods_art_no_list = [x["goods_art_no"] for x in original_photo_list]
  1242. goods_art_no_list = list(set(goods_art_no_list))
  1243. goods_art_no_list = [x for x in goods_art_no_list if "NUM" not in x]
  1244. if goods_art_no_list:
  1245. goods_art_no_dict = self.get_data_from_hqt_with_goods_art_no(
  1246. goods_art_no_list=goods_art_no_list
  1247. )
  1248. for i in original_photo_list:
  1249. if i["goods_art_no"] in goods_art_no_dict:
  1250. i["real_goods_art_no"] = i["goods_art_no"]
  1251. i["real_goods_number"] = "NUM{}".format(
  1252. goods_art_no_dict[i["goods_art_no"]]["编号"]
  1253. )
  1254. # 批量请求编号对应信息
  1255. goods_number_list = [x["goods_art_no"] for x in original_photo_list]
  1256. goods_number_list = list(set(goods_number_list))
  1257. goods_number_list = [x for x in goods_number_list if "NUM" in x]
  1258. if goods_number_list:
  1259. goods_number_dict = self.get_data_from_hqt(
  1260. goods_number_list=goods_number_list
  1261. )
  1262. for i in original_photo_list:
  1263. if i["goods_art_no"] in goods_number_dict:
  1264. i["real_goods_number"] = i["goods_art_no"]
  1265. i["real_goods_art_no"] = goods_number_dict[i["goods_art_no"]][
  1266. "商品货号"
  1267. ]
  1268. # 排序需要基于拍照的文件序号进行处理
  1269. original_photo_list.sort(
  1270. key=lambda x: "{}-{}-{}".format(
  1271. x["goods_art_no"], x["image_index"], x["file"]
  1272. )
  1273. )
  1274. # print(original_photo_list)
  1275. # 对有拍摄信息的图片进行数据库比对,如有比对上,则移动至货号文件夹,否则移入历史文件夹
  1276. total_num = len(original_photo_list)
  1277. # 当天日期作为文件夹
  1278. seconds = time.time()
  1279. output_path = "output/{f_name}".format(
  1280. f_name=time.strftime("%Y-%m-%d", time.localtime(seconds))
  1281. )
  1282. # 遍历每个匹配好的数据进行处理
  1283. n = 0
  1284. for photo_dict in original_photo_list:
  1285. n += 1
  1286. # 进度条
  1287. goods_art_no = photo_dict["goods_art_no"]
  1288. original_image_path = photo_dict["file_path"]
  1289. # 输出货号文件夹
  1290. if photo_dict["real_goods_art_no"]:
  1291. goods_art_no = "{}@{}".format(
  1292. photo_dict["real_goods_art_no"], photo_dict["real_goods_number"]
  1293. )
  1294. goods_art_no_path = "{output_path}/{goods_art_no}".format(
  1295. output_path=output_path, goods_art_no=goods_art_no
  1296. )
  1297. # 创建货号下的一系列文件夹
  1298. self.create_folder(goods_art_no_path)
  1299. # 重命名并进行移动
  1300. print(
  1301. "开始移动:{} {} 命名为:{}".format(
  1302. goods_art_no, original_image_path, goods_art_no_path
  1303. )
  1304. )
  1305. self.move_images(
  1306. goods_art_no, goods_art_no_path, original_image_path
  1307. ) # 货号、货号文件路径、原始图路径
  1308. time.sleep(0.2)
  1309. # self.progress_sign.emit({"type": "移动原始图片", "progress_bar_value": int(n / total_num * 100)})
  1310. # self.show_progress_detail("货号{} 相关文件夹创建完成,已移动原图~".format(goods_art_no))
  1311. if callback_func:
  1312. callback_func(
  1313. "货号{} 相关文件夹创建完成,已移动原图~".format(goods_art_no)
  1314. )
  1315. print("已完成移动处理")
  1316. if n != 0:
  1317. # if settings.MattingPics:
  1318. # # 检查所有未处理的货号文件夹,查看是否有完成图片加工处理
  1319. # self.deal_images()
  1320. # 自动生成一个货号表
  1321. print("output_path", output_path)
  1322. GenerateGoodsArtNoTable.deal(output_path)
  1323. # 完成处理
  1324. # self.set_state(state_value=2)
  1325. return {"code": 0, "msg": "处理完成", "target_path": output_path, "data": {}}
  1326. def check_path(self, image_dir: str):
  1327. if not os.path.exists(image_dir):
  1328. os.mkdir(image_dir)
  1329. return True
  1330. def get_date_time_original(self, file_path):
  1331. with open(file_path, "rb") as file_data:
  1332. tags = exifread.process_file(file_data)
  1333. if "EXIF DateTimeOriginal" in tags:
  1334. return str(tags["EXIF DateTimeOriginal"])
  1335. else:
  1336. return False
  1337. def create_folder(self, path):
  1338. def check_folder(__path):
  1339. if not os.path.exists(__path):
  1340. os.makedirs(__path)
  1341. return False
  1342. return True
  1343. # 文件夹不存在,创建货号子集文件夹
  1344. if not check_folder(path):
  1345. for name in ["原始图", "原始图_已抠图", "800x800", "200images"]:
  1346. other_path = path + "/" + name
  1347. check_folder(other_path)
  1348. def move_images(self, goods_art_no, goods_art_no_path, old_image_path):
  1349. """
  1350. 步骤:
  1351. 1、移动到原始图
  1352. Args:
  1353. goods_art_no:
  1354. goods_art_no_path:
  1355. old_image_path:
  1356. Returns:
  1357. """
  1358. # 移动到原始图
  1359. file = os.path.split(old_image_path)[1]
  1360. # 扩展名
  1361. e = os.path.splitext(file)[1]
  1362. # 获取图片序列
  1363. self.goods_images_count_dict[goods_art_no] += 1
  1364. # A9999(1).jpg
  1365. new_file_name = "{}({})".format(
  1366. goods_art_no, self.goods_images_count_dict[goods_art_no]
  1367. )
  1368. original_image_path = "{}/原始图/{}{}".format(
  1369. goods_art_no_path, new_file_name, e
  1370. )
  1371. # 移动图片
  1372. shutil.move(old_image_path, original_image_path)
  1373. def pixianRemoveImageBg(
  1374. self, file_path: str, out_file_path: str, callbackek_func=None
  1375. ):
  1376. url = self.dataModeMatchPhoto.get_online_data.uploadImage(local_path=file_path)
  1377. remonveUrl = settings.DOMAIN + "/api/ai_image/main/remove_background"
  1378. param = {"base_image": url}
  1379. post_headers = {
  1380. "Authorization": self.token,
  1381. "Content-Length": "",
  1382. "Content-Type": "application/json",
  1383. "Accept": "application/json",
  1384. }
  1385. result = requests.post(
  1386. remonveUrl, data=json.dumps(param), headers=post_headers
  1387. ).json()
  1388. print(result)
  1389. if "code" in result and result["code"] == 0:
  1390. response = requests.get(result["data"]["image"][0])
  1391. with open(out_file_path, "wb") as file:
  1392. file.write(response.content)
  1393. return result["data"]["image"][0]
  1394. else:
  1395. callbackek_func("精细化抠图处理失败 {}".format(result["message"]))
  1396. return ""
  1397. def list_dir(self, path):
  1398. listdir = os.listdir(path)
  1399. return natsorted(listdir, alg=ns.PATH)