Create utils.py

utils.py

@@ -0,0 +1,161 @@
+"""
+Utility functions for the Attendance Analyzer application
+"""
+
+import cv2
+import numpy as np
+from datetime import datetime, date
+import os
+import logging
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+def setup_logging():
+    """Setup logging configuration"""
+    log_format = '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+    logging.basicConfig(
+        level=logging.INFO,
+        format=log_format,
+        handlers=[
+            logging.FileHandler('attendance_analyzer.log'),
+            logging.StreamHandler()
+        ]
+    )
+
+def validate_image(image):
+    """Validate uploaded image"""
+    if image is None:
+        return False, "No image provided"
+    
+    try:
+        if isinstance(image, np.ndarray):
+            if len(image.shape) != 3:
+                return False, "Invalid image format"
+            return True, "Valid image"
+        else:
+            return True, "Valid image"
+    except Exception as e:
+        return False, f"Image validation error: {str(e)}"
+
+def resize_image(image, max_size=(800, 600)):
+    """Resize image while maintaining aspect ratio"""
+    try:
+        height, width = image.shape[:2]
+        max_width, max_height = max_size
+        
+        if width <= max_width and height <= max_height:
+            return image
+        
+        # Calculate scaling factor
+        scale = min(max_width / width, max_height / height)
+        new_width = int(width * scale)
+        new_height = int(height * scale)
+        
+        resized = cv2.resize(image, (new_width, new_height), interpolation=cv2.INTER_AREA)
+        return resized
+    except Exception as e:
+        logger.error(f"Error resizing image: {e}")
+        return image
+
+def format_datetime(dt=None):
+    """Format datetime for database storage"""
+    if dt is None:
+        dt = datetime.now()
+    return dt.strftime("%Y-%m-%d %H:%M:%S")
+
+def format_date(d=None):
+    """Format date for database storage"""
+    if d is None:
+        d = date.today()
+    return d.strftime("%Y-%m-%d")
+
+def create_directory_if_not_exists(directory_path):
+    """Create directory if it doesn't exist"""
+    try:
+        if not os.path.exists(directory_path):
+            os.makedirs(directory_path)
+            logger.info(f"Created directory: {directory_path}")
+        return True
+    except Exception as e:
+        logger.error(f"Error creating directory {directory_path}: {e}")
+        return False
+
+def cleanup_old_files(directory, days_old=30):
+    """Clean up old files from specified directory"""
+    try:
+        current_time = datetime.now()
+        for filename in os.listdir(directory):
+            file_path = os.path.join(directory, filename)
+            if os.path.isfile(file_path):
+                file_time = datetime.fromtimestamp(os.path.getctime(file_path))
+                if (current_time - file_time).days > days_old:
+                    os.remove(file_path)
+                    logger.info(f"Removed old file: {filename}")
+    except Exception as e:
+        logger.error(f"Error cleaning up old files: {e}")
+
+def validate_name(name):
+    """Validate person name"""
+    if not name or not name.strip():
+        return False, "Name cannot be empty"
+    
+    if len(name.strip()) < 2:
+        return False, "Name must be at least 2 characters long"
+    
+    if len(name.strip()) > 50:
+        return False, "Name cannot exceed 50 characters"
+    
+    # Check for valid characters (letters, spaces, hyphens, apostrophes)
+    import re
+    if not re.match(r"^[a-zA-Z\s\-'\.]+$", name.strip()):
+        return False, "Name contains invalid characters"
+    
+    return True, "Valid name"
+
+def validate_email(email):
+    """Validate email address"""
+    if not email:
+        return True, "Email is optional"
+    
+    import re
+    email_pattern = r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
+    if re.match(email_pattern, email):
+        return True, "Valid email"
+    else:
+        return False, "Invalid email format"
+
+def get_system_info():
+    """Get system information for debugging"""
+    import platform
+    import psutil
+    
+    info = {
+        "platform": platform.platform(),
+        "python_version": platform.python_version(),
+        "cpu_count": psutil.cpu_count(),
+        "memory_total": f"{psutil.virtual_memory().total / (1024**3):.2f} GB",
+        "memory_available": f"{psutil.virtual_memory().available / (1024**3):.2f} GB",
+    }
+    return info
+
+def benchmark_face_detection(image, iterations=5):
+    """Benchmark face detection performance"""
+    import time
+    import face_recognition
+    
+    times = []
+    for _ in range(iterations):
+        start_time = time.time()
+        face_locations = face_recognition.face_locations(image)
+        end_time = time.time()
+        times.append(end_time - start_time)
+    
+    avg_time = sum(times) / len(times)
+    return {
+        "average_time": avg_time,
+        "min_time": min(times),
+        "max_time": max(times),
+        "iterations": iterations
+    }