file2mail.py

@@ -8,57 +8,68 @@ import mail
 import os
 import psutil
 import report
-import signal
 import time
 
-CYCLE_TIME = 60
+LOCK_FILE = os.path.join(config.TARGET_FOLDER, "lock")
-SLEEP_TIME = .5
+PID = os.getpid()
 
 logger = logging.getLogger('root')
 report.stdoutLoggingConfigure(log_name_lvl=[('root', config.LOG_LEVEL), ])
 
 
-class GracefulKiller:
+def has_handle(fpath):
-  kill_now = False
+    for proc in psutil.process_iter():
-  def __init__(self):
-    signal.signal(signal.SIGINT, self.exit_gracefully)
-    signal.signal(signal.SIGTERM, self.exit_gracefully)
-
-  def exit_gracefully(self, signum, frame):
-    self.kill_now = True
-
-
-def file2mail():
-    # Scan for files to send
-    for filename in fstools.filelist(config.TARGET_FOLDER):
-        logger.info("Found %s to send via E-Mail to %s.", filename, config.SEND_TO)
-        # Send File as E-Mail
         try:
-            mail.send_mail(
+            for item in proc.open_files():
-                config.SEND_FROM,
+                if fpath == item.path:
-                config.SEND_TO,
+                    return True
-                "Gesanntes Dokument - %s" % os.path.basename(filename),
-                "Hier ist das gescannte Dokument vom Scanner in Buchen:",
-                files=[filename],
-                server=config.SMTP_SERVER,
-                username=config.SMTP_USER,
-                password=config.SMTP_PASSWORD
-            )
         except Exception:
-            logger.exception("Exception while sending an E-Mail")
+            pass
+
+    return False
+
+
+if __name__ == "__main__":
+    while True:
+        # Check if LOCK_FILE EXISTS
+        if not os.path.exists(LOCK_FILE):
+            # Create LOCK_FILE
+            with open(LOCK_FILE, 'w') as fh:
+                fh.write(str(PID))
+            logger.debug("Creating lock-file %s with PID %d", LOCK_FILE, PID)
+            # Read LOCK_FILE content
+            with open(LOCK_FILE, 'r') as fh:
+                lf_pid = json.loads(fh.read())
+            # Check if LOCK_FILE has my PID
+            if lf_pid == PID:
+                # Scan for files to send
+                for filename in fstools.filelist(config.TARGET_FOLDER):
+                    # Exclude LOCK_FILE
+                    if filename != LOCK_FILE and not has_handle(filename):
+                        logger.info("Found %s to send via E-Mail to %s.", filename, config.SEND_TO)
+                        # Send File as E-Mail
+                        try:
+                            mail.send_mail(
+                                config.SEND_FROM,
+                                config.SEND_TO,
+                                "Gesanntes Dokument - %s" % os.path.basename(filename),
+                                "Hier ist das gescannte Dokument vom Scanner in Buchen:",
+                                files=[filename],
+                                server=config.SMTP_SERVER,
+                                username=config.SMTP_USER,
+                                password=config.SMTP_PASSWORD
+                            )
+                        except Exception:
+                            logger.exception("Exception while sending an E-Mail")
+                        else:
+                            # Remove file
+                            logger.debug("Removing file %s", filename)
+                            os.remove(filename)
+                # Remove LOCK_FILE
+                logger.debug("Removing lock-file %s", LOCK_FILE)
+                os.remove(LOCK_FILE)
+            else:
+                logger.warning("LOCK_FILE has PID %d and this proces %d. No action!", lf_pid, PID)
         else:
-            # Remove file
+            logger.warning("LOCK_FILE %s already exists. No action!", LOCK_FILE)
-            logger.debug("Removing file %s", filename)
+        time.sleep(60)
-            os.remove(filename)
-
-
-if __name__ == '__main__':
-  killer = GracefulKiller()
-  i = 0
-  while not killer.kill_now:
-    time.sleep(.5)
-    if i >= CYCLE_TIME / SLEEP_TIME:
-        file2mail()
-        i = 0
-    else:
-        i += 1