feat: add console argument to skip error caused by unsupported version type

Author: nobbyfix

src/azlassets/classes.py

@@ -4,10 +4,15 @@
 from typing import Self
 
 
+class UnknownVersionTypeError(NotImplementedError):
+	def __init__(self, version_name, *args):
+		super().__init__(f"Unknown versionname {version_name}.", *args)
+		self.version_name = version_name
+
+
 CompareType = Enum('CompareType', 'New Changed Unchanged Deleted')
 DownloadType = Enum('DownloadType', 'NoChange Removed Success Failed ForDeletionNoChange')
 
-
 @dataclass
 class VersionTypeDataMixin:
 	hashname: str

src/azlassets/versioncontrol.py

@@ -3,20 +3,20 @@
 from pathlib import Path
 from collections.abc import Generator, Iterable
 
-from .classes import DownloadType, HashRow, UpdateResult, VersionType, VersionResult, SimpleVersionResult
+from .classes import DownloadType, HashRow, UpdateResult, VersionType, VersionResult, SimpleVersionResult, UnknownVersionTypeError
 
 
 def parse_version_string(rawstring: str) -> VersionResult:
 	"""
 	Tries to parse the raw version string as returned by the game server into a `VersionResult`.
 
-	Raises `NotImplementedError` if the versiontype does not exist.
+	Raises `UnknownVersionTypeError` if the versiontype does not exist.
 	"""
 	parts = rawstring.split('$')[1:]
 	versionname = parts[0]
 	versiontype = VersionType.from_hashname(versionname)
 	if not versiontype:
-		raise NotImplementedError(f'Unknown versionname {versionname}.')
+		raise UnknownVersionTypeError(version_name=versionname)
 
 	if versiontype == VersionType.AZL:
 		version = '.'.join(parts[1:-1])

src/downloader.py

@@ -5,9 +5,18 @@
 from pathlib import Path
 
 from azlassets import __version__, config, protobuf, versioncontrol, updater, repair, downloader
-from azlassets.classes import Client
+from azlassets.classes import Client, UnknownVersionTypeError, VersionResult
 
 
+def try_parse_version_string(vstring: str, skip_error: bool = False) -> VersionResult:
+	try:
+		return versioncontrol.parse_version_string(vstring)
+	except UnknownVersionTypeError as e:
+		if skip_error:
+			print(f"WARN: Unknown version type '{e.version_name}' cannot be processed, but this error has been skipped.")
+			print("WARN: Update application as soon as possible to support this missing version type.")
+		else: raise
+
 async def execute(args):
 	# load config data from files
 	userconfig = config.load_user_config()
@@ -31,7 +40,7 @@ async def execute(args):
 		sys.exit(1)
 
 	version_string = version_response.pb.version
-	versionlist = [versioncontrol.parse_version_string(v) for v in version_string if v.startswith("$")]
+	versionlist = [try_parse_version_string(v, args.skip_unknown_version_error) for v in version_string if v.startswith("$")]
 
 	async with downloader.AzurlaneAsyncDownloader(clientconfig.cdnurl, useragent=userconfig.useragent) as downloader_session:
 		for vresult in versionlist:
@@ -71,6 +80,8 @@ def main():
 		help="Checks if all files are correct using the local hash file.")
 	parser.add_argument("--ignore-hashfile", default=False, action=argparse.BooleanOptionalAction,
 		help="Ignores the local hashfile and downloads ALL files again. This is only intended for testing purposes.")
+	parser.add_argument("--skip-unknown-version-error", default=False, action=argparse.BooleanOptionalAction,
+		help="Skips the UnknownVersionTypeError termination as a temporary fix if a new version type gets added.")
 	args = parser.parse_args()
 
 	args.client = Client[args.client]