From 0dfad9f4af1a40febb954bcb2eed642d229d2c8a Mon Sep 17 00:00:00 2001 From: Michael Tremer Date: Sat, 6 Dec 2025 18:10:13 +0000 Subject: [PATCH] dnsbl: Allow to pass the output format Signed-off-by: Michael Tremer --- src/dnsbl/lists.py | 11 ++++++++--- src/scripts/dnsbl.in | 4 +++- 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/src/dnsbl/lists.py b/src/dnsbl/lists.py index 21b4a2e..023fb40 100644 --- a/src/dnsbl/lists.py +++ b/src/dnsbl/lists.py @@ -200,9 +200,14 @@ class List(sqlmodel.SQLModel, database.BackendMixin, table=True): # Export! - def export(self, f): + def export(self, f, format): """ Exports the list """ - for domain in self.domains: - f.write("%s\n" % domain) + # Write the domains as they are + if format == "domains": + for domain in self.domains: + f.write("%s\n" % domain) + + else: + raise ValueError("Unknown output format: %s" % format) diff --git a/src/scripts/dnsbl.in b/src/scripts/dnsbl.in index 8ecb609..57f76e7 100644 --- a/src/scripts/dnsbl.in +++ b/src/scripts/dnsbl.in @@ -96,6 +96,8 @@ class CLI(object): # export export = subparsers.add_parser("export", help=_("Exports a list")) export.add_argument("list", help=_("The name of the list")) + export.add_argument("--format", default="domains", choices=("domains",), + help=_("Output Format")) export.set_defaults(func=self.__export) # add-source @@ -255,7 +257,7 @@ class CLI(object): list = backend.lists.get_by_slug(args.list) # Export! - list.export(sys.stdout) + list.export(sys.stdout, format=args.format) def __add_source(self, backend, args): """ -- 2.47.3