65 lines
1.9 KiB
Python
Executable File
65 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import click
|
|
|
|
from helpers import CliContext, Logger, EepromProgrammer
|
|
|
|
|
|
@click.group()
|
|
@click.option('--device', '-d', default='/dev/ttyUSB0', show_default=True,
|
|
metavar='DEVICE', help="Set the serial device")
|
|
@click.option('--baud', '-b', default=38400, show_default=True,
|
|
metavar='BAUDRATE', help="Set the baud rate of the serial device")
|
|
@click.option('--verbose', '-v', is_flag=True, help='Print debug output')
|
|
@click.pass_context
|
|
def eepprog(ctx: click.Context, device: str, baud: int, verbose: bool):
|
|
# Create dependencies
|
|
logger = Logger(verbose=verbose)
|
|
eeprom_programmer = EepromProgrammer(logger=logger, device=device, baudrate=baud)
|
|
|
|
logger.debug("Creating CLI context (device: {}, bauds: {})".format(device, baud))
|
|
|
|
# Create CLI context
|
|
ctx.obj = CliContext(
|
|
logger=logger,
|
|
eeprom_programmer=eeprom_programmer,
|
|
)
|
|
|
|
# Define clean up functions
|
|
ctx.call_on_close(eeprom_programmer.close)
|
|
|
|
|
|
@eepprog.command('hello', short_help='Say hello. :)')
|
|
@click.pass_obj
|
|
def hello(context: CliContext):
|
|
"""
|
|
Say hello. :)
|
|
|
|
Just a test command that tests the logger.
|
|
"""
|
|
context.logger.debug('hello hello', 413, 'hellooo')
|
|
context.logger.info('hello!')
|
|
context.logger.info('the same', click.style('but in green!', fg='green'))
|
|
context.logger.warn('this is fine')
|
|
context.logger.error('ohno')
|
|
context.logger.error('error, but a friendly one :)', fg='green')
|
|
context.logger.success('yay!')
|
|
|
|
|
|
@eepprog.command('test', short_help="Send INIT command to the programmer and read answer")
|
|
@click.pass_obj
|
|
def test(context: CliContext):
|
|
"""
|
|
Send an 'INIT' command to the programmer and read the answer.
|
|
|
|
Test command for debugging.
|
|
"""
|
|
context.eeprom_programmer.test()
|
|
|
|
# TODO command: list-devices -> serial.tools.list_ports
|
|
# TODO shell: Run an interactive shell
|
|
|
|
|
|
if __name__ == '__main__':
|
|
eepprog()
|