Kumi
e82ccb2701
- Extended copyright to reflect the current year. - Incremented project version to indicate new features and fixes. - Added a new script entry for easier execution, increasing utility and accessibility. - Updated project URLs for better alignment with current infrastructure. - Refactored settings path for simplicity and consistency across deployments. - Improved code readability and maintenance across several modules by cleaning up redundant code, adding missing type annotations, and ensuring consistent code formatting. - Enhanced logging capabilities and error handling to improve diagnostics and troubleshooting, supporting more robust error recovery mechanisms. - Implemented more graceful handling of termination signals to ensure clean shutdown and resource cleanup, enhancing the robustness of the application in production environments. - Introduced command-line argument parsing for configuration file path customization, improving flexibility in different runtime environments. These changes collectively improve the project's maintainability, reliability, and user experience, laying a stronger foundation for future development.
92 lines
2.9 KiB
Python
92 lines
2.9 KiB
Python
from watchdog.events import (
|
|
FileSystemEventHandler,
|
|
FileSystemEvent,
|
|
FileCreatedEvent,
|
|
FileDeletedEvent,
|
|
FileModifiedEvent,
|
|
FileMovedEvent,
|
|
)
|
|
|
|
from multiprocessing import Queue
|
|
|
|
from .logger import Logger
|
|
|
|
import os.path
|
|
import time
|
|
|
|
|
|
class DogHandler(FileSystemEventHandler):
|
|
"""Class implementing a watchdog event handler"""
|
|
|
|
def __init__(self, directory, queue: Queue, *args, **kwargs) -> None:
|
|
"""Initialize a new DogHandler object
|
|
|
|
Args:
|
|
directory (classes.directory.Directory): Directory to watch
|
|
queue (multiprocessing.Queue): Queue to put detected events on
|
|
"""
|
|
super().__init__(*args, **kwargs)
|
|
self._directory = directory
|
|
self._queue = queue
|
|
self._logger = Logger()
|
|
|
|
def dispatch(self, event: FileSystemEvent):
|
|
"""Dispatch events to the appropriate event handlers
|
|
|
|
Args:
|
|
event (watchdog.events.FileSystemEvent): Event to handle
|
|
"""
|
|
if not event.is_directory:
|
|
super().dispatch(event)
|
|
|
|
def on_created(self, event: FileCreatedEvent):
|
|
"""Put file creation events on the queue
|
|
|
|
Args:
|
|
event (watchdog.events.FileCreatedEvent): Event describing the
|
|
created file
|
|
"""
|
|
self._logger.debug(f"Detected creation event of {event.src_path}")
|
|
|
|
size = os.path.getsize(event.src_path)
|
|
time.sleep(5)
|
|
if size == os.path.getsize(event.src_path):
|
|
self._queue.put((self._directory, os.path.basename(event.src_path)))
|
|
|
|
def on_modified(self, event: FileModifiedEvent):
|
|
"""Put file modification events on the queue
|
|
|
|
Args:
|
|
event (watchdog.events.FileModifiedEvent): Event describing the
|
|
modified file
|
|
"""
|
|
self._logger.debug(f"Detected modification event of {event.src_path}")
|
|
|
|
size = os.path.getsize(event.src_path)
|
|
time.sleep(5)
|
|
if size == os.path.getsize(event.src_path):
|
|
self._queue.put((self._directory, os.path.basename(event.src_path)))
|
|
|
|
def on_moved(self, event: FileMovedEvent):
|
|
"""Put file move events on the queue
|
|
|
|
Args:
|
|
event (watchdog.events.FileMovedEvent): Event describing the moved
|
|
file (source and destination)
|
|
"""
|
|
self._logger.debug(
|
|
f"Detected move event of {event.src_path} to {event.dest_path}"
|
|
)
|
|
self._queue.put((self._directory, os.path.basename(event.src_path)))
|
|
self._queue.put((self._directory, os.path.basename(event.dest_path)))
|
|
|
|
def on_deleted(self, event: FileDeletedEvent):
|
|
"""Put file deletion events on the queue
|
|
|
|
Args:
|
|
event (watchdog.events.FileDeletedEvent): Event describing the
|
|
deleted file
|
|
"""
|
|
self._logger.debug(f"Detected deletion event of {event.src_path}")
|
|
self._queue.put((self._directory, os.path.basename(event.src_path)))
|