Esempio n. 1
0
 def c_uninstall(self, args):
     if args:
         addons = args.split(',')
         with tqdm(total=len(addons),
                   bar_format='{n_fmt}/{total_fmt} |{bar}|') as pbar:
             for addon in addons:
                 name, version = self.core.del_addon(addon)
                 if name:
                     self.table_data.append([
                         f'{Fore.LIGHTRED_EX}Uninstalled{Fore.RESET}', name,
                         version
                     ])
                 else:
                     self.table_data.append([
                         f'{Fore.LIGHTBLACK_EX}Not installed{Fore.RESET}',
                         addon, ''
                     ])
                 pbar.update(1)
         print(self.table.table)
     else:
         printft(
             HTML(
                 '<ansigreen>Usage:</ansigreen>\n\tThis command accepts a comma-separated list of links or addo'
                 'n names as an argument.\n<ansigreen>Supported URLs:</ansigreen>\n\thttps://www.curseforge.com'
                 '/wow/addons/[addon_name]\n\thttps://www.wowinterface.com/downloads/[addon_name]\n\tElvUI\n\tE'
                 'lvUI:Dev\n\tTukUI'))
Esempio n. 2
0
 def c_uninstall(self, args):
     if args:
         addons = [
             addon.strip()
             for addon in list(reader([args], skipinitialspace=True))[0]
         ]
         with tqdm(total=len(addons),
                   bar_format='{n_fmt}/{total_fmt} |{bar}|') as pbar:
             for addon in addons:
                 name, version = self.core.del_addon(addon)
                 if name:
                     self.tableData.append([
                         f'{AC.LIGHTRED_EX}Uninstalled{AC.RESET}', name,
                         version
                     ])
                 else:
                     self.tableData.append([
                         f'{AC.LIGHTBLACK_EX}Not installed{AC.RESET}',
                         addon, ''
                     ])
                 pbar.update(1)
         self.sanitize_table()
         printft(ANSI(self.table.table))
     else:
         printft(
             HTML(
                 '<ansigreen>Usage:</ansigreen>\n\tThis command accepts a comma-separated list of links as an a'
                 'rgument.\n<ansigreen>Supported URLs:</ansigreen>\n\thttps://www.curseforge.com/wow/addons/[ad'
                 'don_name] <ansiwhite>|</ansiwhite> cf:[addon_name]\n\thttps://www.wowinterface.com/downloads/'
                 '[addon_name] <ansiwhite>|</ansiwhite> wowi:[addon_id]\n\thttps://www.tukui.org/addons.php?id='
                 '[addon_id] <ansiwhite>|</ansiwhite> tu:[addon_id]\n\thttps://www.tukui.org/classic-addons.php'
                 '?id=[addon_id] <ansiwhite>|</ansiwhite> tuc:[addon_id]\n\tElvUI <ansiwhite>|</ansiwhite> ElvU'
                 'I:Dev\n\tTukui'))
Esempio n. 3
0
 def c_help(self, _):
     printft(HTML('<ansigreen>install [URL]</ansigreen>\n\tCommand accepts a comma-separated list of links.\n'
                  '<ansigreen>uninstall [URL/Name]</ansigreen>\n\tCommand accepts a comma-separated list of links or'
                  ' addon names.\n'
                  '<ansigreen>update [URL/Name]</ansigreen>\n\tCommand accepts a comma-separated list of links or ad'
                  'don names.\n\tIf no argument is provided all non-modified addons will be updated.\n'
                  '<ansigreen>force_update [URL/Name]</ansigreen>\n\tCommand accepts a comma-separated list of links'
                  ' or addon names.\n\tSelected addons will be reinstalled or updated regardless of their current st'
                  'ate.\n'
                  '<ansigreen>wa_update</ansigreen>\n\tCommand detects all installed WeakAuras and generate WeakAura'
                  's Companion payload.\n'
                  '<ansigreen>status</ansigreen>\n\tPrints the current state of all installed addons.\n'
                  '<ansigreen>orphans</ansigreen>\n\tPrints list of orphaned directories and files.\n'
                  '<ansigreen>search [Keyword]</ansigreen>\n\tExecutes addon search on CurseForge.\n'
                  '<ansigreen>import</ansigreen>\n\tCommand attempts to import already installed addons.\n'
                  '<ansigreen>export</ansigreen>\n\tCommand prints list of all installed addons in a form suitable f'
                  'or sharing.\n'
                  '<ansigreen>toggle_backup</ansigreen>\n\tEnables/disables automatic daily backup of WTF directory.'
                  '\n<ansigreen>toggle_dev [Name]</ansigreen>\n\tCommand accepts an addon name as argument.\n\tPrior'
                  'itizes alpha/beta versions for the provided addon.\n'
                  '<ansigreen>toggle_wa [Username]</ansigreen>\n\tEnables/disables automatic WeakAuras updates.\n\tI'
                  'f a username is provided check will start to ignore the specified author.\n'
                  '<ansigreen>set_wa_api [API key]</ansigreen>\n\tSets Wago API key required to access private auras'
                  '.\n\tIt can be procured here: https://wago.io/account\n'
                  '<ansigreen>set_wa_wow_account [Account name]</ansigreen>\n\tSets WoW account used by WeakAuras up'
                  'dater.\n\tNeeded only if WeakAuras are used on more than one WoW account.\n'
                  '<ansigreen>uri_integration</ansigreen>\n\tEnables integration with CurseForge page. "Install" but'
                  'ton will now start this application.\n'
                  '\n<ansibrightgreen>Supported URL:</ansibrightgreen>\n\thttps://www.curseforge.com/wow/addons/[add'
                  'on_name] <ansiwhite>|</ansiwhite> cf:[addon_name]\n\thttps://www.wowinterface.com/downloads/[addo'
                  'n_name] <ansiwhite>|</ansiwhite> wowi:[addon_id]\n\thttps://www.tukui.org/addons.php?id=[addon_id'
                  '] <ansiwhite>|</ansiwhite> tu:[addon_id]\n\thttps://www.tukui.org/classic-addons.php?id=[addon_id'
                  '] <ansiwhite>|</ansiwhite> tuc:[addon_id]\n\tElvUI <ansiwhite>|</ansiwhite> ElvUI:Dev\n\tTukui'))
Esempio n. 4
0
 def c_install(self, args):
     if args:
         addons = [addon.strip() for addon in args.split(',')]
         with tqdm(total=len(addons),
                   bar_format='{n_fmt}/{total_fmt} |{bar}|') as pbar:
             for addon in addons:
                 installed, name, version = self.core.add_addon(addon)
                 if installed:
                     self.tableData.append(
                         [f'{AC.GREEN}Installed{AC.RESET}', name, version])
                 else:
                     self.tableData.append([
                         f'{AC.LIGHTBLACK_EX}Already installed{AC.RESET}',
                         name, version
                     ])
                 pbar.update(1)
         printft(ANSI(self.table.table))
     else:
         printft(
             HTML(
                 '<ansigreen>Usage:</ansigreen>\n\tThis command accepts a comma-separated list of links as an a'
                 'rgument.\n<ansigreen>Supported URLs:</ansigreen>\n\thttps://www.curseforge.com/wow/addons/[ad'
                 'don_name] <ansiwhite>|</ansiwhite> cf:[addon_name]\n\thttps://www.wowinterface.com/downloads/'
                 '[addon_name] <ansiwhite>|</ansiwhite> wowi:[addon_id]\n\thttps://www.tukui.org/addons.php?id='
                 '[addon_id]\n\thttps://www.tukui.org/classic-addons.php?id=[addon_id]\n\tElvUI <ansiwhite>|</a'
                 'nsiwhite> ElvUI:Dev\n\tTukui'))
Esempio n. 5
0
 def c_install(self, args):
     if args:
         if args.startswith('-i '):
             args = args[3:]
             optignore = True
         else:
             optignore = False
         addons = [addon.strip() for addon in args.split(',')]
         with tqdm(total=len(addons),
                   bar_format='{n_fmt}/{total_fmt} |{bar}|') as pbar:
             for addon in addons:
                 installed, name, version = self.core.add_addon(
                     addon, optignore)
                 if installed:
                     self.tableData.append(
                         [f'{AC.GREEN}Installed{AC.RESET}', name, version])
                 else:
                     self.tableData.append([
                         f'{AC.LIGHTBLACK_EX}Already installed{AC.RESET}',
                         name, version
                     ])
                 pbar.update(1)
         self.sanitize_table()
         printft(ANSI(self.table.table))
     else:
         printft(
             HTML(
                 '<ansigreen>Usage:</ansigreen>\n\tThis command accepts a comma-separated list of links as an a'
                 'rgument.\n\tOption <ansiwhite>-i</ansiwhite> will disable the client version check.\n<ansigre'
                 'en>Supported URLs:</ansigreen>\n\thttps://www.curseforge.com/wow/addons/[addon_name] <ansiwhi'
                 'te>|</ansiwhite> cf:[addon_name]\n\thttps://www.wowinterface.com/downloads/[addon_name] <ansi'
                 'white>|</ansiwhite> wowi:[addon_id]\n\thttps://www.tukui.org/addons.php?id=[addon_id] <ansiwh'
                 'ite>|</ansiwhite> tu:[addon_id]\n\thttps://www.tukui.org/classic-addons.php?id=[addon_id] <an'
                 'siwhite>|</ansiwhite> tuc:[addon_id]\n\tElvUI <ansiwhite>|</ansiwhite> ElvU'
                 'I:Dev\n\tTukui'))
Esempio n. 6
0
 def c_force_update(self, args):
     if args:
         self.c_update(args, False, True, True)
     else:
         printft(
             HTML(
                 '<ansigreen>Usage:</ansigreen>\n\tThis command accepts a comma-separated list of links or addo'
                 'n names as an argument.'))
Esempio n. 7
0
 def c_uri_integration(self, _):
     self.core.create_reg()
     printft('CurseBreaker.reg file was created. Attempting to import...')
     out = os.system('Reg import CurseBreaker.reg')
     if out != 0:
         printft('Import failed. Please try to import REG file manually.')
     else:
         os.remove('CurseBreaker.reg')
Esempio n. 8
0
 def handle_exception(self, e):
     if len(self.table_data) > 1:
         print(self.table.table)
     if getattr(sys, 'frozen', False):
         printft(HTML(f'\n<ansibrightred>{str(e)}</ansibrightred>'))
     else:
         sys.tracebacklimit = 1000
         traceback.print_exc()
Esempio n. 9
0
 def c_update(self, args, addline=False, update=True, force=False):
     if len(self.core.cfCache) > 0 or len(self.core.wowiCache) > 0:
         self.core.cfCache = {}
         self.core.wowiCache = {}
     if args:
         addons = [
             addon.strip()
             for addon in list(reader([args], skipinitialspace=True))[0]
         ]
     else:
         addons = sorted(self.core.config['Addons'],
                         key=lambda k: k['Name'].lower())
         self.core.bulk_check(addons)
     with tqdm(total=len(addons),
               bar_format='{n_fmt}/{total_fmt} |{bar}|') as pbar:
         exceptions = []
         for addon in addons:
             try:
                 name, versionnew, versionold, modified = self.core.\
                     update_addon(addon if isinstance(addon, str) else addon['URL'], update, force)
                 if versionold:
                     if versionold == versionnew:
                         if modified:
                             self.tableData.append([
                                 f'{AC.LIGHTRED_EX}Modified{AC.RESET}',
                                 name, versionold
                             ])
                         else:
                             self.tableData.append([
                                 f'{AC.GREEN}Up-to-date{AC.RESET}', name,
                                 versionold
                             ])
                     else:
                         if modified:
                             self.tableData.append([
                                 f'{AC.LIGHTRED_EX}Update suppressed{AC.RESET}',
                                 name, versionold
                             ])
                         else:
                             self.tableData.append([
                                 f'{AC.YELLOW}{"Updated " if update else "Update available"}'
                                 f'{AC.RESET}', name,
                                 f'{AC.YELLOW}{versionnew}{AC.RESET}'
                             ])
                 else:
                     self.tableData.append([
                         f'{AC.LIGHTBLACK_EX}Not installed{AC.RESET}',
                         addon, ''
                     ])
             except Exception as e:
                 exceptions.append(e)
             pbar.update(1)
     self.sanitize_table()
     printft(ANSI('\n' + self.table.table if addline else self.table.table))
     if len(exceptions) > 0:
         self.handle_exception(exceptions, False)
Esempio n. 10
0
 def c_symlink_protection(self, _):
     if self.os == 'Windows':
         printft(HTML('<ansigreen>Directories tweaked:</ansigreen>'))
         for root, dirs, _ in os.walk(self.core.path / '..' / '..'):
             for d in dirs:
                 path = Path(root) / d
                 if os.path.islink(path):
                     set_icon(path, Path("C:/Windows/System32/SHELL32.dll"), 4)
                     print(path.resolve())
     else:
         printft('This feature is available only on Windows.')
Esempio n. 11
0
 def c_uri_integration(self, _):
     self.core.create_reg()
     printft('CurseBreaker.reg file was created. Attempting to import...')
     out = os.system(
         '"' +
         str(Path(os.path.dirname(sys.executable), 'CurseBreaker.reg')) +
         '"')
     if out != 0:
         printft('Import failed. Please try to import REG file manually.')
     else:
         os.remove('CurseBreaker.reg')
Esempio n. 12
0
 def c_toggle_wa(self, args):
     if args:
         if args == self.core.config['WAUsername']:
             printft(
                 HTML(
                     f'WeakAuras version check is now: <ansigreen>ENABLED</ansigreen>\n'
                     f'Auras created by <ansiwhite>{self.core.config["WAUsername"]}</ansiwhite>'
                     f' are now included.'))
             self.core.config['WAUsername'] = ''
         else:
             self.core.config['WAUsername'] = args.strip()
             printft(
                 HTML(
                     f'WeakAuras version check is now: <ansigreen>ENABLED</ansigreen>\n'
                     f'Auras created by <ansiwhite>{self.core.config["WAUsername"]}</ansiwhite>'
                     f' are now ignored.'))
     else:
         if self.core.config['WAUsername'] == 'DISABLED':
             self.core.config['WAUsername'] = ''
             printft(
                 HTML(
                     'WeakAuras version check is now: <ansigreen>ENABLED</ansigreen>'
                 ))
         else:
             self.core.config['WAUsername'] = '******'
             shutil.rmtree(Path('Interface/AddOns/WeakAurasCompanion'),
                           ignore_errors=True)
             printft(
                 HTML(
                     'WeakAuras version check is now: <ansired>DISABLED</ansired>'
                 ))
     self.core.save_config()
Esempio n. 13
0
 def c_install(self, args):
     if args:
         if args.startswith('-i '):
             args = args[3:]
             optignore = True
         else:
             optignore = False
         addons = [
             addon.strip()
             for addon in list(reader([args], skipinitialspace=True))[0]
         ]
         with tqdm(total=len(addons),
                   bar_format='{n_fmt}/{total_fmt} |{bar}|') as pbar:
             results = []
             pool = ThreadPool(len(addons) if len(addons) <= 10 else 10)
             queue = Queue()
             last = False
             for addon in addons:
                 results.append(
                     pool.apply_async(self.core.add_addon,
                                      args=(addon, optignore, queue)))
             for _ in range(len(addons)):
                 queue.get()
                 pbar.update(1)
             pool.close()
             pool.join()
             self.core.save_config()
             for result in range(len(results)):
                 installed, name, version = results[result].get()
                 if installed:
                     self.tableData.append(
                         [f'{AC.GREEN}Installed{AC.RESET}', name, version])
                 else:
                     self.tableData.append([
                         f'{AC.LIGHTBLACK_EX}Already installed{AC.RESET}',
                         name, version
                     ])
         self.sanitize_table()
         printft(ANSI(self.table.table))
     else:
         printft(
             HTML(
                 '<ansigreen>Usage:</ansigreen>\n\tThis command accepts a comma-separated list of links as an a'
                 'rgument.\n\tOption <ansiwhite>-i</ansiwhite> will disable the client version check.\n<ansigre'
                 'en>Supported URLs:</ansigreen>\n\thttps://www.curseforge.com/wow/addons/[addon_name] <ansiwhi'
                 'te>|</ansiwhite> cf:[addon_name]\n\thttps://www.wowinterface.com/downloads/[addon_name] <ansi'
                 'white>|</ansiwhite> wowi:[addon_id]\n\thttps://www.tukui.org/addons.php?id=[addon_id] <ansiwh'
                 'ite>|</ansiwhite> tu:[addon_id]\n\thttps://www.tukui.org/classic-addons.php?id=[addon_id] <an'
                 'siwhite>|</ansiwhite> tuc:[addon_id]\n\tElvUI <ansiwhite>|</ansiwhite> ElvU'
                 'I:Dev\n\tTukui'))
Esempio n. 14
0
 def c_orphans(self, _):
     orphansd, orphansf = self.core.find_orphans()
     printft(HTML('<ansigreen>Directories that are not part of any installed addon:</ansigreen>'))
     for orphan in sorted(orphansd):
         printft(HTML(orphan.replace('[GIT]', '<ansiyellow>[GIT]</ansiyellow>')))
     printft(HTML('\n<ansigreen>Files that are leftovers after no longer installed addons:</ansigreen>'))
     for orphan in sorted(orphansf):
         printft(orphan)
Esempio n. 15
0
 def runner( list, cwd):
     
     p = subprocess.Popen(list, cwd=cwd,
                  stdout=subprocess.PIPE,
                  stderr=subprocess.STDOUT)
     
     r = re.compile('pkg2zip v\\d.\\d')
     full_out = ''
         
     for line in iter(p.stdout.readline, b''):
         out = line.rstrip().decode()
         full_out += f"{out}\n"
         # if out.startswith("ERROR") == False and r.match(out) is None:
         #     rich.print(out, style="dim")
     
     # test for corrupted and file not being a pkg
     if "ERROR: not a pkg file" in full_out:
         # corrupted file and feeding inexistent file
         if os.path.isfile(file):
             if file.endswith(".pkg"):
                 rich.print("The provided file is is a .pkg, but seems to be corrupted", style='red on black')
             else:
                 rich.print("The provided file is is not a .pkg", style='red on black')
         else:
             rich.print("Provided file doesn't exist", style='red on black')
         return False
     elif "ERROR: pkg file is too small" in full_out:
         # download not ended
         rich.print("The provided file is too small, it's probably corrupted or didn't fully downloaded", style='red on black')
         return False
     elif "ERROR: failed to read 256 bytes from file" in full_out:
         # feeded a folder to pkg2zip
         if os.path.isdir(file):
             rich.print("The provided file seems to be a folder", style='red on black')
         else:
             rich.print("Unknown extraction error. Output:", style='red on black')
             rich.print(full_out, style='bright_black on black')
         return False
     elif "ERROR: cannot create 'pspemu' folder" in full_out:
         printft(HTML("<red>[PKG2ZIP] cannot create 'pspemu' folder. Do you have reading permissions for your Download folder?</red>"))
     else:
         if "done!" in full_out:
             rich.print(f"File extracted to: [bright_black]{extraction_folder}", style='green')
             return True
         else:
             rich.print("Unknown extraction error. Output:", style='red on black')
             rich.print(full_out, style='bright_black on black')
             return False
Esempio n. 16
0
 def c_toggle_dev(self, args):
     if args:
         status = self.core.dev_toggle(args)
         if status is None:
             printft(HTML('<ansibrightred>This addon does not exist or it is not installed yet.</ansibrightred>'))
         elif status:
             printft('This addon will now prioritize alpha/beta versions.')
         else:
             printft('This addon will not longer prioritize alpha/beta versions.')
     else:
         printft(HTML('<ansigreen>Usage:</ansigreen>\n\tThis command accepts an addon name as an argument.'))
Esempio n. 17
0
 def c_search(self, args):
     if args:
         results = self.core.search(args)
         printft(HTML('<ansigreen>Top results of your search:</ansigreen>'))
         for url in results:
             if self.core.check_if_installed(url):
                 printft(HTML(f'{url} <ansiyellow>[Installed]</ansiyellow>'))
             else:
                 printft(url)
     else:
         printft(HTML('<ansigreen>Usage:</ansigreen>\n\tThis command accepts a search query as an argument.'))
Esempio n. 18
0
 def auto_update(self):
     if getattr(sys, 'frozen', False):
         try:
             if os.path.isfile(sys.executable + '.old'):
                 os.remove(sys.executable + '.old')
             payload = requests.get(
                 'https://api.github.com/repos/AcidWeb/CurseBreaker/releases/latest',
                 headers=HEADERS).json()
             remoteversion = payload['name']
             changelog = payload['body']
             url = payload['assets'][0]['browser_download_url']
             if StrictVersion(
                     remoteversion[1:]) > StrictVersion(__version__):
                 printft(
                     HTML(
                         '<ansigreen>Updating CurseBreaker...</ansigreen>'))
                 shutil.move(sys.executable, sys.executable + '.old')
                 payload = requests.get(url, headers=HEADERS)
                 with open(sys.executable, 'wb') as f:
                     f.write(payload.content)
                 printft(
                     HTML(
                         f'<ansibrightgreen>Update complete! Please restart the application.</ansibrightgreen'
                         f'>\n\n<ansigreen>Changelog:</ansigreen>\n{changelog}\n'
                     ))
                 os.system('pause')
                 sys.exit(0)
         except Exception as e:
             printft(
                 HTML(
                     f'<ansibrightred>Update failed!\n\nReason: {str(e)}</ansibrightred>\n'
                 ))
             os.system('pause')
             sys.exit(1)
Esempio n. 19
0
def process_resumes(out):
    for dict in out:
        term_cols = get_terminal_columns()

        tag = dict["session_tag"]
        index = dict["Index"]
        dicts = dict["session_dict"]
        time = dict["session_time"]
        id = dict["session_id"]
        pretty_time = dict["session_prettytime"]

        printft(HTML("<green>Session</green><red> %s</red>") % index)

        header = f"TAG: {tag} | SAVED AT: {pretty_time}"
        tail = f"UUID: {id}"

        header_tail = f"{header} | {tail}"
        if len(header_tail) < term_cols:
            rest = " " * (term_cols - (len(header) + len(tail)))
            header_tail = f"{header}{rest}{tail}"

        # print(header_tail)
        printft(HTML("<grey>%s</grey>") % header_tail)
        process_search(dicts, show_index=False)
        printft(HTML("<grey>%s</grey>") % fill_term())
Esempio n. 20
0
 def c_set_wa_wow_account(self, args):
     if args:
         args = args.strip()
         if os.path.isfile(Path(f'WTF/Account/{args}/SavedVariables/WeakAuras.lua')):
             printft(HTML(f'WoW account name set to: <ansiwhite>{args}</ansiwhite>'))
             self.core.config['WAAccountName'] = args
             self.core.save_config()
         else:
             printft('Incorrect WoW account name.')
     else:
         printft(HTML('<ansigreen>Usage:</ansigreen>\n\tThis command accepts the WoW account name as an argument.'))
Esempio n. 21
0
 def c_set_wa_api(self, args):
     if args:
         printft('Wago API key is now set.')
         self.core.config['WAAPIKey'] = args.strip()
         self.core.save_config()
     elif self.core.config['WAAPIKey'] != '':
         printft('Wago API key is now removed.')
         self.core.config['WAAPIKey'] = ''
         self.core.save_config()
     else:
         printft(HTML('<ansigreen>Usage:</ansigreen>\n\tThis command accepts API key as an argument.'))
Esempio n. 22
0
 def auto_update(self):
     if getattr(sys, 'frozen', False):
         try:
             if os.path.isfile(sys.executable + '.old'):
                 try:
                     os.remove(sys.executable + '.old')
                 except PermissionError:
                     pass
             payload = requests.get(
                 'https://api.github.com/repos/AcidWeb/CurseBreaker/releases/latest',
                 headers=HEADERS).json()
             if 'name' in payload and 'body' in payload and 'assets' in payload:
                 remoteversion = payload['name']
                 changelog = payload['body']
                 url = None
                 for binary in payload['assets']:
                     if (self.os == 'Windows' and '.exe' in binary['name'])\
                             or (self.os == 'Darwin' and '.zip' in binary['name'])\
                             or (self.os == 'Linux' and '.gz' in binary['name']):
                         url = binary['browser_download_url']
                         break
                 if url and StrictVersion(
                         remoteversion[1:]) > StrictVersion(__version__):
                     printft(
                         HTML(
                             '<ansigreen>Updating CurseBreaker...</ansigreen>'
                         ))
                     shutil.move(sys.executable, sys.executable + '.old')
                     payload = requests.get(url, headers=HEADERS)
                     if self.os == 'Darwin':
                         zipfile.ZipFile(io.BytesIO(
                             payload.content)).extractall()
                     else:
                         with open(sys.executable, 'wb') as f:
                             if self.os == 'Windows':
                                 f.write(payload.content)
                             elif self.os == 'Linux':
                                 f.write(gzip.decompress(payload.content))
                     os.chmod(sys.executable, 0o775)
                     printft(
                         HTML(
                             f'<ansibrightgreen>Update complete! Please restart the application.</ansibrightgre'
                             f'en>\n\n<ansigreen>Changelog:</ansigreen>\n{changelog}\n'
                         ))
                     pause()
                     sys.exit(0)
         except Exception as e:
             printft(
                 HTML(
                     f'<ansibrightred>Update failed!\n\nReason: {str(e)}</ansibrightred>\n'
                 ))
             pause()
             sys.exit(1)
Esempio n. 23
0
def download_save_state(dict, DBFOLDER, id, tag=False):
    """saves downloads sessions"""
    
    epoch_date = int(time.time())
    pretty_date = datetime.utcfromtimestamp(epoch_date).strftime('%Y-%m-%d %H:%M:%S')

    with SqliteDict(f"{DBFOLDER}/downloads.db", autocommit=False) as database:
        if "resumes" not in database:
            database["resumes"] = []
        
        database_editable = database["resumes"]

        # check, by uuid, if session is already saved
        checker = next((item for item in database_editable if item['session_id'] == id), None)

        if tag in [None, False]:
            tag = str(epoch_date)
            printft(HTML("<green>[DOWNLOAD] saving session with the tag: %s</green>") %epoch_date)
        else:
            # check if tag is already used by other session that's not checker
            tag_checker = next((item for item in database_editable if item['session_tag'] == tag), None)
            if tag_checker is not None:
                if checker is not None:
                    # match tag_chcker uuid with checker uuid
                    # make a new one in case the uuids are different
                    if checker['session_id'] != tag_checker['session_id']:
                        tag = f"{tag}{epoch_date}"
                        printft(HTML("<orange>[DOWNLOAD] tag is in use by other session, new tag will be set as: %s</orange>") %tag)
                else:
                    # just make a new tag appeding epoch_date
                    # since duplication is certain
                    tag = f"{tag}{epoch_date}"
                    printft(HTML("<orange>[DOWNLOAD] tag is in use by other session, new tag will be set as: %s</orange>") %tag)
        

        new_dict = {"session_time":epoch_date,
                                        "session_prettytime":pretty_date,
                                        "session_dict":dict,
                                        "session_tag":tag,
                                        "session_id":id
            }

        if checker is None:
            database_editable.append(new_dict)
        else:
            checker_index = database_editable.index(checker)
            database_editable[checker_index].update(new_dict)

        # try commiting
        database["resumes"] = database_editable
        database.commit()
Esempio n. 24
0
 def auth(self) -> None:
     login_sys = Login()
     try:
         if not login_sys.passwd:
             printft('First run - initialise users')
             admin_user = self.session.prompt(
                 'Admin username (default: {}): '.format(
                     DEFAULT_ADMIN_USERNAME))
             if not admin_user:
                 admin_user = DEFAULT_ADMIN_USERNAME
             admin_pass = self.session.prompt('Admin password: '******'Admin password: '******'admin'])
         else:
             login_attempt = 0
             while True:
                 user = None
                 while not user:
                     user = self.session.prompt('Username: '******'Password: '******'Login attempt limit reached. Exiting')
                     self.exit(1)
                 printft('Username or password incorrect. Please try again')
     except KeyboardInterrupt:
         printft('Exiting...')
         self.exit(0)
     self.run()
Esempio n. 25
0
 def run(self) -> None:
     while True:
         cmd = None
         try:
             cmd = self.session.prompt(self.prompt_text, is_password=False)
         except KeyboardInterrupt:
             printft('CTRL-C pressed')
         except EOFError:
             printft('CTRL-D pressed')
         if cmd == 'exit':
             self.exit()
         if cmd == 'repair':
             self.runrepair()
         if cmd == 'clear':
             self.clear()
         printft('cmd is {}'.format(cmd))
Esempio n. 26
0
 def handle_exception(self, e, table=True):
     if len(self.tableData) > 1 and table:
         self.sanitize_table()
         printft(ANSI(self.table.table))
     if getattr(sys, 'frozen', False):
         if isinstance(e, list):
             for es in e:
                 printft(HTML(f'\n<ansibrightred>{str(es)}</ansibrightred>'))
         else:
             printft(HTML(f'\n<ansibrightred>{str(e)}</ansibrightred>'))
     else:
         if isinstance(e, list):
             for es in e:
                 traceback.print_exception(es, es, es.__traceback__, limit=1000)
         else:
             traceback.print_exc(limit=1000)
Esempio n. 27
0
def create_args():
    parser = argparse.ArgumentParser(
        description=
        'pyNPS is a Nopaystation client writen in python 3.7 that, with the help of wget and pkg2zip, can search, download and decrypt/extract PSVita, PSP, PSX and PSM games from Nopaystation database.'
    )

    parser.add_argument(
        "search",
        help=
        "search something to download, you can search by name or ID or use '_all' to return everythning.",
        type=str,
        nargs="?")
    parser.add_argument("-c",
                        "--console",
                        help="the console you wanna get content with NPS.",
                        type=str,
                        required=False,
                        action='append',
                        choices=["psv", "psp", "psx", "psm", "ps3"])
    parser.add_argument("-r",
                        "--region",
                        help="the region for the pkj you want.",
                        type=str,
                        required=False,
                        action='append',
                        choices=["usa", "eur", "jap", "asia", "int"])
    parser.add_argument(
        "-s",
        "--sort",
        help=
        "sort search output by column name, can string multiple names by using a comma. Available options are: console or c, title_id or id, region or r, type or t, game_name or n, size or s. Default value: c,t,r,n",
        type=str,
        required=False)
    parser.add_argument("-G",
                        "-dg",
                        "--games",
                        help="to download PSV/PSP/PSX/PSM games.",
                        action="store_true")
    parser.add_argument("-D",
                        "-dd",
                        "--dlcs",
                        help="to download PSV/PSP dlcs.",
                        action="store_true")
    parser.add_argument("-T",
                        "-dt",
                        "--themes",
                        help="to download PSV/PSP themes.",
                        action="store_true")
    parser.add_argument("-U",
                        "-du",
                        "--updates",
                        help="to download PSV/PSP game updates.",
                        action="store_true")
    parser.add_argument("-E",
                        "-dde",
                        "--demos",
                        help="to download PSV demos.",
                        action="store_true")
    parser.add_argument("-A",
                        "-da",
                        "--avatars",
                        help="to download PS3 avatars.",
                        action="store_true")
    parser.add_argument(
        "-k",
        "--keepkg",
        help="using this flag will keep the pkg after the extraction",
        action="store_true")
    parser.add_argument(
        "-eb",
        "--eboot",
        help="use this argument to unpack PSP games as EBOOT.PBP",
        action="store_true")
    parser.add_argument(
        "-cso",
        "--compress_cso",
        help=
        "use this argument to unpack PSP games as a compressed .cso file. You can use any number beetween 1 and 9 for compression factors, were 1 is less compressed and 9 is more compressed.",
        type=str,
        required=False,
        choices=[str(x) for x in range(1, 10)])
    parser.add_argument("-zip",
                        "--compress_zip",
                        help="extract pkgs into zip files instead of folders.",
                        action="store_true")
    parser.add_argument(
        "-l",
        "--limit_rate",
        help="limit download speed, input is the same as wget's.",
        type=str,
        required=False)
    parser.add_argument("-u",
                        "--update",
                        help="update database.",
                        action="store_true")
    parser.add_argument(
        "-p",
        "--print",
        help=
        "just print the result and exit, you can use this option to redirect the output to a file!",
        action="store_true")
    parser.add_argument("-R",
                        "--resume_session",
                        help="update database.",
                        action="store_true")
    parser.add_argument('--version',
                        action='version',
                        version=f"%(prog)s version {variables.VERSION}")
    a = parser.parse_args()

    if a.console is not None:
        a.console = list({x.upper() for x in a.console})
    else:
        # consoles that are shown by default, i.e in case the -c flag is not used
        a.console = ["PSV", "PSP", "PSX", "PSM", "PS3"]

    # warn zip, cso and eboot won't work with PS3
    if "PS3" in a.console:
        if a.compress_zip is True:
            printft(
                HTML(
                    "<orange>[WARNING] PS3 games can't be extracted as zip files</orange>"
                ))
        if a.compress_cso is True:
            printft(
                HTML(
                    "<orange>[WARNING] PS3 games can't be compressed as cso files</orange>"
                ))
        if a.eboot is True:
            printft(
                HTML(
                    "<orange>[WARNING] PS3 games can't be packed as eboot files</orange>"
                ))

    # exclusions
    test = [
        a.console, a.region, a.games, a.dlcs, a.themes, a.updates, a.demos,
        a.eboot, a.compress_cso, a.update, a.avatars
    ] == [['PSV', 'PSP', 'PSX', 'PSM', 'PS3'], None, False, False, False,
          False, False, False, None, False, False]
    if a.resume_session is True and test is False:
        printft(
            HTML(
                "<red>[ERROR] you can only use -R/--resume_session alongside the -l/--limit_rate and -k/--keepkg arguments</red>"
            ))
        sys.exit(1)

    # unsuported download types
    if "PSP" in a.console and True in [a.demos, a.avatars]:
        if len(a.console) > 1:
            printft(
                HTML(
                    "<orange>[WARNING] NPS has no support for demos or avatars with the Playstation Portable (PSP)</orange>"
                ))
        else:
            printft(
                HTML(
                    "<red>[ERROR] NPS has no support for demos or avatars with the Playstation Portable (PSP)</red>"
                ))
            sys.exit(1)

    if "PSX" in a.console and True in [
            a.dlcs, a.themes, a.updates, a.demos, a.avatars
    ]:
        if len(a.console) > 1:
            printft(
                HTML(
                    "<orange>[WARNING] NPS only supports game downlaods for the Playstation (PSX)</orange>"
                ))
        else:
            printft(
                HTML(
                    "<red>[ERROR] NPS only supports game downlaods for the Playstation (PSX)</red>"
                ))
            sys.exit(1)
    if "PS3" in a.console and a.updates == True:
        if len(a.console) > 1:
            printft(
                HTML(
                    "<orange>[WARNING] NPS has no support for updates with the Playstation 3 (PS3)</orange>"
                ))
        else:
            printft(
                HTML(
                    "<red>[ERROR] NPS has no support for updates with the Playstation 3 (PS3))</red>"
                ))
            sys.exit(1)
    if "PSV" in a.console and a.avatars == True:
        if len(a.console) > 1:
            printft(
                HTML(
                    "<orange>[WARNING] NPS has no support for avatars with the Playstation Vita (PSV)</orange>"
                ))
        else:
            printft(
                HTML(
                    "<red>[ERROR] NPS has no support for avatars with the Playstation Vita (PSV)</red>"
                ))
            sys.exit(1)
    if "PSM" in a.console and a.avatars == True:
        if len(a.console) > 1:
            printft(
                HTML(
                    "<orange>[WARNING] NPS has no support for avatars with the Playstation Mobile (PSM)</orange>"
                ))
        else:
            printft(
                HTML(
                    "<red>[ERROR] NPS has no support for avatars with the Playstation Mobile (PSM)</red>"
                ))
            sys.exit(1)

    # limit rate string
    if a.limit_rate is not None:
        # check how it ends
        if a.limit_rate[:-1].isdigit() is False or a.limit_rate[-1].lower(
        ) not in ["k", "m", "g", "t"]:
            printft(HTML("<red>[ERROR] invalid format for --limit_rate</red>"))
            sys.exit(1)

    # eboot and cso can't be used at the same time
    if a.eboot is True and a.compress_cso is not None:
        printft(
            HTML(
                "<red>[ERROR] you can't use --eboot and --compress_cso at the same time</red>"
            ))
        sys.exit(1)

    # check order list
    if a.sort is not None:
        for i in a.sort.split(","):
            if i.lower() not in [
                    "c", "id", "r", "t", "gn", "s", "console", "title_id",
                    "region", "type", "game_name", "size"
            ]:
                printft(HTML("<red>[ERROR] invalid list for --order</red>"))
                sys.exit(1)

    return a, parser
Esempio n. 28
0
def dl_file(dict, system, DLFOLDER, WGET, limit_rate):
    """this function downloads the games"""

    system_name = variables.FULL_SYSTEM_NAME[system.upper()]

    url = dict['PKG direct link']
    filename = url.split('/')[-1]
    name = dict['Name']
    title_id = dict['Title ID']

    printft(HTML("<grey>%s</grey>") % fill_term())
    printft(
        HTML("<green>[DOWNLOAD] %s (%s) [%s] for %s</green>") %
        (name, dict['Region'], title_id, system))

    dl_folder = f"{DLFOLDER}/PKG/{system}/{dict['Type']}"

    # making folder
    create_folder(dl_folder)

    # check if file exists
    if os.path.isfile(f"{dl_folder}/{filename}"):
        printft(
            HTML(
                "<orange>[DOWNLOAD] file exists, wget will decide if the file is completely downloaded, if it's not the download will be resumed</orange>"
            ))

    try:
        if limit_rate is None:
            process = subprocess.run(
                [WGET, "-q", "--show-progress", "-c", dl_folder, url],
                cwd=dl_folder)
        else:
            process = subprocess.run([
                WGET, "-q", "--show-progress", "-c", "--limit-rate",
                limit_rate, dl_folder, url
            ],
                                     cwd=dl_folder)
    except KeyboardInterrupt:
        # TODO: add infor about resuming
        printft(
            HTML(
                "\n<orange>[DOWNLOAD] File was partially downloaded, you can resume this download by searching for same pkg again</orange>"
            ))
        printft(
            HTML(
                "<orange>[DOWNLOAD] File location:</orange> <grey>%s/%s</grey>"
            ) % (dl_folder, filename))
        printft(HTML("<grey>Download interrupted by user</grey>"))
        return False
    return True
Esempio n. 29
0
 def c_export(self, _):
     printft(self.core.export_addons())
Esempio n. 30
0
 def c_import(self, args):
     hit, partial_hit, miss = self.core.detect_addons()
     if args == 'install' and len(hit) > 0:
         self.c_install(','.join(hit))
     else:
         printft(HTML(f'<ansigreen>Addons found:</ansigreen>'))
         for addon in hit:
             printft(addon)
         printft(HTML(f'\n<ansiyellow>Possible matches:</ansiyellow>'))
         for addon in partial_hit:
             printft(HTML(' <ansiwhite>or</ansiwhite> '.join(addon)))
         printft(HTML(f'\n<ansired>Unknown directories:</ansired>'))
         for addon in miss:
             printft(f'{addon}')
         printft(
             HTML(
                 f'\nExecute <ansiwhite>import install</ansiwhite> command to install all detected addons.\n'
                 f'Possible matches need to be installed manually with the <ansiwhite>install</ansiwhite>'
                 f' command.'))