Support nested inheritance from socialmediascraper.base.Scraper

This commit is contained in:
JustAnotherArchivist
2018-03-02 18:50:32 +01:00
parent 02473876d7
commit d03c82d413

View File

@@ -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()