예제 #1
0
파일: models.py 프로젝트: cpkurotori/CHECK
 def __init__(self, name, categoryId, id=None, testing=False, **kwargs):
     self.name = name
     self.categoryId = categoryId
     if id is None:
         id = str(id_gen())
     self.id = id
     if testing:
         self.testing = testing
예제 #2
0
파일: models.py 프로젝트: cpkurotori/CHECK
 def __init__(self, name, attributes, id=None, testing=False, **kwargs):
     self.name = name
     self.attributes = attributes
     if id is None:
         id = str(id_gen())
     self.id = id
     if testing:
         self.testing = testing
예제 #3
0
파일: models.py 프로젝트: cpkurotori/CHECK
 def __init__(self, name, id=None, items=None, testing=False, **kwargs):
     self.name = name
     if id is None:
         id = str(id_gen())
     self.id = id
     if items is None:
         items = []
     self.items = items
     if testing:
         self.testing = testing
예제 #4
0
파일: models.py 프로젝트: cpkurotori/CHECK
 def __init__(self, itemId, specs, id=None, testing=False, **kwargs):
     self.itemId = itemId
     if id is None:
         id = str(id_gen())
     self.id = id
     self.specs = specs
     if testing:
         self.testing = testing
     item = getItem(itemId)
     category = getCategory(item.getCategoryId())
     if not specs:
         for attr in category.getAttributes():
             self.specs[attr] = None
예제 #5
0
파일: models.py 프로젝트: cpkurotori/CHECK
 def setSpec(self,
             attr,
             value,
             override=False,
             id=str(id_gen()),
             testing=False,
             **kwargs):
     if attr not in getCategory(getItem(
             self.itemId).getCategoryId()).getAttributes():
         raise InvalidAttributeError
     elif self.specs[attr] is not None and not override:
         raise OverrideError
     else:
         self.specs[attr] = value
     updateChecklistItem(self)
예제 #6
0
파일: models.py 프로젝트: cpkurotori/CHECK
 def __init__(self,
              name,
              id=None,
              login=None,
              pin=None,
              checklists=None,
              admin=False,
              testing=False,
              **kwargs):
     self.name = name
     if checklists is None:
         checklists = {}
     self.checklists = checklists
     if id is None:
         id = str(id_gen())
     self.id = id
     self.pin = pin
     self.admin = admin
     if testing:
         self.testing = testing
예제 #7
0
파일: cli.py 프로젝트: evertonstz/pynps
def cli_main(maindir=""):
    """implement pynps cli interface"""
    runing_from = get_pyinstaller()
    if get_system() == 'Linux':
        base_folder = os.getenv('HOME')
        CONFIGFOLDER = f"{base_folder}/.config/pyNPS"
        config_file = f"{CONFIGFOLDER}/settings.ini"
    elif get_system() == 'Windows':
        if runing_from == 'pi-onefile':
            # one file will have all the related directories in the same folder as the .exe
            base_folder = maindir
            CONFIGFOLDER = f"{base_folder}/pynps_config/"
            config_file = f"{CONFIGFOLDER}/settings.ini"
        elif runing_from in ['python', 'pi-onefolder']:
            # python and one-fodler will have related directories inside Documents
            # python won't come undled with wget.exe and pkg2zip.exe!
            base_folder = os.path.expanduser("~")
            if runing_from == 'python':
                CONFIGFOLDER = f"{base_folder}/Documents/pyNPS"
            else:
                CONFIGFOLDER = f"{base_folder}/Documents/pyNPSbin"
            config_file = f"{CONFIGFOLDER}/settings.ini"

    # create conf file
    if os.path.isfile(config_file) == False:
        create_config(config_file, CONFIGFOLDER, base_folder)
        if CONFIGFOLDER != "":
            if runing_from != 'pi-onefolder':
                create_folder(CONFIGFOLDER + "/lib/")

    # read conf file
    config = configparser.ConfigParser()
    config.read(config_file)

    # test sections
    if sorted(config.sections()) != sorted([
            'pyNPS', 'PSV_Links', 'PSP_Links', 'PSX_Links', 'PSM_Links',
            'PS3_Links', 'BinaryLocations'
    ]):
        rich.print(
            "Error: config file is missing sections: you need the following sections in your config file: 'PSV_Links', 'PSP_Links', 'PSX_Links', 'PSM_Links', 'PS3_Links', 'BinaryLocations'",
            style='red on black')
        sys.exit(1)
    if sorted(list(config["PSV_Links"])) != sorted(
        ['games', 'dlcs', 'themes', 'updates', 'demos']):
        rich.print(
            "Error: config file is missing options in the PSV_Links section: you need the following options in your PSV_Links section: 'games', 'dlcs', 'themes', 'updates', 'demos'",
            style='red on black')
        sys.exit(1)
    if sorted(list(config["PSP_Links"])) != sorted(
        ['games', 'dlcs', 'themes', 'updates']):
        rich.print(
            "Error: config file is missing options in the PSP_Links section: you need the following options in your PSP_Links section: 'games', 'dlcs', 'themes', 'updates'",
            style='red on black')
        sys.exit(1)
    if sorted(list(config["PSX_Links"])) != sorted(['games']):
        rich.print(
            "Error: config file is missing options in the PSX_Links section: you need the following options in your PSX_Links section: 'games'",
            style='red on black')
        sys.exit(1)
    if sorted(list(config["PSM_Links"])) != sorted(['games']):
        rich.print(
            "Error: config file is missing options in the PSM_Links section: you need the following options in your PSM_Links section: 'games'",
            style='red on black')
        sys.exit(1)
    if sorted(list(config["PS3_Links"])) != sorted(
        ['games', 'dlcs', 'themes', 'demos', 'avatars']):
        rich.print(
            "Error: config file is missing options in the PS3_Links section: you need the following options in your PS3_Links section: 'games', 'dlcs', 'themes', 'demos', 'avatars'",
            style='red on black')
        sys.exit(1)

    # making vars
    DBFOLDER = fix_folder_syntax(config['pyNPS']['databasefolder'], maindir)
    DLFOLDER = fix_folder_syntax(config['pyNPS']['downloadfolder'], maindir)
    PKG2ZIP = fix_folder_syntax(config['BinaryLocations']['pkg2zip_location'],
                                maindir)
    WGET = fix_folder_syntax(config['BinaryLocations']['wget_location'],
                             maindir)

    # tests existence of pkg2zip
    PKG2ZIP = check_pkg2zip(PKG2ZIP, CONFIGFOLDER)
    if PKG2ZIP == False:
        rich.print(
            "You don't have a valid pkg2zip installation or binary in your system, extraction will be skipped. PS3 games don't need pkg2zip",
            style='dark_orange')

    # tests existence of wget
    WGET = check_wget(WGET, CONFIGFOLDER)
    if WGET == False:
        rich.print(
            "Error: you don't have a valid wget installation or binary in your system, this program can't work without it",
            style='red on black')
        sys.exit(1)

    # makin dicts for links
    database_psv_links = {}
    for key in config["PSV_Links"]:
        database_psv_links[key] = config["PSV_Links"][key]

    database_psp_links = {}
    for key in config["PSP_Links"]:
        database_psp_links[key] = config["PSP_Links"][key]

    database_psx_links = {}
    for key in config["PSX_Links"]:
        database_psx_links[key] = config["PSX_Links"][key]

    database_psm_links = {}
    for key in config["PSM_Links"]:
        database_psm_links[key] = config["PSM_Links"][key]

    database_ps3_links = {}
    for key in config["PS3_Links"]:
        database_ps3_links[key] = config["PS3_Links"][key]

    # create args
    args, parser = create_args()

    limit_rate = args.limit_rate
    keepkg = args.keepkg
    system = args.console
    cso_factor = args.compress_cso

    if args.resume_session:
        # in this case args.search will be considered a tag to fast resume a session

        input_tag = args.search
        if input_tag is not None:
            if input_tag.isalnum() is False:
                rich.print("Tags can only be alphanumeric without spaces",
                           style='dark_orange')
                input_tag = None

        ##load download db
        with SqliteDict(f"{DBFOLDER}/downloads.db",
                        autocommit=False) as database:
            try:
                db = database['resumes']
                if len(db) == 0:
                    raise
            except:
                rich.print("There are no saved download sessions to resume",
                           style='dark_orange')
                sys.exit(0)

        checker = next(
            (item for item in db if item['session_tag'] == input_tag), None)

        ## if tag is prvided and there's a match in the db
        if input_tag is not None and checker is not None:
            # has something in db
            session = checker

        elif checker is None or input_tag is None:
            if input_tag is not None:
                rich.print("There's no such tag in the download sessions",
                           style='dark_orange')

                # validation resume input
                class Check_resume_input_y_n(Validator):
                    def validate(self, document):
                        text = document.text
                        if len(text) > 0:
                            if text.lower() not in ['y', 'n']:
                                # break
                                raise ValidationError(
                                    message='Use "y" for yes and "n" for no',
                                    cursor_position=0)
                        else:
                            raise ValidationError(
                                message=
                                'Enter something or press Ctrl+C to close.',
                                cursor_position=0)

                try:
                    yn_check = prompt(
                        "Wanna see all currently download sessions? [y/n]: ",
                        validator=Check_resume_input_y_n())
                    if yn_check.lower() != "y":
                        raise
                except KeyboardInterrupt:
                    rich.print('Interrupted by user', style='bright_black')
                    sys.exit(0)
                except:
                    rich.print('Interrupted by user', style='bright_black')
                    sys.exit(0)

            p_db = []
            for index_file, i in enumerate(db):
                i["Index"] = index_file + 1
                p_db.append(i)

            process_resumes(p_db)

            # validating input
            class Check_resume_input(Validator):
                def validate(self, document):
                    text = document.text
                    if len(text) > 0:
                        if text.isdigit():

                            # test if number being typed is bigger than the biggest number from game list
                            num_p = len(db)
                            if num_p == 1 and text != "1":
                                raise ValidationError(
                                    message="Your only option is to type 1.",
                                    cursor_position=0)
                            if int(text) > num_p:
                                raise ValidationError(
                                    message=
                                    f"There are no entries past {num_p}.")
                        else:
                            raise ValidationError(
                                message="Please only use numbers",
                                cursor_position=0)
                    else:
                        raise ValidationError(
                            message=
                            'Enter something or press Ctrl+C to close. Press "h" for help.',
                            cursor_position=0)

            try:
                session_index = prompt(
                    "Enter the number for the download session you want to resume: ",
                    validator=Check_resume_input())
            except KeyboardInterrupt:
                rich.print('Interrupted by user', style='bright_black')
                sys.exit(0)
            except:
                rich.print('Interrupted by user', style='bright_black')
                sys.exit(0)

            session = db[int(session_index) - 1]

        files_to_download = session["session_dict"]

    else:

        what_to_dl = {
            "games": args.games,
            "dlcs": args.dlcs,
            "themes": args.themes,
            "updates": args.updates,
            "demos": args.demos,
            "avatars": args.avatars
        }

        if set(what_to_dl.values()) == set([False]):
            for i in what_to_dl:
                what_to_dl[i] = True

        if args.update == True:
            if [args.region, args.search, args.eboot, args.compress_cso
                ] != [None, None, False, None]:
                # TODO let the user search while updating the database
                rich.print(
                    "Error: you can't search while updating the database",
                    style='red on black')
                sys.exit(1)

            what_to_up = [x for x in what_to_dl if what_to_dl[x] == True]

            for i in system:
                fillterm(
                    f"[green on black][UPDATEDB][/green on black][green] Updating {variables.FULL_SYSTEM_NAME[i]} database"
                )
                if i == "PSV":
                    db = database_psv_links
                elif i == "PSP":
                    db = database_psp_links
                elif i == "PSX":
                    db = database_psx_links
                elif i == "PSM":
                    db = database_psm_links
                elif i == "PS3":
                    db = database_ps3_links

                # parsing supported
                what_to_up_parsed = [x for x in what_to_up if x in db.keys()]

                if len(what_to_up_parsed) > 0:
                    updatedb(db, i, DBFOLDER, WGET, what_to_up_parsed)
                else:
                    rich.print("Nothing to do!", style='green')

            fillterm()

            if get_xmas():
                rich.print(f"Done! [red]Happy Holidays!!![/red] 🎄✨",
                           style='green')
            else:
                rich.print(f"Done!", style='green')

            sys.exit(0)

        elif args.update == False and args.search is None:
            rich.print(
                "Error: No search term provided, you need to search for something, use -h for help",
                style='red on black')
            sys.exit(1)

        # checking for database's existense:
        if os.path.isfile(f"{DBFOLDER}/pynps.db") is False:
            rich.print(
                "Error: theres no database in your system, please update your database and try again, use -h for help",
                style='red on black')
            sys.exit(1)

        # check region
        if args.region == None:
            reg = ["usa", "eur", "jap", "asia", "int"]
        else:
            reg = args.region

        # maybe_download = []
        maybe_download = search_db(system, what_to_dl, args.search, reg,
                                   args.sort, DBFOLDER)

        # test if the result isn't empty
        if len(maybe_download) == 0:
            rich.print(
                "No results found, try searching for something else or updating your database",
                style='dark_orange')
            sys.exit(0)

        # adding indexes to maybe_download
        for i in range(0, len(maybe_download)):
            # maybe_download[i]["Index"] = str(i)
            maybe_download[i]["Index"] = str(i + 1)

        # print possible mathes to the user
        if len(maybe_download) > 1:
            if args.print is True:
                for i in maybe_download:
                    print(
                        f"{i['Index']} {i['System']} | {i['Title ID']} | {variables.REGION_DICT[i['Region']]} | {i['Type']} | {i['Name']} | [{file_size(i['File Size'])}]"
                        .encode("utf-8"))
                sys.exit(0)
            fillterm("[green][SEARCH] here are the matches[/green]")
            process_search(maybe_download)

        if args.noconfirm is False:
            # validating input
            class Check_game_input(Validator):
                def validate(self, document):
                    text = document.text
                    if len(text) > 0:
                        # test if number being typed is bigger than the biggest number from game list
                        num_p = len(maybe_download)
                        if num_p == 1 and text != "1":
                            raise ValidationError(
                                message="Your only option is to type 1.",
                                cursor_position=0)
                        text_processed = text.replace("-", " ").replace(
                            ",", " ").split(" ")
                        last_texttext_processed = [
                            x for x in text_processed if x != ""
                        ]
                        last_text = text_processed[-1]

                        if "0" in text_processed:
                            raise ValidationError(
                                message='Zero is an invalid entry.')

                        if last_text.isdigit():
                            last_text = int(last_text)
                            if last_text > num_p:
                                raise ValidationError(
                                    message=
                                    f"There are no entries past {num_p}.")

                        if text.startswith("-") or text.startswith(","):
                            # break
                            raise ValidationError(
                                message=
                                'Start the input with a number. Press "h" for help.',
                                cursor_position=0)

                        if "--" in text or ",," in text or ",-" in text or "-," in text:
                            raise ValidationError(
                                message=
                                'Use only one symbol to separate numbers. Press "h" for help.'
                            )

                        text = text.replace("-", "").replace(",", "")
                        if text.isdigit() is False and text != "h":
                            raise ValidationError(
                                message='Do not use leters. Press "h" for help.'
                            )
                    else:
                        raise ValidationError(
                            message=
                            'Enter something or press Ctrl+C to close. Press "h" for help.',
                            cursor_position=0)

            if len(maybe_download) > 1:
                try:
                    index_to_download_raw = prompt(
                        "Enter the number for what you want to download, you can enter multiple numbers using commas: ",
                        validator=Check_game_input())
                except KeyboardInterrupt:
                    rich.print('Interrupted by user', style='bright_black')
                    sys.exit(0)
                except:
                    rich.print('Interrupted by user', style='bright_black')
                    sys.exit(0)
            else:
                index_to_download_raw = "1"

            # provides help
            if index_to_download_raw.lower() == "h":
                rich.print("\tSuppose you have 10 files to select from:",
                           style='bright_black')
                rich.print("\tTo download file 2, you type: 2",
                           style='bright_black')
                rich.print(
                    "\tTo download files 1 to 9, the m*******t method, you type: 1,2,3,4,5,6,7,8,9",
                    style='bright_black')
                rich.print(
                    "\tTo download files 1 to 9, the cool-kid method, you type: 1-9",
                    style='bright_black')
                rich.print(
                    "\tTo download files 1 to 5 and files 8 to 10: 1-5,8-10",
                    style='bright_black')
                rich.print(
                    "\tTo download files 1, 4 and files 6 to 10: 1,4,6-10:",
                    style='bright_black')
                rich.print(
                    "\tTo download files 1, 4 and files 6 to 10, the crazy way, as the software doesn't care about order or duplicates: 10-6,1,4,6",
                    style='bright_black')
                rich.print("Exiting", style='bright_black')
                sys.exit(0)

            # parsing indexes
            index_to_download_raw = index_to_download_raw.replace(
                " ", "").split(",")
            index_to_download = []
            for i in index_to_download_raw:
                if "-" in i and i.count("-") == 1:
                    # spliting range
                    range_0, range_1 = i.split("-")
                    # test if is digit
                    if range_0.isdigit() == True and range_1.isdigit() == True:
                        # test if there's a zero
                        range_0 = int(range_0)
                        range_1 = int(range_1)

                        if range_0 < 1 or range_1 < 1:
                            rich.print(
                                "Error: invalid syntax, please only use non-zero positive integer numbers",
                                style='red on black')
                            sys.exit(1)
                        # test if digit 0 is bigger than digit 1
                        if range_0 < range_1:
                            # populate the index list
                            for a in range(range_0, range_1 + 1):
                                if a not in index_to_download:
                                    index_to_download.append(a)
                        elif range_0 > range_1:
                            for a in range(range_1, range_0 + 1):
                                if a not in index_to_download:
                                    index_to_download.append(a)
                        else:  # range_0 == range_1
                            if range_0 not in index_to_download:
                                index_to_download.append(range_0)
                    else:
                        rich.print(
                            "Error: syntax, please only use non-zero positive integer numbers",
                            style='red on black')
                        sys.exit(1)
                elif i.isdigit() == True:
                    if int(i) not in index_to_download:
                        index_to_download.append(int(i))
                else:
                    rich.print(
                        "Error: syntax, please only use non-zero positive integer numbers",
                        style='red on black')
                    sys.exit(1)

            # fixing indexes for python syntax and sorting the list
            index_to_download = sorted([int(x) - 1 for x in index_to_download])

            files_to_download = [maybe_download[i] for i in index_to_download]

            # if index_to_download_raw == "1":
            fillterm(
                "[green][SEARCH] you're going to download the following files[/green]"
            )

            process_search(files_to_download)

            # validation 2
            class Check_game_input_y_n(Validator):
                def validate(self, document):
                    text = document.text
                    if len(text) > 0:
                        if text.lower() not in ['y', 'n']:
                            # break
                            raise ValidationError(
                                message='Use "y" for yes and "n" for no',
                                cursor_position=0)
                    else:
                        raise ValidationError(
                            message='Enter something or press Ctrl+C to close.',
                            cursor_position=0)

            try:
                accept = prompt("Download files? [y/n]: ",
                                validator=Check_game_input_y_n())
                if accept.lower() != "y":
                    raise
            except KeyboardInterrupt:
                rich.print('Interrupted by user', style='bright_black')
                sys.exit(0)
            except:
                rich.print('Interrupted by user', style='bright_black')
                sys.exit(0)

        else:  # noconfirm download
            try:
                rich.print(
                    "ATTENTION! Since you used --noconfirm, you'll be downloading all the listed files, download will start in 4 seconds, you can use control+c to cancel now or at any time.",
                    style='green')
                sleep(4)
            except KeyboardInterrupt:
                rich.print('Interrupted by user', style='bright_black')
                sys.exit(0)
            except:
                rich.print('Interrupted by user', style='bright_black')
                sys.exit(0)

            files_to_download = maybe_download

####### skip all inputs to here in case of a resume :)
    """fully process game by game inside a single "for"
    this is useful to change the state for every game in the resume list
    proposed "Status" key in the dictionary will be as follow:
    
    0 - PKG not downloaded:
            this is already checked by the dl_file() fucntion!
            Also, this should be the default parameter?
    
    1 - PKG downloaded, but not extracted:
            the status will be updated to 1 after the download passes checksum
            a good place to do it could be around ### 1 HERE
    
    2 - PKG downloaded and extracted!
            the package should be removed from the resume dictionary when it reaches this status
            a good place to do it could be around ### 2 HERE
    """
    files_downloaded = []
    for i in files_to_download:
        # download file
        dl_result = dl_file(i, i['System'], DLFOLDER, WGET, limit_rate)

        if dl_result is True:
            rich.print("Download finished", style='dim green')
        elif dl_result is False:
            rich.print(
                'Unable to download .pkg file. Do you have a internet connection?',
                style='red on black')

        if dl_result == 'interrupted':
            # interrupted by user
            resume_dict = []
            for i in files_to_download:
                if i not in files_downloaded:
                    resume_dict.append(i)

            # run saving function
            # TODO: alert about closing with control + c to not save session
            # TODO don't let duplicate tags

            fillterm()

            # tag save validation
            class Check_tag_save(Validator):
                def validate(self, document):
                    text = document.text
                    if text.isalnum(
                    ) is False and text != "":  #only let alphanumeric
                        # break
                        raise ValidationError(
                            message=
                            'Only alphanumeric characters, without spaces, are supported for tag naming.',
                            cursor_position=0)

            try:
                tag = prompt(
                    "If you wanna save this download session to easily resume it latter, give this session a tag (you can use control+c to not save it or just leave blank to use a generated tag): ",
                    validator=Check_tag_save())
            except KeyboardInterrupt:
                rich.print('Interrupted by user', style='bright_black')
                sys.exit(0)

            if tag == "":
                tag = False

            # check if the current session already has a loaded UUID
            try:
                session_id = session['session_id']
                if tag == False:
                    tag = session['session_tag']
            except:
                # this means it's a new session!
                session_id = str(id_gen())
            download_save_state(resume_dict, DBFOLDER, id=session_id, tag=tag)

            sys.exit(0)

        downloaded_file_loc = f"{DLFOLDER}/PKG/{i['System']}/{i['Type']}/{i['PKG direct link'].split('/')[-1]}"

        # checksum
        # TODO: transform into function?
        if "SHA256" in i.keys():
            fillterm(
                f"[green on black][CHECKSUM][/green on black][green] Checking if downloaded file is valid"
            )
            if i["SHA256"] == "":
                rich.print('No checksum provided by NPS, skipping check',
                           style='dark_orange')
            else:
                sha256_dl = checksum_file(downloaded_file_loc)

                try:
                    sha256_exp = i["SHA256"]
                except:
                    sha256_exp = ""

                if sha256_dl != sha256_exp:
                    loc = f"{DLFOLDER}/PKG/{i['System']}/{i['Type']}/{i['PKG direct link'].split('/')[-1]}"
                    rich.print(
                        'Checksum not matching, pkg file is probably corrupted, delete it at your download folder and redownload the pkg',
                        style='red on black')
                    rich.print(
                        f'Corrupted file is located at [bright_black]{loc}[/bright_black]',
                        style='red on black')
                    break  # skip file
                else:
                    rich.print('Download is not corrupted :^)', style='green')

        ### 1 HERE

        dl_dile_loc = f"{DLFOLDER}/PKG/{i['System']}/{i['Type']}/{i['PKG direct link'].split('/')[-1]}"
        if PKG2ZIP != False and i['System'] != 'PS3':
            zrif = ""
            if args.compress_zip == True:
                dl_location = f"{DLFOLDER}/ZIP/{i['System']}/{i['Type'].capitalize()}"
            else:
                dl_location = f"{DLFOLDER}/Extracted"

            try:
                zrif = i['zRIF']
            except:
                pass

            # expose the exact directory were the pkg was extracted!
            extraction_folder = ""
            if i['System'] == "PSV":
                if i["Type"] in ["GAMES", "DEMOS"]:
                    extraction_folder = f"{DLFOLDER}/Extracted/app/{i['Title ID']}"
                if i["Type"] == "UPDATES":
                    extraction_folder = f"{DLFOLDER}/Extracted/patch/{i['Title ID']}"
                if i["Type"] == "DLCS":
                    extraction_folder = f"{DLFOLDER}/Extracted/addcont/{i['Title ID']}"
                if i["Type"] == "THEMES":
                    theme_folder_name = get_theme_folder_name(
                        f"{DLFOLDER}/Extracted/bgdl/t/")
                    extraction_folder = f"{DLFOLDER}/Extracted/bgdl/t/{theme_folder_name}/{i['Title ID']}"

            if i['System'] == "PSP":
                if i["Type"] == "GAMES":
                    if cso_factor == None and args.eboot == False:
                        extraction_folder = f"{DLFOLDER}/Extracted/pspemu/ISO/<i>game_name</i> [{i['Title ID']}].iso"
                    elif cso_factor in [str(x) for x in range(1, 10)]:
                        extraction_folder = f"{DLFOLDER}/Extracted/pspemu/ISO/<i>game_name</i> [{i['Title ID']}].cso"
                    elif args.eboot == True:
                        extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}"
                if i["Type"] == "DLCS":
                    extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}"
                if i["Type"] == "THEMES":
                    extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/THEME/<i>theme_name</i>.PTF"
                if i["Type"] == "UPDATES":
                    extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}"

            if i['System'] == "PSX":
                extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}"

            if i['System'] == "PSM":
                extraction_folder = f"{DLFOLDER}/Extracted/psm/{i['Title ID']}"

            # -x is default argument to not create .zip files
            if args.compress_zip == False:
                pkg2zip_args = ["-x"]
            else:
                pkg2zip_args = []

            if cso_factor != None and i["Type"] == "GAMES" and i[
                    'System'] == "PSP":
                pkg2zip_args.append("-c" + cso_factor)
            elif cso_factor != None and i["Type"] != "UPDATES" and i[
                    'System'] != "PSP":
                rich.print(
                    f"cso is only supported for PSP games, since you're extracting a {i['System']} {i['Type'][:-1].lower()} the compression will be skipped",
                    style='dark_orange')

            if args.eboot == True and i["Type"] == "GAMES" and i[
                    'System'] == "PSP":
                pkg2zip_args.append("-p")
            # append more commands here if needed!

            if args.compress_zip == True:
                extraction_folder = dl_location

            fillterm(
                f"[green on black][PKG2ZIP][/green on black][green] Attempting extraction"
            )

            if i['System'] == "PSV" and zrif not in ["", "MISSING", None]:
                delete = run_pkg2zip(dl_dile_loc, dl_location, PKG2ZIP,
                                     pkg2zip_args, extraction_folder, i, zrif)
            else:
                delete = run_pkg2zip(dl_dile_loc, dl_location, PKG2ZIP,
                                     pkg2zip_args, extraction_folder, i)

            #testing if extraction was completion and delete file if needed
            if delete == True and keepkg == False:
                # delete file
                try:
                    os.remove(dl_dile_loc)
                    rich.print("The compressed .pkg was deleted",
                               style='green')
                except:
                    rich.print(
                        f"Unable to delete file, you may want to it manually at [bright_black]{dl_dile_loc}[/bright_black]",
                        style='red on black')

            ### 2 HERE
            files_downloaded.append(i)
        else:
            if i['System'] != 'PS3':
                fillterm(
                    f"[green on black][PKG2ZIP][/green on black][green] Attempting extraction"
                )
                rich.print(
                    "Skipping extraction since there's no pkg2zip binary in your system",
                    style='red on black')
            else:
                # move if it's a PS3 pkg
                # move to ~/PS3/packages
                # rap files go to ~/PS3/exdata
                fillterm(
                    f"[green on black][MOVE][/green on black][green] Attempting to move files"
                )

                # get downloaded file name
                pkg_location = f"{DLFOLDER}/PKG/{i['System']}/{i['Type']}/{i['PKG direct link'].split('/')[-1]}"

                # new name
                pkg_new_location = f"{DLFOLDER}/PS3/{i['Type']}/packages/{i['Name']} ({i['Region']}) - {i['Content ID']}.pkg"

                delete = False
                try:
                    # touch new folder
                    create_folder(os.path.dirname(pkg_new_location))
                    # copy file and set delte as True if it's sucessfull
                    copyfile(pkg_location, pkg_new_location)
                    rich.print(
                        f"PS3 pkg file moved to [bright_black]{pkg_new_location}[/bright_black]",
                        style='green')
                    delete = True
                except:
                    rich.print(f"Unable to move PS3 pkg file",
                               style='red on black')

                #testing if extraction was completion and delete file if needed
                if delete == True and keepkg == False:
                    # delete file
                    try:
                        os.remove(dl_dile_loc)
                        rich.print(f"The compressed .pkg was deleted",
                                   style='green')
                    except:
                        rich.print(
                            f"Unable to delete file, you may want to it manually at [bright_black]{dl_dile_loc}[/bright_black]",
                            style='red on black')

                # download rap file
                fillterm(
                    f"[green on black][RAP][/green on black][green] Trying to download RAP file"
                )

                if i['RAP'] == "NOT REQUIRED":
                    rich.print(f"RAP files aren't required for this game :^)",
                               style='green')
                elif i['RAP'] == "MISSING" or i['RAP'] == "":
                    rich.print(
                        f"Unfortunatelly there are no RAP files available for this game on Nopaystation",
                        style='dark_orange')
                elif i['RAP'] == "UNLOCK/LICENSE BY DLC":
                    rich.print(f"This pkg is unlocked by a DLC",
                               style='dark_orange')

                else:
                    rap_url = f"https://nopaystation.com/tools/rap2file/{i['Content ID']}/{i['RAP']}"
                    rap_folder = f"{DLFOLDER}/PS3/{i['Type']}/exdata/{i['Content ID']}.rap"
                    # printft(HTML("<green>[RAP] downloaing RAP file</green>"))
                    rap_state = get_rap(i, WGET, rap_folder, rap_url)

                    if rap_state:
                        rich.print(
                            f"PS3 RAP file downloaded to: [bright_black]{rap_folder}[/bright_black]",
                            style='green')
                    else:
                        rich.print(
                            'Unable to download .rap file. Do you have a internet connection?',
                            style='red on black')

            ### 2 HERE
            files_downloaded.append(i)

    fillterm()
    if get_xmas():
        rich.print(f"Done! [red]Happy Holidays!!![/red] 🎄✨",
                   style='green')
    else:
        rich.print(f"Done!", style='green')
예제 #8
0
파일: cli.py 프로젝트: Doctus/pynps
def cli_main(maindir=""):
    """implement pynps cli interface"""

    if get_system() == 'Linux':
        CONFIGFOLDER = f"{os.getenv('HOME')}/.config/pyNPS"
        config_file = f"{CONFIGFOLDER}/settings.ini"
    elif get_system() == 'Windows':
        CONFIGFOLDER = f"{maindir}/pynps_config/"
        config_file = f"{CONFIGFOLDER}/settings.ini"
    else:
        CONFIGFOLDER = ""
        config_file = ""

    # create conf file
    if os.path.isfile(config_file) == False:
        create_config(config_file, CONFIGFOLDER)
        create_folder(CONFIGFOLDER+"/lib/")

    # read conf file
    config = configparser.ConfigParser()
    config.read(config_file)

    # test sections
    if sorted(config.sections()) != sorted(['pyNPS', 'PSV_Links', 'PSP_Links', 'PSX_Links', 'PSM_Links', 'BinaryLocations']):
        printft(HTML("<red>[ERROR] config file: missing sections</red>"))
        print("You need the following sections in your config file: 'PSV_Links', 'PSP_Links', 'PSX_Links', 'PSM_Links', 'BinaryLocations'")
        sys.exit(1)
    if sorted(list(config["PSV_Links"])) != sorted(['games', 'dlcs', 'themes', 'updates', 'demos']):
        printft(HTML("<red>[ERROR] config file: missing options in the PSV_Links section</red>"))
        print("You need the following options in your PSV_Links section: 'games', 'dlcs', 'themes', 'updates', 'demos'")
        sys.exit(1)
    if sorted(list(config["PSP_Links"])) != sorted(['games', 'dlcs', 'themes', 'updates']):
        printft(HTML("<red>[ERROR] config file: missing options in the PSP_Links section</red>"))
        print("You need the following options in your PSP_Links section: 'games', 'dlcs', 'themes', 'updates'")
        sys.exit(1)
    if sorted(list(config["PSX_Links"])) != sorted(['games']):
        printft(HTML("<red>[ERROR] config file: missing options in the PSX_Links section</red>"))
        print("You need the following options in your PSX_Links section: 'games'")
        sys.exit(1)
    if sorted(list(config["PSM_Links"])) != sorted(['games']):
        printft(HTML("<red>[ERROR] config file: missing options in the PSM_Links section</red>"))
        print("You need the following options in your PSM_Links section: 'games'")
        sys.exit(1)

    # making vars
    DBFOLDER = fix_folder_syntax(config['pyNPS']['databasefolder'], maindir)
    DLFOLDER = fix_folder_syntax(config['pyNPS']['downloadfolder'], maindir)
    PKG2ZIP = fix_folder_syntax(config['BinaryLocations']['pkg2zip_location'], maindir)
    WGET = fix_folder_syntax(config['BinaryLocations']['wget_location'], maindir)

    # tests existence of pkg2zip
    PKG2ZIP = check_pkg2zip(PKG2ZIP, CONFIGFOLDER)
    if PKG2ZIP == False:
        printft(HTML("<orange>[PKG2ZIP] you don't have a valid pkg2zip installation or binary in your system, extraction will be skipped</orange>"))
        # sys.exit(1)

    # tests existence of wget
    WGET = check_wget(WGET, CONFIGFOLDER)
    if WGET == False:
        printft(HTML("<red>[ERROR] you don't have a valid wget installation or binary in your system, this program can't work without it</red>"))
        sys.exit(1)

    # makin dicts for links
    database_psv_links = {}
    for key in config["PSV_Links"]:
        database_psv_links[key] = config["PSV_Links"][key]

    database_psp_links = {}
    for key in config["PSP_Links"]:
        database_psp_links[key] = config["PSP_Links"][key]

    database_psx_links = {}
    for key in config["PSX_Links"]:
        database_psx_links[key] = config["PSX_Links"][key]

    database_psm_links = {}
    for key in config["PSM_Links"]:
        database_psm_links[key] = config["PSM_Links"][key]

    # create args
    args, parser = create_args()

    limit_rate = args.limit_rate
    keepkg = args.keepkg
    system = args.console
    cso_factor = args.compress_cso
    
    if args.resume_session:
        # in this case args.search will be considered a tag to fast resume a session

        input_tag = args.search
        if input_tag is not None:
            if input_tag.isalnum() is False:
                printft(HTML("<orange>[DOWNLOAD] Tags can only be alphanumeric without spaces</orange>"))
                input_tag = None
           
        ##load download db
        with SqliteDict(f"{DBFOLDER}/downloads.db", autocommit=False) as database:
            try:
                db = database['resumes']
                if len(db) == 0:
                    raise
            except:
                printft(HTML("<orange>[DOWNLOAD] There are no saved download sessions to resume</orange>"))
                sys.exit(0)

        checker = next((item for item in db if item['session_tag'] == input_tag), None)
        
        ## if tag is prvided and there's a match in the db
        if input_tag is not None and checker is not None:
            # has something in db
            session = checker

        elif checker is None or input_tag is None:
            if input_tag is not None:
                printft(HTML("<orange>[DOWNLOAD] There's no such tag in download sessions</orange>"))

                # validation resume input
                class Check_resume_input_y_n(Validator):
                    def validate(self, document):
                        text = document.text
                        if len(text) > 0:
                            if text.lower() not in ['y', 'n']:
                                # break
                                raise ValidationError(message='Use "y" for yes and "n" for no',
                                                    cursor_position=0)
                        else:
                            raise ValidationError(message='Enter something or press Ctrl+C to close.',
                                                cursor_position=0)

                try:
                    yn_check = prompt("Wanna see all currently download sessions? [y/n]: ",
                                    validator=Check_resume_input_y_n())
                    if yn_check.lower() != "y":
                        raise
                except KeyboardInterrupt:
                    printft(HTML("<grey>Interrupted by user</grey>"))
                    sys.exit(0)
                except:
                    printft(HTML("<grey>Interrupted by user</grey>"))
                    sys.exit(0)
            
            printft(HTML("<grey>%s</grey>") %fill_term())
            
            p_db = []
            for index_file, i in enumerate(db):
                i["Index"] = index_file + 1
                p_db.append(i)

            
            process_resumes(p_db)

            # validating input
            class Check_resume_input(Validator):
                def validate(self, document):
                    text = document.text
                    if len(text) > 0:
                        if text.isdigit():

                            # test if number being typed is bigger than the biggest number from game list
                            num_p = len(db)
                            if num_p == 1 and text != "1":
                                raise ValidationError(message="Your only option is to type 1.",
                                                    cursor_position=0)
                            if int(text) > num_p:
                                raise ValidationError(
                                    message=f"There are no entries past {num_p}.")
                        else:
                            raise ValidationError(message="Please only use numbers",
                                                    cursor_position=0)
                    else:
                        raise ValidationError(message='Enter something or press Ctrl+C to close. Press "h" for help.',
                                            cursor_position=0)


            try:
                session_index = prompt("Enter the number for the download session you want to resume: ", 
                                                validator=Check_resume_input())
            except KeyboardInterrupt:
                printft(HTML("<grey>Interrupted by user</grey>"))
                sys.exit(0)
            except:
                printft(HTML("<grey>Interrupted by user</grey>"))
                sys.exit(0)

            session = db[int(session_index) - 1]

        files_to_download = session["session_dict"]

    else:

        what_to_dl = {"games": args.games, "dlcs": args.dlcs, "themes": args.themes,
                    "updates": args.updates, "demos": args.demos}

        if set(what_to_dl.values()) == set([False]):
            for i in what_to_dl:
                what_to_dl[i] = True

        if args.update == True:
            if [args.region, args.search, args.eboot, args.compress_cso] != [None, None, False, None]:
                printft(HTML("<red>[UPDATEDB] you can't search while updating the database</red>"))
                sys.exit(1)

            printft(HTML("<grey>%s</grey>") %fill_term())
            what_to_up = [x for x in what_to_dl if what_to_dl[x] == True]

            for i in system:
                printft(HTML("<green>[UPDATEDB] %s:</green>") %variables.FULL_SYSTEM_NAME[i])
                if i == "PSV":
                    db = database_psv_links
                elif i == "PSP":
                    db = database_psp_links
                elif i == "PSX":
                    db = database_psx_links
                elif i == "PSM":
                    db = database_psm_links
                
                # parsing supported
                what_to_up_parsed = [x for x in what_to_up if x in db.keys()]
                
                if len(what_to_up_parsed) > 0:
                    updatedb(db, i, DBFOLDER, WGET, what_to_up_parsed)
                else:
                    printft(HTML("<blue>Nothing to do!</blue>"))

            printft(HTML("<blue>Done!</blue>"))
            sys.exit(0)

        elif args.update == False and args.search is None:
            printft(HTML("<red>[SEARCH] No search term provided, you need to search for something</red>"))
            parser.print_help()
            sys.exit(1)

        # checking for database's existense:
        if os.path.isfile(f"{DBFOLDER}/pynps.db") is False:
            printft(HTML("<red>[UPDATEDB] theres no database in your system, please update your database and try again</red>"))
            sys.exit(1)


        # check region
        if args.region == None:
            reg = ["usa", "eur", "jap", "asia", "int"]
        else:
            reg = args.region

        # maybe_download = []
        maybe_download = search_db(system, what_to_dl, args.search, reg, DBFOLDER)

        # test if the result isn't empty
        if len(maybe_download) == 0:
            printft(HTML("<orange>[SEARCH] No results found, try searching for something else or updating your database</orange>"))
            sys.exit(0)

        # adding indexes to maybe_download
        for i in range(0, len(maybe_download)):
            # maybe_download[i]["Index"] = str(i)
            maybe_download[i]["Index"] = str(i + 1)

        # print possible mathes to the user
        if len(maybe_download) > 1:
            if args.print is True:
                for i in maybe_download:
                    print(f"{i['Index']} {i['System']} | {i['Title ID']} | {variables.REGION_DICT[i['Region']]} | {i['Type']} | {i['Name']} | [{file_size(i['File Size'])}]".encode("utf-8"))
                sys.exit(0)
            printft(HTML("<grey>%s</grey>") %fill_term())
            printft(HTML("<green>[SEARCH] here are the matches:</green>"))
            process_search(maybe_download)
            
        # validating input
        class Check_game_input(Validator):
            def validate(self, document):
                text = document.text
                if len(text) > 0:
                    # test if number being typed is bigger than the biggest number from game list
                    num_p = len(maybe_download)
                    if num_p == 1 and text != "1":
                        raise ValidationError(message="Your only option is to type 1.",
                                            cursor_position=0)
                    text_processed = text.replace(
                        "-", " ").replace(",", " ").split(" ")
                    last_texttext_processed = [
                        x for x in text_processed if x != ""]
                    last_text = text_processed[-1]

                    if "0" in text_processed:
                        raise ValidationError(message='Zero is an invalid entry.')

                    if last_text.isdigit():
                        last_text = int(last_text)
                        if last_text > num_p:
                            raise ValidationError(
                                message=f"There are no entries past {num_p}.")

                    if text.startswith("-") or text.startswith(","):
                        # break
                        raise ValidationError(message='Start the input with a number. Press "h" for help.',
                                            cursor_position=0)

                    if "--" in text or ",," in text or ",-" in text or "-," in text:
                        raise ValidationError(
                            message='Use only one symbol to separate numbers. Press "h" for help.')

                    text = text.replace("-", "").replace(",", "")
                    if text.isdigit() is False and text != "h":
                        raise ValidationError(
                            message='Do not use leters. Press "h" for help.')
                else:
                    raise ValidationError(message='Enter something or press Ctrl+C to close. Press "h" for help.',
                                        cursor_position=0)

        if len(maybe_download) > 1:
            try:
                index_to_download_raw = prompt("Enter the number for what you want to download, you can enter multiple numbers using commas: ", 
                                                validator=Check_game_input())
            except KeyboardInterrupt:
                printft(HTML("<grey>Interrupted by user</grey>"))
                sys.exit(0)
            except:
                printft(HTML("<grey>Interrupted by user</grey>"))
                sys.exit(0)
        else:
            index_to_download_raw = "1"

        # provides help
        if index_to_download_raw.lower() == "h":
            printft(HTML("<grey>\tSuppose you have 10 files to select from:</grey>"))
            printft(HTML("<grey>\tTo download file 2, you type: 2</grey>"))
            printft(HTML("<grey>\tTo download files 1 to 9, the m*******t method, you type: 1,2,3,4,5,6,7,8,9</grey>"))
            printft(HTML("<grey>\tTo download files 1 to 9, the cool-kid method, you type: 1-9</grey>"))
            printft(HTML("<grey>\tTo download files 1 to 5 and files 8 to 10: 1-5,8-10</grey>"))
            printft(HTML("<grey>\tTo download files 1, 4 and files 6 to 10: 1,4,6-10</grey>"))
            printft(HTML("<grey>\tTo download files 1, 4 and files 6 to 10, the crazy way, as the software doesn't care about order or duplicates: 10-6,1,4,6</grey>"))
            printft(HTML("<grey>Exiting</grey>"))
            sys.exit(0)

        # parsing indexes
        index_to_download_raw = index_to_download_raw.replace(" ", "").split(",")
        index_to_download = []
        for i in index_to_download_raw:
            if "-" in i and i.count("-") == 1:
                # spliting range
                range_0, range_1 = i.split("-")
                # test if is digit
                if range_0.isdigit() == True and range_1.isdigit() == True:
                    # test if there's a zero
                    range_0 = int(range_0)
                    range_1 = int(range_1)

                    if range_0 < 1 or range_1 < 1:
                        print("ERROR: Invalid syntax, please only use non-zero positive integer numbers")
                        sys.exit(1)
                    # test if digit 0 is bigger than digit 1
                    if range_0 < range_1:
                        # populate the index list
                        for a in range(range_0, range_1+1):
                            if a not in index_to_download:
                                index_to_download.append(a)
                    elif range_0 > range_1:
                        for a in range(range_1, range_0+1):
                            if a not in index_to_download:
                                index_to_download.append(a)
                    else:  # range_0 == range_1
                        if range_0 not in index_to_download:
                            index_to_download.append(range_0)
                else:
                    print("ERROR: Invalid syntax, please only use non-zero positive integer numbers")
                    sys.exit(1)
            elif i.isdigit() == True:
                if int(i) not in index_to_download:
                    index_to_download.append(int(i))
            else:
                print("ERROR: Invalid syntax, please only use non-zero positive integer numbers")
                sys.exit(1)

        # fixing indexes for python syntax and sorting the list
        index_to_download = sorted([int(x)-1 for x in index_to_download])

        files_to_download = [maybe_download[i] for i in index_to_download]

        # if index_to_download_raw == "1":
        printft(HTML("<grey>%s</grey>") %fill_term())
        printft(HTML("<green>[SEARCH] you're going to download the following files:</green>"))


        process_search(files_to_download)


        # validation 2
        class Check_game_input_y_n(Validator):
            def validate(self, document):
                text = document.text
                if len(text) > 0:
                    if text.lower() not in ['y', 'n']:
                        # break
                        raise ValidationError(message='Use "y" for yes and "n" for no',
                                            cursor_position=0)
                else:
                    raise ValidationError(message='Enter something or press Ctrl+C to close.',
                                        cursor_position=0)

        try:
            accept = prompt("Download files? [y/n]: ",
                            validator=Check_game_input_y_n())
            if accept.lower() != "y":
                raise
        except KeyboardInterrupt:
            printft(HTML("<grey>Interrupted by user</grey>"))
            sys.exit(0)
        except:
            printft(HTML("<grey>Interrupted by user</grey>"))
            sys.exit(0)

####### skip all inputs to here in case of a resume :)
    """fully process game by game inside a single "for"
    # this is useful to change the state for every game in the resume list
    # proposed "Status" key in the dictionary will be as follow:
    
    # 0 - PKG not downloaded:
    #         this is already checked by the dl_file() fucntion!
    #         Also, this should be the default parameter?
    
    # 1 - PKG downloaded, but not extracted:
    #         the status will be updated to 1 after the download passes checksum
    #         a good place to do it could be around ### 1 HERE
    
    # 2 - PKG downloaded and extracted!
    #         the package should be removed from the resume dictionary when it reaches this status
    #         a good place to do it could be around ### 2 HERE

    'files_downloaded' starts with no itens
    'files_to_download' starts with all itens to be downloaded

    for i in files_to_download:
        downloads i

        if dl_results is False:
            do processing for resuming download latter
        
        save download locationg into 'downloaded_file_loc' var

        call pkg_checksum() function (WIP) into 'chekcsum' var

        if checksum is True:
            if PKG2ZIP == False:
                append i to 'files_downloaded' list
            else:
                proceed to extract pkg with pkg2zip

                if extraction is ok:
                    delete pkg file if needed
                    append i to 'files_downloaded' list
                    
                remove i from list in the database!!! 
        else:
            skip file
    """
    files_downloaded = []
    for i in files_to_download:
        # download file
        dl_result = dl_file(i,  i['System'], DLFOLDER, WGET, limit_rate)

        if dl_result is False:
            # interrupted by user
            resume_dict = []
            for i in files_to_download:
                if i not in files_downloaded:
                    resume_dict.append(i)
            
            # run saving function
            # TODO: alert about closing with control + c to not save session
            # TODO don't let duplicate tags
            
            printft(HTML("<grey>%s</grey>") %fill_term())
            # tag save validation
            class Check_tag_save(Validator):
                def validate(self, document):
                    text = document.text
                    if text.isalnum() is False and text != "": #only let alphanumeric
                        # break
                        raise ValidationError(message='Only alphanumeric characters, without spaces, are supported for tag naming.',
                                            cursor_position=0)

            try:
                tag = prompt("If you wanna save this download session to easily resume it latter, give this session a tag (you can use control+c to not save it or just leave blank to use a generated tag): ",
                                validator=Check_tag_save())
            except KeyboardInterrupt:
                printft(HTML("<grey>Interrupted by user</grey>"))
                sys.exit(0)
            
            if tag == "":
                tag = False

            # check if the current session already has a loaded UUID
            try:
                session_id = session['session_id']
                if tag == False:
                    tag = session['session_tag']
            except:
                # this means it's a new session!
                session_id = str(id_gen())
            download_save_state(resume_dict, DBFOLDER, id=session_id,  tag=tag)

            sys.exit(0)

        downloaded_file_loc = f"{DLFOLDER}/PKG/{i['System']}/{i['Type']}/{i['PKG direct link'].split('/')[-1]}"

        # checksum
        # TODO: transform into function?
        if "SHA256" in i.keys():
            printft(HTML("<grey>%s</grey>") %fill_term())
            if i["SHA256"] == "":
                printft(HTML("<orange>[CHECKSUM] No checksum provided by NPS, skipping check</orange>"))
            else:

                sha256_dl = checksum_file(downloaded_file_loc)

                try:
                    sha256_exp = i["SHA256"]
                except:
                    sha256_exp = ""

                if sha256_dl != sha256_exp:
                    loc = f"{DLFOLDER}/PKG/{i['System']}/{i['Type']}/{i['PKG direct link'].split('/')[-1]}"
                    printft(HTML("<red>[CHECKSUM] checksum not matching, pkg file is probably corrupted, delete it at your download folder and redownload the pkg</red>"))
                    printft(HTML("<red>[CHECKSUM] corrupted file location: %s</red>") %loc)
                    break # skip file
                else:
                    printft(HTML("<green>[CHECKSUM] downloaded is not corrupted!</green>"))
        
        ### 1 HERE
        

        printft(HTML("<grey>%s</grey>") %fill_term())

        if PKG2ZIP != False:
            zrif = ""
            dl_dile_loc = f"{DLFOLDER}/PKG/{i['System']}/{i['Type']}/{i['PKG direct link'].split('/')[-1]}"
            dl_location = f"{DLFOLDER}/Extracted"

            try:
                zrif = i['zRIF']
            except:
                pass

            # expose the exact directory were the pkg was extracted!
            extraction_folder = ""
            if i['System'] == "PSV":
                if i["Type"] in ["GAMES", "DEMOS"]:
                    extraction_folder = f"{DLFOLDER}/Extracted/app/{i['Title ID']}"
                    # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/app/{i['Title ID']}</green>"))
                if i["Type"] == "UPDATES":
                    extraction_folder = f"{DLFOLDER}/Extracted/patch/{i['Title ID']}"
                    # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/patch/{i['Title ID']}</green>"))
                if i["Type"] == "DLCS":
                    extraction_folder = f"{DLFOLDER}/Extracted/addcont/{i['Title ID']}"
                    # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/addcont/{i['Title ID']}</green>"))
                if i["Type"] == "THEMES":
                    theme_folder_name = get_theme_folder_name(f"{DLFOLDER}/Extracted/bgdl/t/")
                    extraction_folder = f"{DLFOLDER}/Extracted/bgdl/t/{theme_folder_name}/{i['Title ID']}"
                    # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/bgdl/t/{theme_folder_name}/{i['Title ID']}</green>"))

            if i['System'] == "PSP":
                if i["Type"] == "GAMES":
                    if cso_factor == None and args.eboot == False:
                        extraction_folder = f"{DLFOLDER}/Extracted/pspemu/ISO/<i>game_name</i> [{i['Title ID']}].iso"
                        # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/pspemu/ISO/<i>game_name</i> [{i['Title ID']}].iso</green>"))
                    elif cso_factor in [str(x) for x in range(1, 10)]:
                        extraction_folder = f"{DLFOLDER}/Extracted/pspemu/ISO/<i>game_name</i> [{i['Title ID']}].cso"
                        # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/pspemu/ISO/<i>game_name</i> [{i['Title ID']}].cso</green>"))
                    elif args.eboot == True:
                        extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}"
                        # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}</green>"))
                if i["Type"] == "DLCS":
                    extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}"
                    # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}</green>"))
                if i["Type"] == "THEMES":
                    extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/THEME/<i>theme_name</i>.PTF"
                    # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/pspemu/PSP/THEME/<i>theme_name</i>.PTF</green>"))
                if i["Type"] == "UPDATES":
                    extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}"
                    # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}</green>"))

            if i['System'] == "PSX":
                extraction_folder = f"{DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}"
                # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/pspemu/PSP/GAME/{i['Title ID']}</green>"))

            if i['System'] == "PSM":
                extraction_folder = f"{DLFOLDER}/Extracted/psm/{i['Title ID']}"
                # printft(HTML(f"<green>[EXTRACTION] {i['Name']} ➔ {DLFOLDER}/Extracted/psm/{i['Title ID']}</green>"))

            # -x is default argument to not create .zip files
            pkg2zip_args = ["-x"]
            if cso_factor != None and i["Type"] == "GAMES" and i['System'] == "PSP":
                pkg2zip_args.append("-c"+cso_factor)
            elif cso_factor != None and i["Type"] != "UPDATES" and i['System'] != "PSP":
                printft(HTML("<orange>[EXTRACTION] cso is only supported for PSP games, since you're extracting a %s %s the compression will be skipped</orange>") %(i['System'], i['Type'][:-1].lower()))

            if args.eboot == True and i["Type"] == "GAMES" and i['System'] == "PSP":
                pkg2zip_args.append("-p")
            # append more commands here if needed!

            printft(HTML("<green>[PKG2ZIP] Attempting to extract [%s]%s</green>") %(i['Title ID'], i['Name']))

            if i['System'] == "PSV" and zrif not in ["", "MISSING", None]:
                delete = run_pkg2zip(dl_dile_loc, dl_location, PKG2ZIP, pkg2zip_args, extraction_folder, zrif)
            else:
                delete = run_pkg2zip(dl_dile_loc, dl_location, PKG2ZIP, pkg2zip_args, extraction_folder)

            #testing if extraction was completion and delete file if needed
            
            if delete == True and keepkg == False:
                # delete file
                printft(HTML("<green>[EXTRACTION] Attempting to delete .pkg file</green>"))
                try:
                    os.remove(dl_dile_loc)
                    printft(HTML("<green>[EXTRACTION] Success, the compressed .pkg was deleted</green>"))
                except:
                    printft(HTML("<red>[EXTRACTION] Unable to delete, you may want to it manually at: </red><grey>%s</grey>") %dl_dile_loc)

            ### 2 HERE
            files_downloaded.append(i)
        else:
            printft(HTML("<orange>[EXTRACTION] skipping extraction since there's no pkg2zip binary in your system</orange>"))
            
            ### 2 HERE
            files_downloaded.append(i)

    printft(HTML("<grey>%s</grey>") %fill_term())
    printft(HTML("<blue>Done!</blue>"))