100 lines
2.9 KiB
Python
100 lines
2.9 KiB
Python
#!/usr/bin/env python
|
|
import argparse
|
|
import ast
|
|
import json
|
|
import os
|
|
from os.path import isfile, join
|
|
import re
|
|
import subprocess
|
|
import tempfile
|
|
import time
|
|
|
|
# from colorama import init, Fore, Back, Style
|
|
|
|
# Default flavor json directory
|
|
IMAGE_DIR = './image_dir'
|
|
CLI_PATH = '../ormcli/orm'
|
|
|
|
|
|
def read_jsonfile(file):
|
|
return json.loads(open(file).read())
|
|
|
|
|
|
def sh(cmd):
|
|
# run a shell command, echoing output, painting error lines red,
|
|
# print runtime and status
|
|
# return status and output
|
|
|
|
print '>> Starting: ' + cmd
|
|
start = time.time()
|
|
output = ''
|
|
errpat = re.compile('error', re.I)
|
|
p = subprocess.Popen(cmd.split(), shell=False, stdout=subprocess.PIPE)
|
|
for line in iter(p.stdout.readline, b''):
|
|
out = line.rstrip()
|
|
print(">>> " + out)
|
|
output += out
|
|
end = time.time()
|
|
span = end - start
|
|
retcode = p.wait()
|
|
print '>> Ended: %s [%s, %d:%02d]' % (cmd, retcode, span / 60, span % 60)
|
|
return retcode, output
|
|
|
|
|
|
parser = argparse.ArgumentParser(prog='regionator',
|
|
description='batch add region to image')
|
|
# parser.add_argument('--image_dir',
|
|
# type=str,
|
|
# default='./image_dir',
|
|
# help='<JSON image directory, default: ./image_dir>')
|
|
# parser.add_argument('--host',
|
|
# type=str,
|
|
# help='<orm host ip>')
|
|
# parser.add_argument('--cli_command',
|
|
# type=str,
|
|
# default='/opt/app/orm/ormcli/ormcli/orm',
|
|
# help='<path to cli command>')
|
|
parser.add_argument('regions',
|
|
type=str,
|
|
default='',
|
|
help='<comma-separated regions to add, e.g. region1,'
|
|
'region2>')
|
|
args = parser.parse_args()
|
|
|
|
regions = args.regions.split(',')
|
|
if not regions:
|
|
print "Must specify at least one region"
|
|
exit(0)
|
|
data = {'regions': [{'name': r} for r in regions]}
|
|
fh, file_name = tempfile.mkstemp()
|
|
os.write(fh, json.dumps(data))
|
|
os.close(fh)
|
|
|
|
# Prepare images dict with pairs {image_name:image_id}
|
|
img_dict = {}
|
|
# harg = '--orm-base-url %s' % args.host if args.host else ''
|
|
res, output = sh(
|
|
'python %s ims %s list_images test ' % (CLI_PATH, ''))
|
|
if not res:
|
|
images = ast.literal_eval(output)
|
|
for img in images['images']:
|
|
img_dict[img['name']] = img['id']
|
|
print img_dict
|
|
|
|
for file in [f for f in os.listdir(IMAGE_DIR) if
|
|
isfile(join(IMAGE_DIR, f))]:
|
|
f = read_jsonfile(join(IMAGE_DIR, file))
|
|
|
|
print f
|
|
image_name = f['name']
|
|
if image_name in img_dict:
|
|
image_id = img_dict[image_name]
|
|
print 'image_id: ' + image_id
|
|
res, output = sh('python %s ims add_regions test %s %s' % (
|
|
CLI_PATH, image_id, file_name))
|
|
else:
|
|
print 'python image_name: {} does not exist. ' \
|
|
'ignore.'.format(image_name)
|
|
|
|
os.unlink(file_name)
|