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

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