Merge pull request #22 from keshav2212/master

Added main() function to genicons.py

Author: TimidRobot

scripts/genicons.py

@@ -8,6 +8,8 @@
 import os
 import os.path
 from functools import reduce
+import sys
+import traceback
 
 # Third-party
 import cairo
@@ -132,59 +134,76 @@ def genicon(
     return ctx
 
 
-font_map = pangocairo.font_map_get_default()
-font_families = [family.get_name() for family in font_map.list_families()]
-if "CC Icons" not in font_families:
-    raise Exception(
-        "CC Icons font not installed. See" " <https://wiki.debian.org/Fonts>."
-    )
+def main():
+    font_map = pangocairo.font_map_get_default()
+    font_families = [family.get_name() for family in font_map.list_families()]
+    if "CC Icons" not in font_families:
+        raise Exception(
+            "CC Icons font not installed. See"
+            " <https://wiki.debian.org/Fonts>."
+        )
 
-script_dir = os.path.dirname(__file__)
-basedir = os.path.realpath(
-    os.path.abspath(os.path.join(script_dir, "..", "www", "i"))
-)
-print("# basedir:", basedir)
-
-for suite, licenses in SUITES.items():
-    for lic, module_chars in licenses.items():
-        for chars, dimensions, background, foreground in product(
-            module_chars, DIMENSIONS, BACKGROUNDS, FOREGROUNDS
-        ):
-            # e.g. white on white
-            if foreground == background:
-                continue
-            path = os.path.realpath(
-                os.path.abspath(
-                    os.path.join(
-                        basedir, icon_path(suite, lic, background, foreground),
+    script_dir = os.path.dirname(__file__)
+    basedir = os.path.realpath(
+        os.path.abspath(os.path.join(script_dir, "..", "www", "i"))
+    )
+    print("# basedir:", basedir)
+
+    for suite, licenses in SUITES.items():
+        for lic, module_chars in licenses.items():
+            for chars, dimensions, background, foreground in product(
+                module_chars, DIMENSIONS, BACKGROUNDS, FOREGROUNDS
+            ):
+                # e.g. white on white
+                if foreground == background:
+                    continue
+                path = os.path.realpath(
+                    os.path.abspath(
+                        os.path.join(
+                            basedir,
+                            icon_path(suite, lic, background, foreground),
+                        )
+                    )
+                )
+                filepath = os.path.realpath(
+                    os.path.abspath(
+                        os.path.join(path, icon_filename(dimensions, chars))
                     )
                 )
-            )
-            filepath = os.path.realpath(
-                os.path.abspath(
-                    os.path.join(path, icon_filename(dimensions, chars))
+                if os.path.exists(filepath):
+                    continue
+                width = dimensions[0]
+                height = dimensions[1]
+                font_size = dimensions[2]
+                padding = dimensions[3]
+                ctx = genicon(
+                    suite,
+                    chars,
+                    font_size,
+                    padding,
+                    width,
+                    height,
+                    background,
+                    foreground,
                 )
-            )
-            if os.path.exists(filepath):
-                continue
-            width = dimensions[0]
-            height = dimensions[1]
-            font_size = dimensions[2]
-            padding = dimensions[3]
-            ctx = genicon(
-                suite,
-                chars,
-                font_size,
-                padding,
-                width,
-                height,
-                background,
-                foreground,
-            )
-            try:
-                os.makedirs(path)
-            except OSError as e:
-                if e.errno != errno.EEXIST:
-                    raise
-            # Will raise and exception on error
-            ctx.get_target().write_to_png(filepath)
+                try:
+                    os.makedirs(path)
+                except OSError as e:
+                    if e.errno != errno.EEXIST:
+                        raise
+                # Will raise and exception on error
+                ctx.get_target().write_to_png(filepath)
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except SystemExit as e:
+        sys.exit(e.code)
+    except KeyboardInterrupt:
+        print("INFO (130) Halted via KeyboardInterrupt.", file=sys.stderr)
+        sys.exit(130)
+    except Exception:
+        print("ERROR (1) Unhandled exception:", file=sys.stderr)
+        print(traceback.print_exc(), file=sys.stderr)
+        sys.exit(1)