ardmk-init arg parse options and made V1.0
This commit is contained in:
John Whittington
parent
8819e761fb
commit
f3bb8249c1
1 changed files with 107 additions and 40 deletions
147
bin/ardmk-init
147
bin/ardmk-init
|
|
@ -7,8 +7,8 @@ import argparse
|
|||
from clint.textui import prompt, validators, colored, puts
|
||||
|
||||
## Global Vars
|
||||
VERSION = "0.1"
|
||||
directory = ""
|
||||
VERSION = "1.0"
|
||||
directory = os.getcwd()
|
||||
ard_template = "\n\
|
||||
#include <Arduino.h>\n\
|
||||
#include <Wire.h>\n\
|
||||
|
|
@ -22,48 +22,100 @@ void loop() {\n\
|
|||
"
|
||||
|
||||
## Command Parser
|
||||
parser = argparse.ArgumentParser(description='Arduino Makefile and boilerplate project generator. For use with Ard-Makefile')
|
||||
parser.add_argument('-v', '--verbose', action='store_true', help="Print file contents during creation")
|
||||
parser.add_argument('-d', '--directory', default='.', help='Directory to run generator')
|
||||
parser = argparse.ArgumentParser(description='Arduino Makefile and boilerplate project generator. For use with Ard-Makefile: https://github.com/sudar/Arduino-Makefile. Script created by John Whittington https://github.com/tuna-f1sh 2017\n\nVersion: ' + VERSION, usage='"ardmk-init -b uno -quiet -project" Automonously create boiler plate Arduino Uno project in current working directory with same name. See --help for more.')
|
||||
parser.add_argument('-v', '--verbose', action='store_true', help="print file contents during creation")
|
||||
parser.add_argument('-d', '--directory', default=directory, help='directory to run generator')
|
||||
parser.add_argument('-b', '--board', default='uno', help='board tag')
|
||||
parser.add_argument('-u', '--micro', default='AUTO', help='microcontroller on board')
|
||||
parser.add_argument('-f', '--freq', default='AUTO', help='mlock frequency')
|
||||
parser.add_argument('-p', '--port', default='AUTO', help='monitor port')
|
||||
parser.add_argument('-n', '--name', default=os.path.basename(directory), help='project name')
|
||||
parser.add_argument('-q', '--quiet', action='store_true', help='run quiet without user prompts')
|
||||
parser.add_argument('-P', '--project', action='store_true', help='create boilerplate project with src, lib and bin folder structure')
|
||||
parser.add_argument('-t', '--template', action='store_true', help='create bare minimum Arduino source file')
|
||||
args = parser.parse_args()
|
||||
|
||||
directory = args.directory
|
||||
|
||||
def generateMakefile():
|
||||
src_dir = ""
|
||||
|
||||
puts(colored.cyan("Generating Arduino Ard-Makefile project in " + os.path.abspath(directory)))
|
||||
puts(colored.red("Any existing Makefile in " + os.path.abspath(directory) + " will be overwritten!!"))
|
||||
# Header
|
||||
fileContents = "# Generated by ard-make version " + VERSION + "\n\n"
|
||||
btag = prompt.query('Board tag?', default='uno')
|
||||
if not btag == 'uno':
|
||||
bsub = prompt.query('Board sub micro?', default='atmega328')
|
||||
f_cpu = prompt.query('Board frequency', default='16000000L')
|
||||
|
||||
# Basic
|
||||
if not args.quiet:
|
||||
puts(colored.cyan("Generating Arduino Ard-Makefile project in " + os.path.abspath(directory)))
|
||||
btag = prompt.query('Board tag?', default='uno')
|
||||
if not btag == 'uno':
|
||||
bsub = prompt.query('Board sub micro?', default='atmega328')
|
||||
f_cpu = prompt.query('Board frequency', default='16000000L')
|
||||
else:
|
||||
bsub = 'AUTO'
|
||||
f_cpu = 'AUTO'
|
||||
monitor_port = prompt.query('Arduino port?', default='AUTO')
|
||||
else:
|
||||
bsub = 'AUTO'
|
||||
f_cpu = 'AUTO'
|
||||
monitor_port = prompt.query('Arduino port?', default='AUTO')
|
||||
btag = args.board
|
||||
bsub = args.micro
|
||||
f_cpu = args.freq
|
||||
monitor_port = args.port
|
||||
|
||||
fileContents += checkDefine('BOARD_TAG', btag)
|
||||
fileContents += checkDefine('BOARD_SUB', bsub)
|
||||
fileContents += checkDefine('F_CPU', f_cpu)
|
||||
fileContents += checkDefine('MONITOR_PORT', monitor_port)
|
||||
|
||||
if not prompt.yn('Extended options?', default='n'):
|
||||
if not prompt.yn('Define local folders?', default='n'):
|
||||
src_dir = prompt.query('Sources folder (Makefile will be created here)?', default='', validators=[])
|
||||
userlibs = prompt.query('Library folder (will create if does not exist) - AUTO is Sketchbook directory?', default='AUTO', validators=[])
|
||||
obj_dir = prompt.query('Output directory?', default='AUTO', validators=[])
|
||||
boards_txt = prompt.query('Boards file?', default='AUTO')
|
||||
isp_prog = prompt.query('ISP programmer?', default='atmelice_isp')
|
||||
isp_port = prompt.query('ISP port?', default='AUTO')
|
||||
if not prompt.yn('Quiet make?', default='n'):
|
||||
fileContents += "ARDUINO_QUIET = 1\n"
|
||||
# Extended
|
||||
if not args.quiet:
|
||||
if not prompt.yn('Extended options?', default='n'):
|
||||
if not prompt.yn('Define local folders?', default='n'):
|
||||
src_dir = prompt.query('Sources folder (Makefile will be created here)?', default='', validators=[])
|
||||
userlibs = prompt.query('Library folder (will create if does not exist) - AUTO is Sketchbook directory?', default='AUTO', validators=[])
|
||||
obj_dir = prompt.query('Output directory?', default='AUTO', validators=[])
|
||||
else:
|
||||
src_dir = ''
|
||||
userlibs = 'AUTO'
|
||||
obj_dir = 'AUTO'
|
||||
boards_txt = prompt.query('Boards file?', default='AUTO')
|
||||
isp_prog = prompt.query('ISP programmer?', default='atmelice_isp')
|
||||
isp_port = prompt.query('ISP port?', default='AUTO')
|
||||
if not prompt.yn('Quiet make?', default='n'):
|
||||
fileContents += "ARDUINO_QUIET = 1\n"
|
||||
|
||||
fileContents += checkDefine('ISP_PROG', isp_prog)
|
||||
fileContents += checkDefine('ISP_PORT', isp_port)
|
||||
fileContents += checkDefine('BOARDS_TXT', boards_txt)
|
||||
fileContents += checkDefine('ISP_PROG', isp_prog)
|
||||
fileContents += checkDefine('ISP_PORT', isp_port)
|
||||
fileContents += checkDefine('BOARDS_TXT', boards_txt)
|
||||
|
||||
# Check andd create folders
|
||||
checkCreateFolder(src_dir)
|
||||
checkCreateFolder(userlibs)
|
||||
checkCreateFolder(obj_dir)
|
||||
|
||||
# Makefile will be in src_dir so lib and bin must be relative
|
||||
if src_dir:
|
||||
userlibs = "../" + userlibs
|
||||
obj_dir = "../" + obj_dir
|
||||
|
||||
fileContents += checkDefine('USER_LIB_PATH', userlibs)
|
||||
fileContents += checkDefine('OBJDIR', obj_dir)
|
||||
else:
|
||||
src_dir = ''
|
||||
|
||||
if args.template or not prompt.yn('Create template Arduino source?', default='n'):
|
||||
sourceFilename = prompt.query('Name of project?', default=os.path.basename(os.getcwd()))
|
||||
if src_dir:
|
||||
writeTemplate(src_dir + "/" + sourceFilename)
|
||||
else:
|
||||
writeTemplate(sourceFilename)
|
||||
fileContents += checkDefine('TARGET', sourceFilename)
|
||||
|
||||
else:
|
||||
if args.project:
|
||||
src_dir = 'src'
|
||||
userlibs = 'lib'
|
||||
obj_dir = 'bin'
|
||||
else:
|
||||
src_dir = ''
|
||||
userlibs = 'AUTO'
|
||||
obj_dir = 'AUTO'
|
||||
|
||||
# Check andd create folders
|
||||
checkCreateFolder(src_dir)
|
||||
|
|
@ -78,18 +130,19 @@ def generateMakefile():
|
|||
fileContents += checkDefine('USER_LIB_PATH', userlibs)
|
||||
fileContents += checkDefine('OBJDIR', obj_dir)
|
||||
|
||||
if not prompt.yn('Create template Arduino source?', default='n'):
|
||||
sourceFilename = prompt.query('Name of project?', default='')
|
||||
if src_dir:
|
||||
writeTemplate(src_dir + "/" + sourceFilename)
|
||||
else:
|
||||
writeTemplate(sourceFilename)
|
||||
fileContents += checkDefine('TARGET', sourceFilename)
|
||||
if args.project or args.template:
|
||||
if src_dir:
|
||||
writeTemplate(src_dir + "/" + args.name)
|
||||
else:
|
||||
writeTemplate(args.name)
|
||||
fileContents += checkDefine('TARGET', args.name)
|
||||
|
||||
if not "ARDMK_DIR" in os.environ:
|
||||
ardmk = prompt.query('Arduino Makefile path?', default='/usr/share/arduino', validators=[validators.PathValidator()])
|
||||
ardmk = "ARDMK_DIR := " + ardmk + "\n"
|
||||
|
||||
if args.quiet:
|
||||
puts(colored.magenta('Warning: ARDMK_DIR environment variable not defined. Must be defined for Makefile to work'))
|
||||
else:
|
||||
ardmk = prompt.query('Arduino Makefile path?', default='/usr/share/arduino', validators=[validators.PathValidator()])
|
||||
ardmk = "ARDMK_DIR := " + ardmk + "\n"
|
||||
|
||||
fileContents += "\ninclude $(ARDMK_DIR)/Arduino.mk"
|
||||
|
||||
|
|
@ -110,8 +163,15 @@ def writeToMakefile(fileContents, path):
|
|||
makefile.close()
|
||||
|
||||
def writeTemplate(filename):
|
||||
src = open((filename + ".ino"),'w')
|
||||
puts(colored.cyan("Writing " + os.path.abspath(filename) + ".ino..."))
|
||||
if os.path.isfile(filename + '.ino'):
|
||||
if args.quiet:
|
||||
puts(colored.red(filename + '.ino' + ' already exists! Stopping.'))
|
||||
return
|
||||
puts(colored.red(filename + '.ino' + ' already exists! Overwrite?'))
|
||||
if prompt.yn('Continue?', default='n'):
|
||||
return
|
||||
src = open((filename + ".ino"),'w')
|
||||
if args.verbose:
|
||||
puts(colored.yellow(ard_template))
|
||||
src.write("/* Project: " + filename + " */\n" + ard_template)
|
||||
|
|
@ -134,6 +194,13 @@ def main():
|
|||
checkCreateFolder(directory)
|
||||
# Change to dir so all commands are run relative
|
||||
os.chdir(directory)
|
||||
if os.path.isfile('Makefile'):
|
||||
if args.quiet:
|
||||
puts(colored.red('Makefile in ' + os.path.abspath(directory) + ' already exists! Stopping.'))
|
||||
return
|
||||
puts(colored.red('Makefile in ' + os.path.abspath(directory) + ' already exists! Overwrite?'))
|
||||
if prompt.yn('Continue?', default='n'):
|
||||
return
|
||||
generateMakefile();
|
||||
|
||||
main()
|
||||
|
|
|
|||
Loading…
Reference in a new issue