summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDaniel Kahn Gillmor <dkg@fifthhorseman.net>2019-11-02 01:28:18 -0400
committerSean Whitton <spwhitton@spwhitton.name>2019-11-02 08:44:19 -0700
commita9a3a085c2f407f1dac144eba58087a7ebcf4e35 (patch)
treeac436e4dadc574017fceaf362aeaa3e51a8633d4
parent75dbd9eb55cae90d6c962e9eb914ffa05d05d69e (diff)
downloadmailscripts-a9a3a085c2f407f1dac144eba58087a7ebcf4e35.tar.gz
email-print-mime-structure: put main() into its own function
No functional changes. This is a refactoring commit to provide some non-global scoping and easier readability. Signed-off-by: Daniel Kahn Gillmor <dkg@fifthhorseman.net>
-rwxr-xr-xemail-print-mime-structure16
1 files 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()