def test_ft232h(self):
with open('pyftdi/tests/resources/ft232h_x2.yaml', 'rb') as yfp:
self.loader.load(yfp)
eeprom = FtdiEeprom()
eeprom.open('ftdi://::FT1ABC1/1', ignore=True)
eeprom.erase()
eeprom.initialize()
# default EEPROM config does not have any CBUS configured as GPIO
self.assertEqual(eeprom.cbus_pins, [])
self.assertEqual(eeprom.cbus_mask, 0)
eeprom.set_property('cbus_func_6', 'gpio')
eeprom.set_property('cbus_func_9', 'gpio')
# not yet committed
self.assertEqual(eeprom.cbus_pins, [])
self.assertEqual(eeprom.cbus_mask, 0)
eeprom.sync()
# committed
self.assertEqual(eeprom.cbus_pins, [6, 9])
self.assertEqual(eeprom.cbus_mask, 0xA)
eeprom.set_property('cbus_func_5', 'gpio')
eeprom.set_property('cbus_func_8', 'gpio')
eeprom.sync()
self.assertEqual(eeprom.cbus_pins, [5, 6, 8, 9])
self.assertEqual(eeprom.cbus_mask, 0xF)
# pin1 and pin3 is not configurable as GPIO
self.assertRaises(ValueError, eeprom.set_property, 'cbus_func_1',
'gpio')
self.assertRaises(ValueError, eeprom.set_property, 'cbus_func_3',
'gpio')
eeprom.close()
def test_ft230x(self):
with open('pyftdi/tests/resources/ft230x.yaml', 'rb') as yfp:
self.loader.load(yfp)
eeprom = FtdiEeprom()
eeprom.open('ftdi:///1')
# default EEPROM config does not have any CBUS configured as GPIO
self.assertEqual(eeprom.cbus_pins, [])
self.assertEqual(eeprom.cbus_mask, 0)
# enable CBUS1 and CBUS3 as GPIO
eeprom.set_property('cbus_func_1', 'gpio')
eeprom.set_property('cbus_func_3', 'gpio')
eeprom.sync()
self.assertEqual(eeprom.cbus_pins, [1, 3])
self.assertEqual(eeprom.cbus_mask, 0xA)
# enable CBUS0 and CBUS2 as GPIO
eeprom.set_property('cbus_func_0', 'gpio')
eeprom.set_property('cbus_func_2', 'gpio')
# not yet committed
self.assertEqual(eeprom.cbus_pins, [1, 3])
self.assertEqual(eeprom.cbus_mask, 0xA)
eeprom.sync()
# committed
self.assertEqual(eeprom.cbus_pins, [0, 1, 2, 3])
self.assertEqual(eeprom.cbus_mask, 0xF)
# invalid CBUS pin
self.assertRaises(ValueError, eeprom.set_property,
'cbus_func_4', 'gpio')
# invalid pin function
self.assertRaises(ValueError, eeprom.set_property,
'cbus_func_0', 'gpio_')
# invalid pin
self.assertRaises(ValueError, eeprom.set_property,
'cbus_func', 'gpio')
# valid alternative mode
eeprom.set_property('cbus_func_0', 'txled')
eeprom.set_property('cbus_func_1', 'rxled')
eeprom.sync()
self.assertEqual(eeprom.cbus_pins, [2, 3])
self.assertEqual(eeprom.cbus_mask, 0xC)
eeprom.close()
def main():
"""Main routine"""
debug = False
try:
argparser = ArgumentParser(description=modules[__name__].__doc__)
argparser.add_argument('device',
nargs='?',
default='ftdi:///?',
help='serial port device name')
argparser.add_argument('-x',
'--hexdump',
action='store_true',
help='dump EEPROM content as ASCII')
argparser.add_argument('-X',
'--hexblock',
type=int,
help='dump EEPROM as indented hexa blocks')
argparser.add_argument('-i',
'--input',
type=FileType('rt'),
help='input ini file to load EEPROM content')
argparser.add_argument('-l',
'--load',
default='all',
choices=('all', 'raw', 'values'),
help='section(s) to load from input file')
argparser.add_argument('-o',
'--output',
type=FileType('wt'),
help='output ini file to save EEPROM content')
argparser.add_argument('-s',
'--serial-number',
help='set serial number')
argparser.add_argument('-m',
'--manufacturer',
help='set manufacturer name')
argparser.add_argument('-p', '--product', help='set product name')
argparser.add_argument('-c',
'--config',
action='append',
help='change/configure a property '
'as key=value pair')
argparser.add_argument('-e',
'--erase',
action='store_true',
help='erase the whole EEPROM content')
argparser.add_argument('-u',
'--update',
action='store_true',
help='perform actual update, use w/ care')
argparser.add_argument('-P',
'--vidpid',
action='append',
help='specify a custom VID:PID device ID, '
'may be repeated')
argparser.add_argument('-V',
'--virtual',
type=FileType('r'),
help='use a virtual device, specified as YaML')
argparser.add_argument('-v',
'--verbose',
action='count',
default=0,
help='increase verbosity')
argparser.add_argument('-d',
'--debug',
action='store_true',
help='enable debug mode')
args = argparser.parse_args()
debug = args.debug
if not args.device:
argparser.error('Serial device not specified')
loglevel = max(DEBUG, ERROR - (10 * args.verbose))
loglevel = min(ERROR, loglevel)
if debug:
formatter = Formatter(
'%(asctime)s.%(msecs)03d %(name)-20s '
'%(message)s', '%H:%M:%S')
else:
formatter = Formatter('%(message)s')
FtdiLogger.set_formatter(formatter)
FtdiLogger.set_level(loglevel)
FtdiLogger.log.addHandler(StreamHandler(stderr))
if args.virtual:
from pyftdi.usbtools import UsbTools
# Force PyUSB to use PyFtdi test framework for USB backends
UsbTools.BACKENDS = ('pyftdi.tests.backend.usbvirt', )
# Ensure the virtual backend can be found and is loaded
backend = UsbTools.find_backend()
loader = backend.create_loader()()
loader.load(args.virtual)
try:
add_custom_devices(Ftdi, args.vidpid)
except ValueError as exc:
argparser.error(str(exc))
eeprom = FtdiEeprom()
eeprom.open(args.device)
if args.erase:
eeprom.erase()
if args.input:
eeprom.load_config(args.input, args.load)
if args.serial_number:
eeprom.set_serial_number(args.serial_number)
if args.manufacturer:
eeprom.set_manufacturer_name(args.manufacturer)
if args.product:
eeprom.set_product_name(args.product)
for conf in args.config or []:
if conf == '?':
helpstr = ', '.join(sorted(eeprom.properties))
print(
fill(helpstr, initial_indent=' ', subsequent_indent=' '))
exit(1)
for sep in ':=':
if sep in conf:
name, value = conf.split(sep, 1)
if not value:
argparser.error('Configuration %s without value' %
conf)
helpio = StringIO()
eeprom.set_property(name, value, helpio)
helpstr = helpio.getvalue()
if helpstr:
print(
fill(helpstr,
initial_indent=' ',
subsequent_indent=' '))
exit(1)
break
else:
argparser.error('Missing name:value separator in %s' % conf)
if args.hexdump:
print(hexdump(eeprom.data))
if args.hexblock is not None:
indent = ' ' * args.hexblock
for pos in range(0, len(eeprom.data), 16):
hexa = ' '.join(
['%02x' % x for x in eeprom.data[pos:pos + 16]])
print(indent, hexa, sep='')
if args.update:
if eeprom.commit(False):
eeprom.reset_device()
if args.verbose > 0:
eeprom.dump_config()
if args.output:
eeprom.save_config(args.output)
except (ImportError, IOError, NotImplementedError, ValueError) as exc:
print('\nError: %s' % exc, file=stderr)
if debug:
print(format_exc(chain=False), file=stderr)
exit(1)
except KeyboardInterrupt:
exit(2)
