Feat: Log reports to files

[bookmarks_db.git] / check_redirects.py

diff --git a/check_redirects.py b/check_redirects.py
index 12c74d8e59ed9b8e4fdc06ef4baab39257a328cd..1f5c2b01137a53b05b725e204995a1996656bb41 100755 (executable)
--- a/check_redirects.py
+++ b/check_redirects.py
@@ -14,15 +14,26 @@ __copyright__ = "Copyright (C) 2023 PhiloSoft Design"
 __license__ = "GNU GPL"
 
 
+def report_redirect(href, moved):
+    if log_file:
+        log_file.write("%s\n%s\n\n" % (href, moved))
+    else:
+        print("URL: %s\nMoved to: %s\n" % (href, moved))
+
+
 def run():
     from getopt import getopt
-    optlist, args = getopt(sys.argv[1:], "s")
+    optlist, args = getopt(sys.argv[1:], "sl:")
 
     report_stats = 1
+    global log_file
+    log_filename = None
 
     for _opt, _arg in optlist:
         if _opt == '-s':
             report_stats = 0
+        if _opt == '-l':
+            log_filename = _arg
     try:
         del _opt, _arg
     except NameError:
@@ -33,9 +44,12 @@ def run():
 
     if args:
         sys.stderr.write("check_redirects: too many arguments\n")
-        sys.stderr.write("Usage: check_redirects [-s]\n")
+        sys.stderr.write("Usage: check_redirects [-s] [-l logfile]\n")
         sys.exit(1)
 
+    if log_filename:
+        log_file = open(log_filename, 'wt')
+
     from storage import storage
     storage = storage()
 
@@ -59,9 +73,7 @@ def run():
                 continue
 
             if hasattr(object, "moved"):
-                print(object.href)
-                print(object.moved)
-                print()
+                report_redirect(object.href, object.moved)
                 del object.moved
 
     if report_stats: