Fixup script structure for entrypoints

2018-03-26 03:35:03 -04:00
parent 30062b3c2b
commit e50d3907ae
6 changed files with 124 additions and 108 deletions
@@ -32,19 +32,23 @@ import sys

 import dkim

-if sys.version_info[0] >= 3:
-    # Make sys.stdin a binary stream.
-    sys.stdin = sys.stdin.detach()
+def main():
+    if sys.version_info[0] >= 3:
+        # Make sys.stdin a binary stream.
+        sys.stdin = sys.stdin.detach()

-message = sys.stdin.read()
-verbose = '-v' in sys.argv
-if verbose:
-  logging.basicConfig(level=10)
-  a = dkim.ARC(message)
-  cv, results, comment = a.verify()
-else:
-  cv, results, comment = dkim.arc_verify(message)
+    message = sys.stdin.read()
+    verbose = '-v' in sys.argv
+    if verbose:
+        logging.basicConfig(level=10)
+        a = dkim.ARC(message)
+        cv, results, comment = a.verify()
+    else:
+        cv, results, comment = dkim.arc_verify(message)

-print("arc verification: cv=%s %s" % (cv, comment))
-if verbose:
-  print(repr(results))
+    print("arc verification: cv=%s %s" % (cv, comment))
+    if verbose:
+        print(repr(results))
+
+if __name__ == "__main__":
+    main()