mirror of
https://github.com/YunoHost/yunorunner.git
synced 2024-09-03 20:05:52 +02:00
[enh] start to prototype a better cli tool
This commit is contained in:
parent
c5ef7b4856
commit
b5bf55265a
1 changed files with 70 additions and 0 deletions
70
cli.py
Normal file
70
cli.py
Normal file
|
@ -0,0 +1,70 @@
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import argh
|
||||||
|
import requests
|
||||||
|
|
||||||
|
from argh.decorators import named
|
||||||
|
|
||||||
|
|
||||||
|
def require_token():
|
||||||
|
if os.path.exists("token") and open("token").read().strip():
|
||||||
|
return
|
||||||
|
|
||||||
|
print("You need a token to be able to uses this command tool for security reasons, please refer to the README on how to add one https://github.com/YunoHost/yunorunner")
|
||||||
|
try:
|
||||||
|
token = input("Token: ").strip()
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
print()
|
||||||
|
token = None
|
||||||
|
|
||||||
|
if not token:
|
||||||
|
print("Error: you need to provide a valid token")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
open("token", "w").write(token)
|
||||||
|
|
||||||
|
|
||||||
|
def request_api(path, domain, verb, data):
|
||||||
|
assert verb in ("get", "post", "put", "delete")
|
||||||
|
|
||||||
|
https = False
|
||||||
|
if domain.split(":")[0] not in ("localhost", "127.0.0.1", "0.0.0.0"):
|
||||||
|
https = True
|
||||||
|
|
||||||
|
response = getattr(requests, verb)(
|
||||||
|
"http%s://%s/api/%s" % ("s" if https else "", domain, path),
|
||||||
|
headers={"X-Token": open("token", "r").read().strip()},
|
||||||
|
json=data,
|
||||||
|
)
|
||||||
|
|
||||||
|
# TODO: real error message
|
||||||
|
assert response.status_code == 200, response.content
|
||||||
|
assert response.content == b"ok", response.content
|
||||||
|
|
||||||
|
|
||||||
|
def add(name, url_or_path, test_type="stable", yunohost_version="unstable", debian_version="stretch", revision="master", domain="localhost:4242"):
|
||||||
|
request_api(
|
||||||
|
path="job",
|
||||||
|
verb="post",
|
||||||
|
domain=domain,
|
||||||
|
data={
|
||||||
|
"name": name,
|
||||||
|
"url_or_path": url_or_path,
|
||||||
|
"yunohost_version": yunohost_version,
|
||||||
|
"test_type": test_type,
|
||||||
|
"debian_version": debian_version,
|
||||||
|
"revision": revision,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@named("list")
|
||||||
|
def list_(all=False, domain="localhost:4242"): pass
|
||||||
|
def delete(job_id, domain="localhost:4242"): pass
|
||||||
|
def update(job_id, domain="localhost:4242"): pass
|
||||||
|
def cancel(job_id, domain="localhost:4242"): pass
|
||||||
|
def resume(job_id, domain="localhost:4242"): pass
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
require_token()
|
||||||
|
argh.dispatch_commands([add, list_, delete, update, cancel, resume])
|
Loading…
Add table
Reference in a new issue