40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import json
|
||
|
|
import requests
|
||
|
|
|
||
|
|
def discovery(url="http://localhost:8898"):
|
||
|
|
ret = {"data": []}
|
||
|
|
req = requests.request(method="GET", url=f"{url}/v1/jobs")
|
||
|
|
try:
|
||
|
|
for res_value in req.json():
|
||
|
|
ret["data"].append({"{#SERVICE}": res_value["name"]})
|
||
|
|
return json.dumps(ret)
|
||
|
|
except:
|
||
|
|
return "error"
|
||
|
|
|
||
|
|
def task(url="http://localhost:8898", task_name=""):
|
||
|
|
ret = ""
|
||
|
|
try:
|
||
|
|
req = requests.request(method="GET", url=f"{url}/v1/jobs/{task_name}")
|
||
|
|
ret = req.json()["status"]
|
||
|
|
except:
|
||
|
|
return "error"
|
||
|
|
return ret
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
parser = argparse.ArgumentParser(prog="dkron")
|
||
|
|
subparser = parser.add_subparsers(help='sub-command help', dest='option')
|
||
|
|
parser_discovery = subparser.add_parser("discovery")
|
||
|
|
parser_service = subparser.add_parser("status")
|
||
|
|
parser_service.add_argument("task")
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
if args.option == "discovery":
|
||
|
|
disc = discovery()
|
||
|
|
print(disc)
|
||
|
|
elif args.option == "status":
|
||
|
|
t = task(task_name=args.task)
|
||
|
|
print(t)
|