268 lines
9.4 KiB
Python
268 lines
9.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import copy
|
|
import json
|
|
import random
|
|
import uuid
|
|
|
|
from oslo_config import cfg
|
|
from oslo_log import log
|
|
from prettytable import PrettyTable
|
|
import requests
|
|
|
|
from faafo import version
|
|
|
|
|
|
LOG = log.getLogger('faafo.client')
|
|
CONF = cfg.CONF
|
|
|
|
|
|
def get_random_task():
|
|
random.seed()
|
|
|
|
if CONF.command.width:
|
|
width = int(CONF.command.width)
|
|
else:
|
|
width = random.randint(int(CONF.command.min_width),
|
|
int(CONF.command.max_width))
|
|
|
|
if CONF.command.height:
|
|
height = int(CONF.command.height)
|
|
else:
|
|
height = random.randint(int(CONF.command.min_height),
|
|
int(CONF.command.max_height))
|
|
|
|
if CONF.command.iterations:
|
|
iterations = int(CONF.command.iterations)
|
|
else:
|
|
iterations = random.randint(int(CONF.command.min_iterations),
|
|
int(CONF.command.max_iterations))
|
|
|
|
if CONF.command.xa:
|
|
xa = float(CONF.command.xa)
|
|
else:
|
|
xa = random.uniform(float(CONF.command.min_xa),
|
|
float(CONF.command.max_xa))
|
|
|
|
if CONF.command.xb:
|
|
xb = float(CONF.command.xb)
|
|
else:
|
|
xb = random.uniform(float(CONF.command.min_xb),
|
|
float(CONF.command.max_xb))
|
|
|
|
if CONF.command.ya:
|
|
ya = float(CONF.command.ya)
|
|
else:
|
|
ya = random.uniform(float(CONF.command.min_ya),
|
|
float(CONF.command.max_ya))
|
|
|
|
if CONF.command.yb:
|
|
yb = float(CONF.command.yb)
|
|
else:
|
|
yb = random.uniform(float(CONF.command.min_yb),
|
|
float(CONF.command.max_yb))
|
|
|
|
task = {
|
|
'uuid': str(uuid.uuid4()),
|
|
'width': width,
|
|
'height': height,
|
|
'iterations': iterations, 'xa': xa,
|
|
'xb': xb,
|
|
'ya': ya,
|
|
'yb': yb
|
|
}
|
|
|
|
return task
|
|
|
|
|
|
def do_get_fractal():
|
|
LOG.error("command 'download' not yet implemented")
|
|
|
|
|
|
def do_show_fractal():
|
|
LOG.info("showing fractal %s" % CONF.command.uuid)
|
|
result = requests.get("%s/v1/fractal/%s" %
|
|
(CONF.endpoint_url, CONF.command.uuid))
|
|
if result.status_code == 200:
|
|
data = json.loads(result.text)
|
|
output = PrettyTable(["Parameter", "Value"])
|
|
output.align["Parameter"] = "l"
|
|
output.align["Value"] = "l"
|
|
output.add_row(["uuid", data['uuid']])
|
|
output.add_row(["duration", "%f seconds" % data['duration']])
|
|
output.add_row(["dimensions", "%d x %d pixels" %
|
|
(data['width'], data['height'])])
|
|
output.add_row(["iterations", data['iterations']])
|
|
output.add_row(["xa", data['xa']])
|
|
output.add_row(["xb", data['xb']])
|
|
output.add_row(["ya", data['ya']])
|
|
output.add_row(["yb", data['yb']])
|
|
output.add_row(["size", "%d bytes" % data['size']])
|
|
output.add_row(["checksum", data['checksum']])
|
|
output.add_row(["generated_by", data['generated_by']])
|
|
print(output)
|
|
else:
|
|
LOG.error("fractal '%s' not found" % CONF.command.uuid)
|
|
|
|
|
|
def do_list_fractals():
|
|
LOG.info("listing all fractals")
|
|
|
|
fractals = get_fractals()
|
|
output = PrettyTable(["UUID", "Dimensions", "Filesize"])
|
|
for fractal in fractals:
|
|
output.add_row([
|
|
fractal["uuid"],
|
|
"%d x %d pixels" % (fractal["width"], fractal["height"]),
|
|
"%d bytes" % (fractal["size"] or 0),
|
|
])
|
|
print(output)
|
|
|
|
|
|
def get_fractals(page=1):
|
|
result = requests.get("%s/v1/fractal?page=%d" %
|
|
(CONF.endpoint_url, page))
|
|
|
|
fractals = []
|
|
if result.status_code == 200:
|
|
data = json.loads(result.text)
|
|
if page < data['total_pages']:
|
|
fractals = data['objects'] + get_fractals(page + 1)
|
|
else:
|
|
return data['objects']
|
|
|
|
return fractals
|
|
|
|
|
|
def do_delete_fractal():
|
|
LOG.info("deleting fractal %s" % CONF.command.uuid)
|
|
result = requests.delete("%s/v1/fractal/%s" %
|
|
(CONF.endpoint_url, CONF.command.uuid))
|
|
LOG.debug("result: %s" %result)
|
|
|
|
|
|
def do_create_fractal():
|
|
random.seed()
|
|
if CONF.command.tasks:
|
|
number = int(CONF.command.tasks)
|
|
else:
|
|
number = random.randint(int(CONF.command.min_tasks),
|
|
int(CONF.command.max_tasks))
|
|
LOG.info("generating %d task(s)" % number)
|
|
for i in xrange(0, number):
|
|
task = get_random_task()
|
|
LOG.debug("created task %s" % task)
|
|
# NOTE(berendt): only necessary when using requests < 2.4.2
|
|
headers = {'Content-type': 'application/json',
|
|
'Accept': 'text/plain'}
|
|
requests.post("%s/v1/fractal" % CONF.endpoint_url,
|
|
json.dumps(task), headers=headers)
|
|
|
|
|
|
def add_command_parsers(subparsers):
|
|
parser = subparsers.add_parser('create')
|
|
parser.set_defaults(func=do_create_fractal)
|
|
parser.add_argument("--height", default=None,
|
|
help="The height of the generate image.")
|
|
parser.add_argument("--min-height", default=256,
|
|
help="The minimum height of the generate image.")
|
|
parser.add_argument("--max-height", default=1024,
|
|
help="The maximum height of the generate image.")
|
|
parser.add_argument("--width", default=None,
|
|
help="The width of the generated image.")
|
|
parser.add_argument("--min-width", default=256,
|
|
help="The minimum width of the generated image.")
|
|
parser.add_argument("--max-width", default=1024,
|
|
help="The maximum width of the generated image.")
|
|
parser.add_argument("--iterations", default=None,
|
|
help="The number of iterations.")
|
|
parser.add_argument("--min-iterations", default=128,
|
|
help="The minimum number of iterations.")
|
|
parser.add_argument("--max-iterations", default=512,
|
|
help="The maximum number of iterations.")
|
|
parser.add_argument("--tasks", default=None,
|
|
help="The number of generated fractals.")
|
|
parser.add_argument("--min-tasks", default=1,
|
|
help="The minimum number of generated fractals.")
|
|
parser.add_argument("--max-tasks", default=10,
|
|
help="The maximum number of generated fractals.")
|
|
parser.add_argument("--xa", default=None,
|
|
help="The value for the parameter 'xa'.")
|
|
parser.add_argument("--min-xa", default=-1.0,
|
|
help="The minimum value for the parameter 'xa'.")
|
|
parser.add_argument("--max-xa", default=-4.0,
|
|
help="The maximum value for the parameter 'xa'.")
|
|
parser.add_argument("--xb", default=None,
|
|
help="The value for the parameter 'xb'.")
|
|
parser.add_argument("--min-xb", default=1.0,
|
|
help="The minimum value for the parameter 'xb'.")
|
|
parser.add_argument("--max-xb", default=4.0,
|
|
help="The maximum value for the parameter 'xb'.")
|
|
parser.add_argument("--ya", default=None,
|
|
help="The value for the parameter 'ya'.")
|
|
parser.add_argument("--min-ya", default=-0.5,
|
|
help="The minimum value for the parameter 'ya'.")
|
|
parser.add_argument("--max-ya", default=-3,
|
|
help="The maximum value for the parameter 'ya'.")
|
|
parser.add_argument("--yb", default=None,
|
|
help="The value for the parameter 'yb'.")
|
|
parser.add_argument("--min-yb", default=0.5,
|
|
help="The minimum value for the parameter 'yb'.")
|
|
parser.add_argument("--max-yb", default=3,
|
|
help="The maximum value for the parameter 'yb'.")
|
|
|
|
parser = subparsers.add_parser('delete')
|
|
parser.set_defaults(func=do_delete_fractal)
|
|
parser.add_argument("uuid", help="Fractal to delete.")
|
|
|
|
parser = subparsers.add_parser('show')
|
|
parser.set_defaults(func=do_show_fractal)
|
|
parser.add_argument("uuid", help="Fractal to show.")
|
|
|
|
parser = subparsers.add_parser('get')
|
|
parser.set_defaults(func=do_get_fractal)
|
|
parser.add_argument("uuid", help="Fractal to download.")
|
|
|
|
parser = subparsers.add_parser('list')
|
|
parser.set_defaults(func=do_list_fractals)
|
|
|
|
|
|
client_commands = cfg.SubCommandOpt('command', title='Commands',
|
|
help='Show available commands.',
|
|
handler=add_command_parsers)
|
|
|
|
CONF.register_cli_opts([client_commands])
|
|
|
|
client_cli_opts = [
|
|
cfg.StrOpt('endpoint-url',
|
|
default='http://localhost',
|
|
help='API connection URL')
|
|
]
|
|
|
|
CONF.register_cli_opts(client_cli_opts)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
log.register_options(CONF)
|
|
log.set_defaults()
|
|
|
|
CONF(project='client', prog='faafo-client',
|
|
version=version.version_info.version_string())
|
|
|
|
log.setup(CONF, 'client',
|
|
version=version.version_info.version_string())
|
|
|
|
CONF.command.func()
|