123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270 |
- # -*- coding: utf-8 -*-
- # Copyright (C) 2015-2016 Peter Magnusson <peter@birchroad.net>
- """This module is the cli for the Uploader class"""
- import argparse
- import logging
- import os
- import glob
- from .uploader import Uploader
- from .term import terminal
- from serial import VERSION as serialversion
- log = logging.getLogger(__name__) # pylint: disable=C0103
- from .version import __version__
- def destination_from_source(sources):
- """
- Split each of the sources in the array on ':'
- First part will be source, second will be destination.
- Modifies the the original array to contain only sources
- and returns an array of destinations.
- """
- destinations = []
- newsources = []
- for i in range(0, len(sources)):
- srcdst = sources[i].split(':')
- if len(srcdst) == 2:
- destinations.append(srcdst[1])
- newsources[i] = srcdst[0]
- else:
- listing = glob.glob(srcdst[0])
- for filename in listing:
- newsources.append(filename)
- #always use forward slash at destination
- destinations.append(filename.replace('\\', '/'))
- return [newsources, destinations]
- def operation_upload(uploader, sources, verify, do_compile, do_file, do_restart):
- """The upload operation"""
- sources, destinations = destination_from_source(sources)
- if len(destinations) == len(sources):
- if uploader.prepare():
- for filename, dst in zip(sources, destinations):
- if do_compile:
- uploader.file_remove(os.path.splitext(dst)[0]+'.lc')
- uploader.write_file(filename, dst, verify)
- #init.lua is not allowed to be compiled
- if do_compile and dst != 'init.lua':
- uploader.file_compile(dst)
- uploader.file_remove(dst)
- if do_file:
- uploader.file_do(os.path.splitext(dst)[0]+'.lc')
- elif do_file:
- uploader.file_do(dst)
- else:
- raise Exception('Error preparing nodemcu for reception')
- else:
- raise Exception('You must specify a destination filename for each file you want to upload.')
- if do_restart:
- uploader.node_restart()
- log.info('All done!')
- def operation_download(uploader, sources):
- """The download operation"""
- destinations = destination_from_source(sources)
- if len(destinations) == len(sources):
- for filename, dst in zip(sources, destinations):
- uploader.read_file(filename, dst)
- else:
- raise Exception('You must specify a destination filename for each file you want to download.')
- log.info('All done!')
- def operation_file(uploader, cmd, filename=''):
- """File operations"""
- if cmd == 'list':
- uploader.file_list()
- if cmd == 'do':
- for path in filename:
- uploader.file_do(path)
- elif cmd == 'format':
- uploader.file_format()
- elif cmd == 'remove':
- for path in filename:
- uploader.file_remove(path)
- elif cmd == 'print':
- for path in filename:
- uploader.file_print(path)
- def arg_auto_int(value):
- """parsing function for integer arguments"""
- return int(value, 0)
- def main_func():
- """Main function for cli"""
- parser = argparse.ArgumentParser(
- description='NodeMCU Lua file uploader',
- prog='nodemcu-uploader'
- )
- parser.add_argument(
- '--verbose',
- help='verbose output',
- action='store_true',
- default=False)
- parser.add_argument(
- '--version',
- help='prints the version and exists',
- action='version',
- version='%(prog)s {version} (serial {serialversion})'.format(version=__version__, serialversion=serialversion)
- )
- parser.add_argument(
- '--port', '-p',
- help='Serial port device',
- default=Uploader.PORT)
- parser.add_argument(
- '--baud', '-b',
- help='Serial port baudrate',
- type=arg_auto_int,
- default=Uploader.BAUD)
- parser.add_argument(
- '--start_baud', '-B',
- help='Initial Serial port baudrate',
- type=arg_auto_int,
- default=Uploader.START_BAUD)
- parser.add_argument(
- '--timeout', '-t',
- help='Timeout for operations',
- type=arg_auto_int,
- default=Uploader.TIMEOUT)
- subparsers = parser.add_subparsers(
- dest='operation',
- help='Run nodemcu-uploader {command} -h for additional help')
- upload_parser = subparsers.add_parser(
- 'upload',
- help='Path to one or more files to be uploaded. Destination name will be the same as the file name.')
- upload_parser.add_argument(
- 'filename',
- nargs='+',
- help='Lua file to upload. Use colon to give alternate destination.'
- )
- upload_parser.add_argument(
- '--compile', '-c',
- help='If file should be uploaded as compiled',
- action='store_true',
- default=False
- )
- upload_parser.add_argument(
- '--verify', '-v',
- help='To verify the uploaded data.',
- action='store',
- nargs='?',
- choices=['none', 'raw', 'sha1'],
- default='none'
- )
- upload_parser.add_argument(
- '--dofile', '-e',
- help='If file should be run after upload.',
- action='store_true',
- default=False
- )
- upload_parser.add_argument(
- '--restart', '-r',
- help='If esp should be restarted',
- action='store_true',
- default=False
- )
- exec_parser = subparsers.add_parser(
- 'exec',
- help='Path to one or more files to be executed line by line.')
- exec_parser.add_argument('filename', nargs='+', help='Lua file to execute.')
- download_parser = subparsers.add_parser(
- 'download',
- help='Path to one or more files to be downloaded. Destination name will be the same as the file name.')
- download_parser.add_argument('filename',
- nargs='+',
- help='Lua file to download. Use colon to give alternate destination.')
- file_parser = subparsers.add_parser(
- 'file',
- help='File functions')
- file_parser.add_argument(
- 'cmd',
- choices=('list', 'do', 'format', 'remove', 'print'),
- help="list=list files, do=dofile given path, format=formate file area, remove=remove given path")
- file_parser.add_argument('filename', nargs='*', help='path for cmd')
- node_parse = subparsers.add_parser(
- 'node',
- help='Node functions')
- node_parse.add_argument('ncmd', choices=('heap', 'restart'), help="heap=print heap memory, restart=restart nodemcu")
- subparsers.add_parser(
- 'terminal',
- help='Run pySerials miniterm'
- )
- args = parser.parse_args()
- default_level = logging.INFO
- if args.verbose:
- default_level = logging.DEBUG
- #formatter = logging.Formatter('%(message)s')
- logging.basicConfig(level=default_level, format='%(message)s')
- if args.operation == 'terminal':
- #uploader can not claim the port
- terminal(args.port, str(args.start_baud))
- return
- # let uploader user the default (short) timeout for establishing connection
- uploader = Uploader(args.port, args.baud, start_baud=args.start_baud)
- # and reset the timeout (if we have the uploader&timeout)
- if args.timeout:
- uploader.set_timeout(args.timeout)
- if args.operation == 'upload':
- operation_upload(uploader, args.filename, args.verify, args.compile, args.dofile,
- args.restart)
- elif args.operation == 'download':
- operation_download(uploader, args.filename)
- elif args.operation == 'exec':
- sources = args.filename
- for path in sources:
- uploader.exec_file(path)
- elif args.operation == 'file':
- operation_file(uploader, args.cmd, args.filename)
- elif args.operation == 'node':
- if args.ncmd == 'heap':
- uploader.node_heap()
- elif args.ncmd == 'restart':
- uploader.node_restart()
- #no uploader related commands after this point
- uploader.close()
|