From 8158df842404dd812f4eb857b6b9014c5ddab3c9 Mon Sep 17 00:00:00 2001
From: Alex Koren <AlexKoren@actiview.local>
Date: Fri, 4 Nov 2016 11:40:42 -0700
Subject: [PATCH] added 2.4" Nextion uploader

---
 Nextion/nextion.py | 97 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 Nextion/nextion.py

diff --git a/Nextion/nextion.py b/Nextion/nextion.py
new file mode 100644
index 0000000..3b201cb
--- /dev/null
+++ b/Nextion/nextion.py
@@ -0,0 +1,97 @@
+import serial
+import time
+import sys
+import os
+
+e = "\xff\xff\xff"
+deviceName = '/dev/ttyUSB0' #MAKE SURE THIS IS THE CORRECT DEVICE
+CHECK_MODEL = 'NX3224T024'
+
+def getBaudrate(ser, fSize=None):
+    for baudrate in (2400, 4800, 9600, 19200, 38400, 57600, 115200):
+        ser.baudrate = baudrate
+        ser.timeout = 3000 / baudrate + .2
+        print 'Trying with baudrate: ' + str(baudrate) + '...'
+        ser.write("\xff\xff\xff")
+        ser.write('connect')
+        ser.write("\xff\xff\xff")
+        r = ser.read(128)
+        if 'comok' in r:
+            print 'Connected with baudrate: ' + str(baudrate) + '...'
+            noConnect = False
+            status, unknown1, model, unknown2, version, serial, flashSize = r.strip("\xff\x00").split(',')
+            print 'Status: ' + status
+            print 'Model: ' + model
+            print 'Version: ' + version
+            print 'Serial: ' + serial
+            print 'Flash size: ' + flashSize
+            if fSize and fSize > flashSize:
+                print 'File too big!'
+                return False
+            if not CHECK_MODEL in model:
+                print 'Wrong Display!'
+                return False
+            return True
+    return False
+
+def setDownloadBaudrate(ser, fSize, baudrate):
+    ser.write("")
+    ser.write("whmi-wri " + str(fSize) + "," + str(baudrate) + ",0" + e)
+    time.sleep(.05)
+    ser.baudrate = baudrate
+    ser.timeout = .5
+    r = ser.read(1)
+    if "\x05" in r:
+        return True
+    return False
+
+def transferFile(ser, filename, fSize):
+    with open(filename, 'rb') as hmif:
+        dcount = 0
+        while True:
+            data = hmif.read(4096)
+            if len(data) == 0:
+                break
+            dcount += len(data)
+            ser.timeout = 5
+            ser.write(data)
+            sys.stdout.write('\rDownloading, %3.1f%%...'% (dcount / float(fSize) * 100.0))
+            sys.stdout.flush()
+            ser.timeout = .5
+            time.sleep(.5)
+            r = ser.read(1)
+            if "\x05" in r:
+                continue
+            else:
+                break
+        print
+    return True
+
+def upload(ser, filename):
+    if not getBaudrate(ser, os.path.getsize(filename)):
+        print 'could not find baudrate'
+        exit(1)
+
+    if not setDownloadBaudrate(ser, os.path.getsize(filename), 115200):
+        print 'could not set download baudrate'
+        exit(1)
+
+    if not transferFile(ser, filename, os.path.getsize(filename)):
+        print 'could not transfer file'
+        exit(1)
+
+    exit(0)
+
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print 'usage:\npython nextion.py file_to_upload.tft'
+        exit(1)
+
+    ser = serial.Serial(deviceName, 9600, timeout=5)
+    if not ser:
+        print 'could not open device'
+        exit(1)
+    if not ser.is_open:
+        ser.open()
+
+    upload(ser, sys.argv[1])