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