From a9a3a085c2f407f1dac144eba58087a7ebcf4e35 Mon Sep 17 00:00:00 2001 From: Daniel Kahn Gillmor Date: Sat, 2 Nov 2019 01:28:18 -0400 Subject: 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 --- email-print-mime-structure | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) (limited to 'email-print-mime-structure') 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() -- cgit v1.2.3