106 lines
2.4 KiB
Python
106 lines
2.4 KiB
Python
|
Import("env")
|
||
|
|
import serial, time, sys, glob
|
||
|
|
|
||
|
|
#
|
||
|
|
# Upload actions
|
||
|
|
#
|
||
|
|
|
||
|
|
portName = False
|
||
|
|
|
||
|
|
def before_upload(source, target, env):
|
||
|
|
print("\n\nSearching for a Smartcitizen kit...")
|
||
|
|
myPort = selectPort(serialPorts())
|
||
|
|
if myPort:
|
||
|
|
print("Asking for upload bridge...")
|
||
|
|
myPort.write("")
|
||
|
|
myPort.write("esp -flash\n")
|
||
|
|
env.Replace(UPLOAD_PORT=portName)
|
||
|
|
time.sleep(1)
|
||
|
|
|
||
|
|
def after_upload(source, target, env):
|
||
|
|
print("All good!!!")
|
||
|
|
global portName
|
||
|
|
|
||
|
|
print("Current build targets", map(str, BUILD_TARGETS))
|
||
|
|
|
||
|
|
env.AddPreAction("upload", before_upload)
|
||
|
|
env.AddPostAction("upload", after_upload)
|
||
|
|
|
||
|
|
env.AddPreAction("uploadfs", before_upload)
|
||
|
|
env.AddPostAction("uploadfs", after_upload)
|
||
|
|
|
||
|
|
|
||
|
|
#
|
||
|
|
# Custom actions when building program/firmware
|
||
|
|
#
|
||
|
|
|
||
|
|
# env.AddPreAction("buildprog", callback...)
|
||
|
|
# env.AddPostAction("buildprog", callback...)
|
||
|
|
|
||
|
|
# #
|
||
|
|
# # Custom actions for specific files/objects
|
||
|
|
# #
|
||
|
|
|
||
|
|
# env.AddPreAction("$BUILD_DIR/firmware.elf", [callback1, callback2,...])
|
||
|
|
# env.AddPostAction("$BUILD_DIR/firmware.hex", callback...)
|
||
|
|
|
||
|
|
# # custom action before building SPIFFS image. For example, compress HTML, etc.
|
||
|
|
# env.AddPreAction("$BUILD_DIR/spiffs.bin", callback...)
|
||
|
|
|
||
|
|
# # custom action for project's main.cpp
|
||
|
|
# env.AddPostAction("$BUILD_DIR/src/main.cpp.o", callback...)
|
||
|
|
|
||
|
|
|
||
|
|
def serialPorts():
|
||
|
|
"""Lists serial ports
|
||
|
|
|
||
|
|
:raises EnvironmentError:
|
||
|
|
On unsupported or unknown platforms
|
||
|
|
:returns:
|
||
|
|
A list of available serial ports
|
||
|
|
"""
|
||
|
|
if sys.platform.startswith('win'):
|
||
|
|
ports = ['COM' + str(i + 1) for i in range(256)]
|
||
|
|
|
||
|
|
elif sys.platform.startswith('linux') or sys.platform.startswith('cygwin'):
|
||
|
|
# this is to exclude your current terminal "/dev/tty"
|
||
|
|
ports = glob.glob('/dev/tty[A-Za-z]*')
|
||
|
|
|
||
|
|
elif sys.platform.startswith('darwin'):
|
||
|
|
ports = glob.glob('/dev/tty.*')
|
||
|
|
|
||
|
|
else:
|
||
|
|
raise EnvironmentError('Unsupported platform')
|
||
|
|
|
||
|
|
result = []
|
||
|
|
|
||
|
|
for port in ports:
|
||
|
|
try:
|
||
|
|
s = serial.Serial(port)
|
||
|
|
s.close()
|
||
|
|
result.append(port)
|
||
|
|
except (OSError, serial.SerialException):
|
||
|
|
pass
|
||
|
|
return result
|
||
|
|
|
||
|
|
def selectPort(ports):
|
||
|
|
|
||
|
|
for port in ports:
|
||
|
|
try:
|
||
|
|
s = serial.Serial(port)
|
||
|
|
for i in range(3):
|
||
|
|
s.write('\r\n\r\n')
|
||
|
|
time.sleep(0.2)
|
||
|
|
response = s.read(s.in_waiting)
|
||
|
|
if 'SCK' in response or 'Sdcard' in response:
|
||
|
|
print('Smartcitizen kit found on ' + port)
|
||
|
|
global portName
|
||
|
|
portName = port
|
||
|
|
return s
|
||
|
|
s.close()
|
||
|
|
except (OSError, serial.SerialException):
|
||
|
|
pass
|
||
|
|
|
||
|
|
print('No Smartcitizen kit found, please check your USB connection')
|
||
|
|
sys.exit(-1)
|