From b0c3dab64e59a3442d3cbcfaec46945e5b3573b8 Mon Sep 17 00:00:00 2001 From: Thomas Oettli Date: Fri, 30 Apr 2021 14:07:30 +0200 Subject: [PATCH] add get command to cli --- pyquarantine/cli.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/pyquarantine/cli.py b/pyquarantine/cli.py index 4f0fa10..cba3cb7 100644 --- a/pyquarantine/cli.py +++ b/pyquarantine/cli.py @@ -286,6 +286,13 @@ def delete(quarantines, args): logger.info("quarantined email deleted successfully") +def get(quarantines, args): + storage = _get_storage(quarantines, args.quarantine) + fp, _ = storage.get_mail(args.quarantine_id) + print(fp.read().decode()) + fp.close() + + class StdErrFilter(logging.Filter): def filter(self, rec): return rec.levelno in (logging.ERROR, logging.WARNING) @@ -452,6 +459,17 @@ def main(): help="Delete email for all recipients.", action="store_true") quarantine_delete_parser.set_defaults(func=delete) + # quarantine get command + quarantine_get_parser = quarantine_subparsers.add_parser( + "get", + description="Get email from quarantine.", + help="Get email from quarantine", + formatter_class=formatter_class) + quarantine_get_parser.add_argument( + "quarantine_id", + metavar="ID", + help="Quarantine ID.") + quarantine_get_parser.set_defaults(func=get) # whitelist command group whitelist_parser = subparsers.add_parser(