diff --git a/cli.py b/cli.py index eb84739..3994211 100644 --- a/cli.py +++ b/cli.py @@ -1,4 +1,4 @@ -from src.logic.cli import avail_check, search_all +from src.logic.search import avail_check, search_all, search_requested import os import argparse from src.logic.utils import move, tag_folder, rename, detect_chapters @@ -15,9 +15,10 @@ def grabber(args): os.system(f"rm -rf {cfg.komgrabber.download_location}") os.mkdir(cfg.komgrabber.download_location) if nyaa is True and komga is True: - search_all() - if args.scan: - scan_komga() + if args.request: + search_requested() + search_all(args.library, args.all) + if cfg.komgrabber.aria2.kill_after_completion: # kill aria2c os.system("killall aria2c") @@ -52,14 +53,30 @@ def main(): subparsers = parser.add_subparsers(dest="command", required=True) # tag subcommand - tag_parser = subparsers.add_parser("search", help="Run search operation") + tag_parser = subparsers.add_parser("search", help="Run search operation. After the search is completed, the library will be scanned to detect new or updated series.") + tag_parser.add_argument( + "library", + type=str, + help="Library to search in (e.g. 'manga', 'anime', leave empty for all)", + ) + tag_parser.add_argument( "-v", "--verbose", action="store_true", help="Enable verbose output" ) + tag_parser.add_argument( - "--scan", + "-a", + "--all", action="store_true", - help="Scan the library after downloading", + default=False, + help="Search for all series in the database", + ) + tag_parser.add_argument( + "-r", + "--request", + action="store_true", + default=False, + help="Search for the requested series in the database", ) tag_parser.set_defaults(func=grabber)