Add check for correct model number of display

This commit is contained in:
phl0
2016-11-06 22:55:15 +01:00
parent b85d702224
commit 3f6a34a8b5

View File

@@ -20,6 +20,7 @@ import serial
import time
import sys
import os
import re
e = "\xff\xff\xff"
@@ -116,4 +117,8 @@ if __name__ == "__main__":
checkModel = None
if len(sys.argv) == 4:
checkModel = sys.argv[3]
pattern = re.compile("^NX\d{4}[TK]\d{3}$")
if not pattern.match(checkModel):
print 'Invalid model name. Please give a correct one (e.g. NX3224T024)'
exit(1)
upload(ser, sys.argv[1], checkModel)