From d03c82d413e37a063e91437ad0d5f62239d9cc82 Mon Sep 17 00:00:00 2001 From: JustAnotherArchivist Date: Fri, 2 Mar 2018 18:50:32 +0100 Subject: [PATCH] Support nested inheritance from socialmediascraper.base.Scraper --- socialmediascraper/cli.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/socialmediascraper/cli.py b/socialmediascraper/cli.py index ad92199..c11e13c 100644 --- a/socialmediascraper/cli.py +++ b/socialmediascraper/cli.py @@ -15,10 +15,12 @@ def parse_args(): parser.add_argument('-n', '--max-results', dest = 'maxResults', type = int, metavar = 'N', help = 'Only return the first N results') subparsers = parser.add_subparsers(dest = 'scraper', help = 'The scraper you want to use') - for cls in socialmediascraper.base.Scraper.__subclasses__(): + classes = socialmediascraper.base.Scraper.__subclasses__() + for cls in classes: subparser = subparsers.add_parser(cls.name, formatter_class = argparse.ArgumentDefaultsHelpFormatter) cls.setup_parser(subparser) subparser.set_defaults(cls = cls) + classes.extend(cls.__subclasses__()) args = parser.parse_args()