Permalink
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
106 lines (80 sloc) 2.44 KB
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)