import time
from concurrent.futures import as_completed, ThreadPoolExecutor, wait
import threading
from .remove_bg_pixian import RemoveBgPiXian
from .other.module_online_data import GetOnlineData
from .deal_one_image import DealOneImage, DealOneImageBeforehand
from .other.log import MyLogger


class DealCutout:

    def __init__(self):
        super().__init__()
        self.lock = threading.Lock()
        self.need_cutout_images = {}
        self.state = 2  # 1进行中 2停止
        self.get_online_data = GetOnlineData()
        self.is_upload_pic_num = 0
        self.is_deal_num = 0
        # 图片列表
        self.upload_pic_dict = {}
        self.logger = MyLogger().logger

    def startDispose(self):
        self.get_online_data.refresh_headers()
        num = 0
        for image_data in self.need_cutout_images:
            num += 1
            upload_pic_dict = {}
            upload_pic_dict = DealOneImageBeforehand(
                image_data=image_data, lock=self.lock, windows=self, num=num
            ).run(upload_pic_dict)
            print("upload_pic_dict", upload_pic_dict)
            DealOneImage(
                image_data=image_data, lock=self.lock, windows=self, num=num
            ).run(image_data, upload_pic_dict)

class DealCloths:

    def __init__(self):
        super().__init__()
        self.lock = threading.Lock()
        self.need_cutout_images = {}
        self.state = 2  # 1进行中 2停止
        self.get_online_data = GetOnlineData()
        self.is_upload_pic_num = 0
        self.is_deal_num = 0
        # 图片列表
        self.upload_pic_dict = {}
        self.logger = MyLogger().logger

    def startDispose(self):
        self.get_online_data.refresh_headers()
        num = 0
        for image_data in self.need_cutout_images:
            num += 1
            upload_pic_dict = {}
            hand = DealOneImageBeforehand(
                image_data=image_data, lock=self.lock, windows=self, num=num
            )
            upload_pic_dict = hand.get_image_cut_cloths()