Add strict set of events to monitor

DMOJ · Apr 21, 2023 · 17b0bcf · 17b0bcf
1 parent 582f444
commit 17b0bcf
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/dmoj/monitor.py b/dmoj/monitor.py
@@ -10,7 +10,13 @@
 
 try:
     from watchdog.observers import Observer
-    from watchdog.events import FileSystemEventHandler
+    from watchdog.events import (
+        FileSystemEventHandler,
+        EVENT_TYPE_MOVED,
+        EVENT_TYPE_DELETED,
+        EVENT_TYPE_MODIFIED,
+        EVENT_TYPE_CREATED,
+    )
 
     has_watchdog_installed = True
 except ImportError:
@@ -52,11 +58,20 @@ def run(self):
 
 
 class SendProblemsHandler(FileSystemEventHandler):
+    ALLOWED_EVENT_TYPES = (
+        EVENT_TYPE_MOVED,
+        EVENT_TYPE_DELETED,
+        EVENT_TYPE_MODIFIED,
+        EVENT_TYPE_CREATED,
+    )
+
     def __init__(self, refresher=None):
         self.refresher = refresher
         self.callback = None
 
     def on_any_event(self, event):
+        if event.event_type not in self.ALLOWED_EVENT_TYPES:
+            return
         if self.callback is not None:
             self.callback()
         if self.refresher is not None: