X-Git-Url: https://sigrok.org/gitweb/?p=sigrok-util.git;a=blobdiff_plain;f=firmware%2Fhantek-dso-extract.py;h=faa771231d694a1ff999f64eec7f1d8a088b9bfd;hp=e99236bef3ea2dc7f5eaa7993251044e0460f3a4;hb=aba15c6379086057921e760758cb96ac7cdec744;hpb=683fe750793df2f56a205ddb3294ddf339b658a6

diff --git a/firmware/hantek-dso-extract.py b/firmware/hantek-dso-extract.py
index e99236b..faa7712 100755
--- a/firmware/hantek-dso-extract.py
+++ b/firmware/hantek-dso-extract.py
@@ -29,12 +29,9 @@ import parsepe
 
 def find_model(filename):
 	filename = os.path.split(filename)[-1]
-	m = re.search('^dso([a-z0-9]+)1.sys$', filename, re.I)
+	m = re.search('^dso([25])[a-z0-9]+1.sys$', filename, re.I)
 	if m:
-		model = m.group(1).upper()
-		model = model.replace('X86', '').replace('AMD64', '').replace('IA64', '')
-		if model == '520A':
-			model = '5200A'
+		model = m.group(1) + 'xxx'
 	else:
 		model = 'unknown'