Release 0.0.5

CHANGELOG.md

@@ -3,6 +3,12 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 
+## [0.0.5] - 2020-08-03
+### Fixed
+- Issue #4: fix broken progressbar
+- Don't download the .ird if it already exists
+- Don't show progressbar if --quiet flag is set
+
 ## [0.0.4] - 2020-08-03
 ### Fixed
 - Build

libray/core.py

@@ -95,11 +95,16 @@ def error(msg):
 
 def warning(msg):
   """Print a warning message"""
-  print('WARNING: %s. Continuing regardless' % msg)
+  print('WARNING: %s. Continuing regardless.' % msg)
 
 
 def download_ird(ird_name):
   """Download an .ird from GET_IRD_NET_LOC"""
+
+  # Check if file already exists and skip if it does
+  if os.path.exists(ird_name):
+    return
+
   ird_link = GET_IRD_NET_LOC + ird_name
   r = requests.get(ird_link, stream=True)
 
@@ -111,7 +116,10 @@ def download_ird(ird_name):
 def ird_by_game_id(game_id):
   """Using a game_id, download the responding .ird from ALL_IRD_NET_LOC"""
   gameid = game_id.replace('-','')
-  r = requests.get(ALL_IRD_NET_LOC, headers = {'User-Agent': 'Anonymous (You)' }, timeout=5)
+  try:
+    r = requests.get(ALL_IRD_NET_LOC, headers = {'User-Agent': 'Anonymous (You)' }, timeout=5)
+  except requests.exceptions.ReadTimeout:
+    core.error('Server timed out, fix your connection or manually specify a key/ird.')
   soup = BeautifulSoup(r.text, "html.parser")
 
   ird_name = False
@@ -121,7 +129,7 @@ def ird_by_game_id(game_id):
       ird_name = url
 
   if not ird_name:
-    error("Unable to download IRD, couldn't find link")
+    error("Unable to download IRD, couldn't find link. You could specify the decryption key with -d if you have it.")
 
   download_ird(ird_name)
 

libray/iso.py

@@ -104,7 +104,7 @@ class ISO:
     if not args.decryption_key:
       if not args.ird:
         if not args.quiet:
-          core.warning('No IRD file specified, downloading required file')
+          core.warning('No IRD file specified, finding required file')
         args.ird = core.ird_by_game_id(self.game_id) # Download ird
 
       self.ird = ird.IRD(args)
@@ -135,7 +135,8 @@ class ISO:
 
       with open(output_name, 'wb') as output_iso:
 
-        pbar = tqdm(total= (self.size // 2048) - 4 )
+        if not args.quiet:
+          pbar = tqdm(total= (self.size // 2048) )
 
         for region in self.regions:
           input_iso.seek(region['start'])
@@ -148,6 +149,9 @@ class ISO:
                 core.warning('Trying to read past the end of the file')
                 break
               output_iso.write(data)
+
+              if not args.quiet:
+                pbar.update(1)
             continue
           # Encrypted region, decrypt then write
           else:
@@ -168,9 +172,12 @@ class ISO:
 
               output_iso.write(decrypted)
 
-          pbar.update(1)
+              if not args.quiet:
+                pbar.update(1)
 
-        pbar.close()
+        if not args.quiet:
+          pbar.close()
+          print('Decryption complete.')
 
 
   def print_info(self):

setup.py

@@ -5,7 +5,7 @@ from setuptools import setup
 
 setup(
   name="libray",
-  version="0.0.4",
+  version="0.0.5",
   description='A Libre (FLOSS) Python application for unencrypting, extracting, repackaging, and encrypting PS3 ISOs',
   author="Nichlas Severinsen",
   author_email="ns@nsz.no",