56 lines
1.5 KiB
Python
Executable File
56 lines
1.5 KiB
Python
Executable File
#!/usr/bin/python3
|
|
import argparse
|
|
import logging
|
|
import os
|
|
|
|
FORMAT = '%(levelname)s: %(message)s'
|
|
logging.basicConfig(level=logging.INFO, format=FORMAT)
|
|
parser = argparse.ArgumentParser(description='Create CVITEK device image')
|
|
|
|
|
|
def parse_Args():
|
|
cur_dir = os.path.abspath(os.getcwd())
|
|
parser.add_argument(
|
|
'--image_dir',
|
|
metavar='path',
|
|
type=str,
|
|
default=cur_dir,
|
|
help='the folder path to dir inclued fip,rootfs kernel and xml')
|
|
parser.add_argument("-v",
|
|
"--verbose",
|
|
help="increase output verbosity",
|
|
action="store_true")
|
|
group = parser.add_mutually_exclusive_group()
|
|
group.add_argument('--serial', action='store_true', default=False)
|
|
group.add_argument('--libusb', action='store_true', default=False)
|
|
|
|
args = parser.parse_args()
|
|
|
|
return args
|
|
|
|
|
|
def main():
|
|
args = parse_Args()
|
|
image_dir = args.image_dir
|
|
|
|
if (not args.serial and not args.libusb) or args.serial:
|
|
driver = "serial"
|
|
else:
|
|
driver = "libusb"
|
|
logging.info("Using %s" % driver)
|
|
|
|
logging.info("cv182x USB download start\n")
|
|
cmd = "python rom_usb_dl/cv182x_rom_usb_download.py --image_dir " + image_dir
|
|
result = os.system(cmd)
|
|
|
|
if result:
|
|
exit(-1)
|
|
|
|
cmd = "python rom_usb_dl/cv182x_uboot_usb_download.py --image_dir " + image_dir + " --" + driver
|
|
os.system(cmd)
|
|
logging.info("cv182x USB download end\n")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|