Feat: Open log files in UTF-8 encoding

[bookmarks_db.git] / check_title.py

diff --git a/check_title.py b/check_title.py
index e8215e8e7815c9a74111d3bbc459d1440b7b72ca..391f5a147ec5882fbb3541604f9b741a3670377b 100755 (executable)
--- a/check_title.py
+++ b/check_title.py
@@ -1,4 +1,4 @@
-#! /usr/bin/env python
+#! /usr/bin/env python3
 """Check and show URLs in the bookmarks database where name != real title
 
 This file is a part of Bookmarks database and Internet robot.
@@ -10,73 +10,84 @@ from bkmk_objects import make_linear, quote_title, unquote_title
 
 
 __author__ = "Oleg Broytman <phd@phdru.name>"
-__copyright__ = "Copyright (C) 2002-2017 PhiloSoft Design"
+__copyright__ = "Copyright (C) 2002-2023 PhiloSoft Design"
 __license__ = "GNU GPL"
 
 
-def run():
-   from getopt import getopt
-   optlist, args = getopt(sys.argv[1:], "s")
+def report_title(href, title, real_title):
+    if real_title is None:
+        real_title = "NO REAL TITLE!!!"
+    s = "%s\n%s\n%s\n" % (href, title, real_title)
+
+    if log_file:
+        log_file.write("%s\n" % s)
+    else:
+        print(s)
 
-   report_stats = 1
 
-   for _opt, _arg in optlist:
-      if _opt == '-s':
-         report_stats = 0
-   try:
-      del _opt, _arg
-   except NameError:
-      pass
+def run():
+    from getopt import getopt
+    optlist, args = getopt(sys.argv[1:], "sl:")
+
+    report_stats = 1
+    global log_file
+    log_filename = None
 
-   if report_stats:
-      print("Broytman check_title, Copyright (C) 2002-2017 PhiloSoft Design")
+    for _opt, _arg in optlist:
+        if _opt == '-s':
+            report_stats = 0
+        if _opt == '-l':
+            log_filename = _arg
+    try:
+        del _opt, _arg
+    except NameError:
+        pass
 
-   if args:
-      sys.stderr.write("check_title: too many arguments\n")
-      sys.stderr.write("Usage: check_title [-s]\n")
-      sys.exit(1)
+    if report_stats:
+        print("Broytman check_title, Copyright (C) 2002-2023 PhiloSoft Design")
 
-   from storage import storage
-   storage = storage()
+    if args:
+        sys.stderr.write("check_title: too many arguments\n")
+        sys.stderr.write("Usage: check_title [-s] [-l logfile]\n")
+        sys.exit(1)
 
-   if report_stats:
-      sys.stdout.write("Loading %s: " % storage.filename)
-      sys.stdout.flush()
+    if log_filename:
+        log_file = open(log_filename, 'wt', encoding='utf-8')
 
-   root_folder = storage.load()
-   make_linear(root_folder)
-   objects = len(root_folder.linear)
+    from storage import storage
+    storage = storage()
 
-   if report_stats:
-      print("Ok")
+    if report_stats:
+        sys.stdout.write("Loading %s: " % storage.filename)
+        sys.stdout.flush()
 
+    root_folder = storage.load()
+    make_linear(root_folder)
+    objects = len(root_folder.linear)
 
-   for object_no in range(objects):
-      object = root_folder.linear[object_no]
+    if report_stats:
+        print("Ok")
 
-      if object.isBookmark:
-         if hasattr(object, "moved") or hasattr(object, "error") \
-               or object.href.startswith('place:'): # Firefox SmartBookmarks
-            continue
+    for object_no in range(objects):
+        object = root_folder.linear[object_no]
 
-         if hasattr(object, "real_title") and (object.real_title is not None):
-            unquoted_title = unquote_title(quote_title(object.real_title))
-            unquoted_name = unquote_title(object.name)
-            if unquoted_name != unquoted_title:
-               print(object.href)
-               print(unquoted_name)
-               print(unquoted_title)
-               print()
-         else:
-            print(object.href)
-            print(object.name)
-            print("NO REAL TITLE!!!")
-            print()
+        if object.isBookmark:
+            if hasattr(object, "moved") or hasattr(object, "error") or \
+                    object.href.startswith('place:'):  # Firefox SmartBookmarks
+                continue
 
+            if hasattr(object, "real_title") \
+                    and (object.real_title is not None):
+                unquoted_title = unquote_title(quote_title(object.real_title))
+                unquoted_name = unquote_title(object.name)
+                if unquoted_name != unquoted_title:
+                    report_title(object.href, unquoted_name, unquoted_title)
+            else:
+                report_title(object.href, object.name, None)
 
-   if report_stats:
-      print(objects, "objects passed")
+    if report_stats:
+        print(objects, "objects passed")
 
 
 if __name__ == '__main__':
-   run()
+    run()