def __init__(self, args):
self.path = args[0]
if len(args) > 2:
self.args = urlencode(parse_qsl(args[1]))
self.method = args[2]
elif len(args) == 2:
if args[1].upper() in ['GET', 'POST']:
self.args = ''
self.method = args[1]
else:
self.args = urlencode(parse_qsl(args[1]))
self.method = 'GET'
else:
self.args = ''
self.method = 'GET'
def del_queue(name, port='8089', host='127.0.0.1'):
data = {'name': name, 'opt': 'del', 'cli': 'true'}
url = 'http://%s:%s/taskq?%s' % (host, port, urlencode(data))
try:
resp = request.urlopen(url)
return resp.read().decode()
except (URLError):
raise Exception('Connection refused. Please check host or port.')
def __call__(self, parser, namespace, values, option_string=None):
port, name = values
data = {'name': name, 'opt': 'cls', 'cli': 'true'}
url = 'http://127.0.0.1:%s/taskq?%s' % (port, urlencode(data))
try:
resp = request.urlopen(url)
print(resp.read().decode())
except(URLError):
print('Connection refused. Please check port.')
exit(0)
def add_task():
path = request.json.get('path')
args = request.json.get('args')
method = request.json.get('method')
name = request.json.get('name')
json = request.json.get('json')
if args or json:
args = dumps(args or json) if json else urlencode(dict_encode(args))
rowcount = db.task.push(path, args, method, name, int(bool(json)))
return 'OK' if rowcount > 0 else 'Not Modified'
def __call__(self, parser, namespace, values, option_string=None):
port, name, mode = values
if mode not in ['con', 'seq']:
print('The mode must be either con or seq.')
exit(1)
if not name.strip():
print('The name is invalid.')
exit(1)
data = {'name': name, 'mode': mode, 'opt': 'add', 'cli': 'true'}
url = 'http://127.0.0.1:%s/taskq?%s' % (port, urlencode(data))
try:
resp = request.urlopen(url)
print(resp.read().decode())
except(URLError):
print('Connection refused. Please check port.')
exit(0)