added limit flag for compression and decompression

nut/Config.py

@@ -20,6 +20,8 @@
 autoUpdateTitleDb = True
 allowNoMetadata = True
 cdnCacheOnly = False
+limit = 0
+limitCount = 0
 
 region = None
 

nut/__init__.py

@@ -280,6 +280,7 @@ def compress(filePath, compressionLevel=19, outputDir=None, copy = False):
  f.write(buffer)
 
  newNsp.close()
+ container.close()
 
  if copy:
  os.unlink(tmpFilePath)
@@ -367,6 +368,12 @@ def compressAll(level=19, copy = False):
  if Config.download.fileSizeMin is not None and lastestNsp.getFileSize() < Config.download.fileSizeMin:
  continue
 
+ if Config.limit:
+ Config.limitCount += 1
+
+ if Config.limitCount > Config.limit:
+ continue
+
  q.put(lastestNsp.path)
 
  except BaseException as e:
@@ -455,6 +462,12 @@ def decompressAll():
  if Config.download.fileSizeMin is not None and lastestNsz.getFileSize() < Config.download.fileSizeMin:
  continue
 
+ if Config.limit:
+ Config.limitCount += 1
+
+ if Config.limitCount > Config.limit:
+ continue
+
  q.put(lastestNsz.path)
 
  except BaseException as e:
@@ -775,6 +788,13 @@ def _ftpsync(url):
  if path[-3:] == 'nsx':
  if len(Titles.get(nsp.titleId).getFiles('nsp')) or len(Titles.get(nsp.titleId).getFiles('nsz')):
  continue
+
+ if Config.limit:
+ Config.limitCount += 1
+
+ if Config.limitCount > Config.limit:
+ continue
+
  q.put(nsp)
  except BaseException as e:
  Print.error(str(e))
@@ -1269,6 +1289,12 @@ def extractNcaMeta(files = []):
  if hasCnmt(nsp.titleId, nsp.version):
  continue
 
+ if Config.limit:
+ Config.limitCount += 1
+
+ if Config.limitCount > Config.limit:
+ continue
+
  q[path] = nsp
  except BaseException:
  Print.info('exception: %s' % (path))

requirements.txt

@@ -11,7 +11,7 @@ pycryptodome~=3.14.1
 pycryptoplus~=0.0.1
 pyopenssl~=22.0.0
 Flask~=2.0.3
-Pillow~=9.0.1
+Pillow~=10.4.0
 zstandard==0.15.2
 future~=0.18.2
 asn1~=2.4.2