CameraMachine/python/mcu/capture/capture_basic_mode.py

import time
import settings
from sys import platform as sys_platform
from utils.utils_func import httpGetHandler
from PIL import Image
from io import BytesIO
import requests

if settings.IS_TEST:
    import piexif
    from PIL import Image
    import threading


class CaptureBasic():
    # program_sign = Signal(dict)0
    base_host = "http://localhost:5513"
    _get_image_index = 0
    _get_folder_index = 0
    is_first_init = 0

    def __init__(self):
        # self.window = window
        # 状态 1 进行中 2 已停止
        self.state = 2
        # 总拍照数量
        self.total_num = 0
        # 是否获取照片
        self.is_get_photo = False
        # self.keyboard = Controller()
        self.hwnd = None
    def getPreviewImage(self)->Image:
        '''获取预览图'''
        yzmdata = requests.get(self.base_host + "/preview.jpg")
        tempIm = BytesIO(yzmdata.content)
        im = Image.open(tempIm)
        return im

    def getCaptureFolderPath(self) -> Image:
        """获取预览图"""
        response = httpGetHandler(
            url=self.base_host,
            params={"slc": "get", "param1": "session.folder"},
            headers={},
        )
        return response.text

    def run_capture_action(self,command):
        '''执行capture动作'''
        response = httpGetHandler(url=self.base_host, params={"CMD":command},headers={})
        return response
    # def photograph(self):
    #     """
    #     Args:
    #         data:
    #         info_show: 信息输出
    #         image_deal_mode:0 表示左脚，1表示右脚
    #     Returns:
    #     """
    #     print("142----------------------------photograph")
    #     # 执行拍照
    #     if not self.hwnd:
    #         return False, {"flag": "{} 未打开".format(settings.CaptureSoftwareName)}
    #     if self.state == 1:
    #         print("当前有未完成的任务")
    #         return False, {"flag": "当前有未完成的任务"}
    #     print("开始拍照")
    #     self.state = 1
    #     if not self.set_foreground_window():
    #         return False, {"flag": "no {}".format(settings.CaptureSoftwareName)}
    #     time.sleep(0.1)
    #     # 自动对焦
    #     self.auto_focus()
    #     # self.send_win32_key(self.hwnd, key_1=win32con.VK_CONTROL, key_2=win32con.VK_F2)

    #     if settings.IS_TEST:
    #         # print("56-----测试拍照")
    #         threading.Thread(target=self.test_add_photo, args=()).start()
    #     # self.program_sign.emit({"type": "add_to_photo_todo_list",
    #     #                         "data": data})
    #     time.sleep(float(settings.PhotographSeconds))

    # def test_add_photo(self):
    #     def copy_file(srcfile, dstpath, dstfile):  # 复制函数
    #         if not os.path.isfile(srcfile):
    #             # print("%s not exist!" % (srcfile))
    #             return
    #         else:
    #             if int(os.path.getsize(srcfile) / 1024) < 10:
    #                 return
    #             # fpath, fname = os.path.split(srcfile)  # 分离文件名和路径
    #             if not os.path.exists(dstpath):
    #                 os.makedirs(dstpath)  # 创建路径
    #             shutil.copy(srcfile, dstfile)  # 复制文件

    #     print("模拟拍照  test_add_photo")
    #     time.sleep(0.9)
    #     s = time.time()
    #     source_path = r"{}\PhotoImageDir\test_source".format(os.getcwd())
    #     check_path(source_path)

    #     # 整理所有测试源照片
    #     images_path_list = []
    #     folder_list = get_folder(source_path)
    #     for folder_dict in folder_list:
    #         folder_path = folder_dict["folder_path"]
    #         _pics = [x["file_path"] for x in get_images(folder_path)]
    #         if _pics:
    #             images_path_list.append(_pics)

    #     if len(images_path_list) <= CaptureBasic._get_folder_index:
    #         CaptureBasic._get_folder_index = 0
    #         CaptureBasic._get_image_index = 0
    #     srcfile = images_path_list[CaptureBasic._get_folder_index][CaptureBasic._get_image_index]

    #     # 新文件夹
    #     CaptureBasic._get_image_index += 1
    #     if len(images_path_list[CaptureBasic._get_folder_index]) <= CaptureBasic._get_image_index:
    #         CaptureBasic._get_folder_index += 1
    #         CaptureBasic._get_image_index = 0

    #     # print("pic:", srcfile)
    #     e = os.path.splitext(os.path.split(srcfile)[1])[1]

    #     # 修改图片时间
    #     if e != ".CR2":
    #         _time = time.strftime("%Y:%m:%d %H:%M:%S", time.localtime(time.time()))
    #         exif_ifd = {piexif.ExifIFD.DateTimeOriginal: _time}
    #         exif_dict = {"Exif": exif_ifd}
    #         exif_bytes = piexif.dump(exif_dict)
    #         im = Image.open(srcfile)
    #         im.save(srcfile, exif=exif_bytes)

    #     dst_path = settings.PhotoOutputDir
    #     # dst_path = "{}/data".format(os.getcwd())
    #     e = os.path.splitext(srcfile)[-1]
    #     new_name = "~{}{}".format(int(time.time() * 10), e)
    #     dst_file = "{}/{}".format(dst_path, new_name)
    #     print("测试数据   dst_file", dst_file)
    #     copy_file(srcfile, dst_path, dst_file)
    #     # print("修改图片耗时：{}".format(time.time() - s))

    def auto_focus(self):
        # 执行拍照
        time.sleep(0.2)
        self.run_capture_action("LiveView_Focus")

    def last_images(self):
        if not self.hwnd:
            return
        # win32gui.SetForegroundWindow(self.hwnd)
        # self.keyboard.press(Key.ctrl)
        # self.keyboard.press(Key.left)
        # self.keyboard.release(Key.left)
        # self.keyboard.release(Key.ctrl)

    def next_images(self):
        # 执行拍照
        if not self.hwnd:
            return
        # win32gui.SetForegroundWindow(self.hwnd)
        # self.keyboard.press(Key.ctrl)
        # self.keyboard.press(Key.right)
        # self.keyboard.release(Key.right)
        # self.keyboard.release(Key.ctrl)

    def test_1(self):
        if not self.hwnd:
            return
        # 执行照片全选与导出
        # win32gui.SetForegroundWindow(self.hwnd)
        # self.keyboard.press(Key.ctrl)
        # self.keyboard.press("a")
        # self.keyboard.release("a")
        # self.keyboard.release(Key.ctrl)
        # time.sleep(1)
        # # 照片导出
        # self.keyboard.press(Key.ctrl)
        # self.keyboard.press("d")
        # self.keyboard.release("d")
        # self.keyboard.release(Key.ctrl)
if __name__ == "__main__":
    cap = CaptureBasic()
    cap.auto_focus()
    print("cccc")