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

diff --git a/firmware/hantek-dso-extract.py b/firmware/hantek-dso-extract.py
index faa7712..045e50a 100755
--- a/firmware/hantek-dso-extract.py
+++ b/firmware/hantek-dso-extract.py
@@ -1,6 +1,6 @@
 #!/usr/bin/python3
 ##
-## This file is part of the sigrok project.
+## This file is part of the sigrok-util project.
 ##
 ## Copyright (C) 2012 Bert Vermeulen <bert@biot.com>
 ##
@@ -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'