Exemple #1
0
def _job_quality(job):
    from libyo.youtube.resolve.profiles import profiles
    profile=job.getnosect("profile",list(profiles.keys())[0]).lower()
    if profile not in profiles:
        print("[ERROR] The Profile \"{0}\" is unknown".format(profile))
        return False
    if job.getnosect("resolution",job.getnosect("quality",None)):
        try:
            resolution = int(job.getnosect("resolution",job.getnosect("quality")).lower().rstrip("p"))
        except ValueError:
            print("[ERROR] Resolution has to be in this format: \"%i\" or \"%ip\" ('360' or '360p')")
            return False
        raw_map = profiles[profile][0]
        if resolution not in raw_map:
            print("[ WARN] The Exact Resolution given ({1}) is not avaiable in this Profile: \"{0}\"".format(profile,resolution))
        l= [ v for k,v in raw_map.items() if k <= resolution ]
        return l
    else:
        return list(profiles[profile][0].values())
Exemple #2
0
def main(argv):
    """
    The main entrypoint.
    
    This only manages commandline arguments and db intialization
    It will dispatch control to [subcommand]_command(args)
    """
    # print welcome message and check versions
    welcome()
    
    #---------------------------------------------
    # parse the commandline arguments
    #---------------------------------------------
    choice_profile = choice.cichoice(profiles.keys())
    choice_quality = choice.qchoice.new(1080, 720, 480, 360, 240)
    
    parser = argparse.ArgumentParser(prog=argv[0])
    parser.add_argument("-db", dest="database", help="The Database to use", default=database)
    parser.add_argument("-V", dest="command_", action="store_const", const="version",
        help="Display version and quit")
    
    subparsers = parser.add_subparsers(dest="command",
        help="Use '%(prog)s (command) -h' to get further help. By Default, the 'run' command is run.",
        title="command", default="run")
    
    # config subcommand
    config_parser = subparsers.add_parser("config", description="Modify application options")
    config_parser.add_argument("mode", choices=("get", "set", "list"),
        help="What you want to do; available options are %(choices)s")
    config_parser.add_argument("key", help="The Key to modify/show (set, get)", nargs="?")
    config_parser.add_argument("value", help="The Value to set (set)", nargs="?")
    
    # job management
    job_parser = subparsers.add_parser("job", description="Manage youfeed jobs")
    job_subparsers = job_parser.add_subparsers(dest="mode",
        help="Job modification modes")
    
    job_add = job_subparsers.add_parser("add", description="Add new job")
    job_add.add_argument("name", help="An unique Identifier for this job")
    job_add.add_argument("type", help="The job type. choices: %(choices)s", choices=("playlist", "favorites"))
    job_add.add_argument("resource", help="The job resource. [playlist -> playlist ID]")
    job_add.add_argument("-target", help="The name of the resulting playlist")
    job_add.add_argument("-profile", help="The job codec profile", choices=choice_profile)
    job_add.add_argument("-quality", help="The job maximum quality", choices=choice_quality)
    job_add.add_argument("-export", help="Export the playlist file")
    job_add.add_argument("-disable", action="store_true", help="Disable the new job")
    job_add.add_argument("-noidcheck", action="store_true", help="Disable Playlist ID check")
    
    job_rm = job_subparsers.add_parser("rm", description="Remove a job")
    job_rm.add_argument("name", help="The job identifier")
    
    job_dis = job_subparsers.add_parser("disable", description="Enable/Disable Job")
    job_dis.add_argument("name", help="The job identifier")
    job_dis.add_argument("disabled", help="disabled flag value. choices: %(choices)s", choices=("yes", "no"))
    
    job_mod = job_subparsers.add_parser("change", description="Modify a job")
    job_mod.add_argument("name", help="The job identifier")
    job_mod.add_argument("-name", dest="newname", help="Change the identifier")
    job_mod.add_argument("-type", help="Change the job type", choices=("playlist",))
    job_mod.add_argument("-resource", help="Change the job resource")
    job_mod.add_argument("-target", help="Change the playlist name")
    job_mod.add_argument("-profile", help="Change the codec profile", choices=choice_profile)
    job_mod.add_argument("-quality", help="Change the maximum quality", choices=choice_quality)
    job_mod.add_argument("-export", help="Change the export location")
    job_mod.add_argument("-noidcheck", action="store_true", help="Disable Playlist ID check")
    
    job_list = job_subparsers.add_parser("list", description="List jobs")
    job_list.add_argument("-showall", help="Don't hide system jobs", action="store_true")
    
    job_show = job_subparsers.add_parser("show", description="Display a job")
    job_show.add_argument("name", help="The job identifier")
    
    # run subcommand
    run_parser = subparsers.add_parser("run",
        description="run job(s). if no jobs are given, run all enabled jobs")
    run_parser.add_argument("names", help="The job identifier(s)", nargs="*")
    run_parser.add_argument("-p", help="Only recreate the playlist file(s)", action="store_true")
    run_parser.add_argument("-d", help="Only check for new videos, don't download anything", action="store_true")
    run_parser.add_argument("-forceall", help="run disabled jobs, too", action="store_true")
    
    # db subcommand
    db_parser = subparsers.add_parser("db", description="Manage the YouFeed Database")
    db_parsers = db_parser.add_subparsers(dest="mode")
    
    db_import2 = db_parsers.add_parser("v2import", description="Import v2 videos")
    db_import2.add_argument("pldir", help="youfeed v2 pl/ directory")
    db_import2.add_argument("-move", help="move contained videos to new schema", action="store_true")
    #db_import2.add_argument
    
    # export subcommand
    export_parser = subparsers.add_parser("export", description="export videos")
    export_parser.add_argument("playlist_id", help="what to export")
    export_parser.add_argument("-profile", choices=choice_profile, help="profile")
    export_parser.add_argument("-quality", choices=choice_quality, help="quality")
    export_parser.add_argument("-startat", type=int, help="first element index", default=1)
    
    export_parsers = export_parser.add_subparsers(dest="mode", help="where to export to")
    
    export_folder = export_parsers.add_parser("folder", description="copy videos to folder")
    export_folder.add_argument("location", help="the folder location")
    export_folder.add_argument("filename_template", help="how to name the video files: %n=index %t=title %e=extension %i=videoId %p=playlistTitle %x=playlistId %u=uploader", default="%n._%t.%e", nargs="?")
    
    # do the parsing
    args = parser.parse_args(argv[1:])
    
    #---------------------------------------------
    # database initialization
    #---------------------------------------------
    if not os.path.exists(args.database):
        db = yfdb.DB.open(args.database)
        db.setOptionValue("playlists_folder", playlists_folder)
        db.setOptionValue("videos_folder", videos_folder)
    else:
        db = yfdb.DB.open(args.database)
        
    # pass db around with args
    args.db     = db
    args.root   = os.path.dirname(os.path.abspath(args.database))
    
    #---------------------------------------------
    # Dispatcher
    #---------------------------------------------
    if args.command_:
        if args.command_ == "version":
            print("YouFeed version: %s" % version)
            print("libyo version:   %s" % libyo.version)
            print("DB version:      %i" % yfdb.DB_VERSION)
            return 0
    elif args.command == "config":
        return config_command(args)
    elif args.command == "job":
        return job_command(args)
    elif args.command == "run":
        return run_command(args)
    elif args.command == "db":
        return db_command(args)
    elif args.command == "export":
        return export_command(args)