improve job class selection code

+ consistent argument order for add_argument() calls
This commit is contained in:
Mike Fährmann
2019-05-10 22:05:57 +02:00
parent e64773ffdd
commit bd9cb3d191
2 changed files with 65 additions and 63 deletions

View File

@@ -204,14 +204,8 @@ def main():
if args.list_urls: if args.list_urls:
jobtype = job.UrlJob jobtype = job.UrlJob
jobtype.maxdepth = args.list_urls jobtype.maxdepth = args.list_urls
elif args.list_keywords:
jobtype = job.KeywordJob
elif args.list_data:
jobtype = job.DataJob
elif args.simulate:
jobtype = job.SimulationJob
else: else:
jobtype = job.DownloadJob jobtype = args.jobtype or job.DownloadJob
urls = args.urls urls = args.urls
if args.inputfile: if args.inputfile:

View File

@@ -11,7 +11,7 @@
import argparse import argparse
import logging import logging
import json import json
from .version import __version__ from . import job, version
class ConfigAction(argparse.Action): class ConfigAction(argparse.Action):
@@ -70,27 +70,27 @@ def build_parser():
) )
general.add_argument( general.add_argument(
"--version", "--version",
action="version", version=__version__, action="version", version=version.__version__,
help="Print program version and exit", help="Print program version and exit",
) )
general.add_argument( general.add_argument(
"-d", "--dest", "-d", "--dest",
metavar="DEST", action=ConfigAction, dest="base-directory", dest="base-directory", metavar="DEST", action=ConfigAction,
help="Destination directory", help="Destination directory",
) )
general.add_argument( general.add_argument(
"-i", "--input-file", "-i", "--input-file",
metavar="FILE", dest="inputfile", dest="inputfile", metavar="FILE",
help="Download URLs found in FILE ('-' for stdin)", help="Download URLs found in FILE ('-' for stdin)",
) )
general.add_argument( general.add_argument(
"--cookies", "--cookies",
metavar="FILE", action=ConfigAction, dest="cookies", dest="cookies", metavar="FILE", action=ConfigAction,
help="File to load additional cookies from", help="File to load additional cookies from",
) )
general.add_argument( general.add_argument(
"--proxy", "--proxy",
metavar="URL", action=ConfigAction, dest="proxy", dest="proxy", metavar="URL", action=ConfigAction,
help="Use the specified proxy", help="Use the specified proxy",
) )
general.add_argument( general.add_argument(
@@ -101,49 +101,57 @@ def build_parser():
output = parser.add_argument_group("Output Options") output = parser.add_argument_group("Output Options")
output.add_argument( output.add_argument(
"-q", "--quiet", dest="loglevel", action="store_const", "-q", "--quiet",
const=logging.ERROR, default=logging.INFO, dest="loglevel", default=logging.INFO,
action="store_const", const=logging.ERROR,
help="Activate quiet mode", help="Activate quiet mode",
) )
output.add_argument( output.add_argument(
"-v", "--verbose", dest="loglevel", action="store_const", "-v", "--verbose",
const=logging.DEBUG, default=logging.INFO, dest="loglevel",
action="store_const", const=logging.DEBUG,
help="Print various debugging information", help="Print various debugging information",
) )
output.add_argument( output.add_argument(
"-g", "--get-urls", dest="list_urls", action="count", "-g", "--get-urls",
dest="list_urls", action="count",
help="Print URLs instead of downloading", help="Print URLs instead of downloading",
) )
output.add_argument( output.add_argument(
"-j", "--dump-json", dest="list_data", action="store_true", "-j", "--dump-json",
dest="jobtype", action="store_const", const=job.DataJob,
help="Print JSON information", help="Print JSON information",
) )
output.add_argument( output.add_argument(
"-s", "--simulate", dest="simulate", action="store_true", "-s", "--simulate",
dest="jobtype", action="store_const", const=job.SimulationJob,
help="Simulate data extraction; do not download anything", help="Simulate data extraction; do not download anything",
) )
output.add_argument( output.add_argument(
"-K", "--list-keywords", dest="list_keywords", action="store_true", "-K", "--list-keywords",
dest="jobtype", action="store_const", const=job.KeywordJob,
help=("Print a list of available keywords and example values " help=("Print a list of available keywords and example values "
"for the given URLs"), "for the given URLs"),
) )
output.add_argument( output.add_argument(
"--list-modules", dest="list_modules", action="store_true", "--list-modules",
dest="list_modules", action="store_true",
help="Print a list of available extractor modules", help="Print a list of available extractor modules",
) )
output.add_argument( output.add_argument(
"--list-extractors", dest="list_extractors", action="store_true", "--list-extractors",
dest="list_extractors", action="store_true",
help=("Print a list of extractor classes " help=("Print a list of extractor classes "
"with description, (sub)category and example URL"), "with description, (sub)category and example URL"),
) )
output.add_argument( output.add_argument(
"--write-log", "--write-log",
metavar="FILE", dest="logfile", action=ConfigAction, dest="logfile", metavar="FILE", action=ConfigAction,
help="Write logging output to FILE", help="Write logging output to FILE",
) )
output.add_argument( output.add_argument(
"--write-unsupported", "--write-unsupported",
metavar="FILE", dest="unsupportedfile", action=ConfigAction, dest="unsupportedfile", metavar="FILE", action=ConfigAction,
help=("Write URLs, which get emitted by other extractors but cannot " help=("Write URLs, which get emitted by other extractors but cannot "
"be handled, to FILE"), "be handled, to FILE"),
) )
@@ -151,37 +159,37 @@ def build_parser():
downloader = parser.add_argument_group("Downloader Options") downloader = parser.add_argument_group("Downloader Options")
downloader.add_argument( downloader.add_argument(
"-r", "--limit-rate", "-r", "--limit-rate",
metavar="RATE", action=ConfigAction, dest="rate", dest="rate", metavar="RATE", action=ConfigAction,
help="Maximum download rate (e.g. 500k or 2.5M)", help="Maximum download rate (e.g. 500k or 2.5M)",
) )
downloader.add_argument( downloader.add_argument(
"-R", "--retries", "-R", "--retries",
metavar="RETRIES", action=ConfigAction, dest="retries", type=int, dest="retries", metavar="RETRIES", type=int, action=ConfigAction,
help="Number of retries (default: 5)", help="Number of retries (default: 5)",
) )
downloader.add_argument( downloader.add_argument(
"--http-timeout", "--http-timeout",
metavar="SECONDS", action=ConfigAction, dest="timeout", type=float, dest="timeout", metavar="SECONDS", type=float, action=ConfigAction,
help="Timeout for HTTP connections (defaut: 30.0)", help="Timeout for HTTP connections (defaut: 30.0)",
) )
downloader.add_argument( downloader.add_argument(
"--sleep", "--sleep",
metavar="SECONDS", action=ConfigAction, dest="sleep", type=float, dest="sleep", metavar="SECONDS", type=float, action=ConfigAction,
help="Number of seconds to sleep before each download", help="Number of seconds to sleep before each download",
) )
downloader.add_argument( downloader.add_argument(
"--no-part", "--no-part",
action=ConfigConstAction, nargs=0, dest="part", const=False, dest="part", nargs=0, action=ConfigConstAction, const=False,
help="Do not use .part files", help="Do not use .part files",
) )
downloader.add_argument( downloader.add_argument(
"--no-check-certificate", "--no-check-certificate",
action=ConfigConstAction, nargs=0, dest="verify", const=False, dest="verify", nargs=0, action=ConfigConstAction, const=False,
help="Disable HTTPS certificate validation", help="Disable HTTPS certificate validation",
) )
downloader.add_argument( downloader.add_argument(
"--abort-on-skip", "--abort-on-skip",
action=ConfigConstAction, nargs=0, dest="skip", const="abort", dest="skip", nargs=0, action=ConfigConstAction, const="abort",
help=("Abort extractor run if a file download would normally be " help=("Abort extractor run if a file download would normally be "
"skipped, i.e. if a file with the same filename already exists"), "skipped, i.e. if a file with the same filename already exists"),
) )
@@ -189,64 +197,64 @@ def build_parser():
configuration = parser.add_argument_group("Configuration Options") configuration = parser.add_argument_group("Configuration Options")
configuration.add_argument( configuration.add_argument(
"-c", "--config", "-c", "--config",
metavar="CFG", dest="cfgfiles", action="append", dest="cfgfiles", metavar="FILE", action="append",
help="Additional configuration files", help="Additional configuration files",
) )
configuration.add_argument(
"--config-yaml",
dest="yamlfiles", metavar="FILE", action="append",
help=argparse.SUPPRESS,
)
configuration.add_argument( configuration.add_argument(
"-o", "--option", "-o", "--option",
metavar="OPT", action=ParseAction, dest="options", default=[], dest="options", metavar="OPT", action=ParseAction, default=[],
help="Additional '<key>=<value>' option values", help="Additional '<key>=<value>' option values",
) )
configuration.add_argument( configuration.add_argument(
"--ignore-config", dest="load_config", action="store_false", "--ignore-config",
dest="load_config", action="store_false",
help="Do not read the default configuration files", help="Do not read the default configuration files",
) )
configuration.add_argument(
"--config-yaml",
metavar="CFG", dest="yamlfiles", action="append",
help=argparse.SUPPRESS,
)
authentication = parser.add_argument_group("Authentication Options") authentication = parser.add_argument_group("Authentication Options")
authentication.add_argument( authentication.add_argument(
"-u", "--username", "-u", "--username",
metavar="USER", action=ConfigAction, dest="username", dest="username", metavar="USER", action=ConfigAction,
help="A username to login with" help="Username to login with",
) )
authentication.add_argument( authentication.add_argument(
"-p", "--password", "-p", "--password",
metavar="PASS", action=ConfigAction, dest="password", dest="password", metavar="PASS", action=ConfigAction,
help="The password associated with the given username", help="Password belonging to the given username",
) )
authentication.add_argument( authentication.add_argument(
"--netrc", "--netrc",
action=ConfigConstAction, nargs=0, dest="netrc", const=True, dest="netrc", nargs=0, action=ConfigConstAction, const=True,
help="Enable .netrc authentication data", help="Enable .netrc authentication data",
) )
selection = parser.add_argument_group("Selection Options") selection = parser.add_argument_group("Selection Options")
selection.add_argument( selection.add_argument(
"--download-archive", "--download-archive",
metavar="FILE", dest="archive", action=ConfigAction, dest="archive", metavar="FILE", action=ConfigAction,
help=("Record all downloaded files in the archive file and " help=("Record all downloaded files in the archive file and "
"skip downloading any file already in it.") "skip downloading any file already in it."),
) )
selection.add_argument( selection.add_argument(
"--range", "--range",
metavar="RANGE", dest="image-range", action=ConfigAction, dest="image-range", metavar="RANGE", action=ConfigAction,
help=("Index-range specifying which images to download. " help=("Index-range(s) specifying which images to download. "
"For example '--range -2,4,6-8,10-' will download images with " "For example '5-10' or '1,3-5,10-'"),
"index 1, 2, 4, 6, 7, 8 and 10 up to the last one"),
) )
selection.add_argument( selection.add_argument(
"--chapter-range", "--chapter-range",
metavar="RANGE", dest="chapter-range", action=ConfigAction, dest="chapter-range", metavar="RANGE", action=ConfigAction,
help=("Like '--range', but applies to manga-chapters " help=("Like '--range', but applies to manga-chapters "
"and other delegated URLs"), "and other delegated URLs"),
) )
selection.add_argument( selection.add_argument(
"--filter", "--filter",
metavar="EXPR", dest="image-filter", action=ConfigAction, dest="image-filter", metavar="EXPR", action=ConfigAction,
help=("Python expression controlling which images to download. " help=("Python expression controlling which images to download. "
"Files for which the expression evaluates to False are ignored. " "Files for which the expression evaluates to False are ignored. "
"Available keys are the filename-specific ones listed by '-K'. " "Available keys are the filename-specific ones listed by '-K'. "
@@ -255,7 +263,7 @@ def build_parser():
) )
selection.add_argument( selection.add_argument(
"--chapter-filter", "--chapter-filter",
metavar="EXPR", dest="chapter-filter", action=ConfigAction, dest="chapter-filter", metavar="EXPR", action=ConfigAction,
help=("Like '--filter', but applies to manga-chapters " help=("Like '--filter', but applies to manga-chapters "
"and other delegated URLs"), "and other delegated URLs"),
) )
@@ -263,33 +271,33 @@ def build_parser():
postprocessor = parser.add_argument_group("Post-processing Options") postprocessor = parser.add_argument_group("Post-processing Options")
postprocessor.add_argument( postprocessor.add_argument(
"--zip", "--zip",
action="append_const", dest="postprocessors",
dest="postprocessors", const={"name": "zip"}, action="append_const", const={"name": "zip"},
help="Store downloaded files in a ZIP archive", help="Store downloaded files in a ZIP archive",
) )
postprocessor.add_argument( postprocessor.add_argument(
"--ugoira-conv", "--ugoira-conv",
action="append_const", dest="postprocessors",
dest="postprocessors", const={"name": "ugoira", "ffmpeg-args": ( action="append_const", const={"name": "ugoira", "ffmpeg-args": (
"-c:v", "libvpx", "-crf", "4", "-b:v", "5000k", "-an")}, "-c:v", "libvpx", "-crf", "4", "-b:v", "5000k", "-an")},
help="Convert Pixiv Ugoira to WebM (requires FFmpeg)", help="Convert Pixiv Ugoira to WebM (requires FFmpeg)",
) )
postprocessor.add_argument( postprocessor.add_argument(
"--write-metadata", "--write-metadata",
action="append_const", dest="postprocessors",
dest="postprocessors", const={"name": "metadata"}, action="append_const", const={"name": "metadata"},
help="Write metadata to separate JSON files", help="Write metadata to separate JSON files",
) )
postprocessor.add_argument( postprocessor.add_argument(
"--write-tags", "--write-tags",
action="append_const", dest="postprocessors",
dest="postprocessors", const={"name": "metadata", "mode": "tags"}, action="append_const", const={"name": "metadata", "mode": "tags"},
help="Write image tags to separate text files", help="Write image tags to separate text files",
) )
parser.add_argument( parser.add_argument(
"urls", "urls",
nargs="*", metavar="URL", metavar="URL", nargs="*",
help=argparse.SUPPRESS, help=argparse.SUPPRESS,
) )