No functional changes.  This is a refactoring commit to provide some
non-global scoping and easier readability.

Signed-off-by: Daniel Kahn Gillmor <d...@fifthhorseman.net>
---
 email-print-mime-structure | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/email-print-mime-structure b/email-print-mime-structure
index b78ae91..5eb94e0 100755
--- a/email-print-mime-structure
+++ b/email-print-mime-structure
@@ -80,10 +80,14 @@ class MimePrinter(object):
         else:
             self.print_part(z, prefix+'─╴')
 
-msg:Union[Message, str, int, Any] = email.message_from_file(sys.stdin)
+def main() -> None:
+    msg:Union[Message, str, int, Any] = email.message_from_file(sys.stdin)
 
-if isinstance(msg, Message):
-    printer:MimePrinter = MimePrinter()
-    printer.test(msg, '└')
-else:
-    logging.error('Input was not an e-mail message')
+    if isinstance(msg, Message):
+        printer:MimePrinter = MimePrinter()
+        printer.test(msg, '└')
+    else:
+        logging.error('Input was not an e-mail message')
+
+if __name__ == '__main__':
+    main()
-- 
2.24.0.rc1

Reply via email to