propusk/module/ImageTool.py

from PIL import Image, UnidentifiedImageError
import os
from logger import logger
from datetime import datetime
import cv2


def rotate_image(path_file: str, gradus: int = -90) -> str:
    try:
        image = Image.open(path_file, 'r')
        rotate_path_image = os.path.join(
            os.path.dirname(image.filename),
            F"rotate_{os.path.split(image.filename)[1]}"
        )
    
        rotate_img = image.rotate(gradus, expand=True)
        rotate_img.save(rotate_path_image)
    
        return rotate_path_image
    
    except FileNotFoundError:
        logger.error(F"Файл не найден: {path_file=}")
        return None
    except UnidentifiedImageError:
        logger.error(F"Не правильный формат: {path_file=}")
        return None



def cupture_face(path_photo: str, new_path: str) -> None:
    try:
        img = cv2.imread(path_photo)
        face_recog = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_alt_tree.xml')
        face_result = face_recog.detectMultiScale(img, scaleFactor=2, minNeighbors=3)
        if len(face_result) != 0:
            logger.info(F"Find face: {face_result}")
            x,y,w,h = face_result[-1]        
            img = img[y-50:y+h+80,x-50:x+50+w]
            cv2.imwrite(new_path, img)
    except cv2.error as err:
        logger.error(err)