1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#!/usr/bin/env python3
"""
simple client for que.run
"""
import argparse
import urllib.request as request
import urllib.parse
import time
import subprocess
import sys
# set to something ridiculously high so we don't run into timeouts while polling
# or waiting for a message
MAX_TIMEOUT = 100000000
def main(argv=None):
cli = argparse.ArgumentParser(description=__doc__)
cli.add_argument(
"--host", default="https://que.run", help="where que-server is running"
)
cli.add_argument(
"--poll", default=False, action="store_true", help="stream data from the que"
)
cli.add_argument(
"--then",
help="when polling, run this shell command after each response, replacing '\que' with the target and '\msg' with the body of the response",
)
cli.add_argument(
"--serve",
default=False,
action="store_true",
help="when posting to the que, do so continuously in a loop. this can be used for serving a webpage or other file continuously",
)
cli.add_argument(
"target", help="namespace and path of the que, like 'ns/path/subpath'"
)
cli.add_argument(
"infile",
nargs="?",
type=argparse.FileType("r"),
help="data to put on the que. Use '-' for stdin, otherwise should be a readable file",
)
if argv is None:
args = cli.parse_args()
else:
args = cli.parse_args(argv)
if args.infile:
# send input data
data = args.infile.read().encode("utf-8").strip()
if args.serve:
# loop until ^C
while not time.sleep(1):
with request.urlopen(
f"{args.host}/{args.target}", data=data, timeout=MAX_TIMEOUT
) as req:
pass
else:
with request.urlopen(
f"{args.host}/{args.target}", data=data, timeout=MAX_TIMEOUT
) as req:
pass
else:
# no input data, do a read instead
params = urllib.parse.urlencode({"poll": args.poll})
url = f"{args.host}/{args.target}?{params}"
with request.urlopen(url) as req:
if args.poll:
while not time.sleep(1):
msg = req.readline().decode("utf-8").strip()
print(msg)
if args.then:
subprocess.run(
args.then.replace("\msg", msg).replace("\que", args.target),
shell=True,
)
else:
msg = req.read().decode("utf-8").strip()
print(msg)
if args.then:
subprocess.run(
args.then.replace("\msg", msg).replace("\que", args.target),
shell=True,
)
if __name__ == "__main__":
try:
main()
except KeyboardInterrupt:
print("Interrupted")
sys.exit(0)
except urllib.error.HTTPError as e:
print(e)
sys.exit(1)
except http.client.IncompleteRead as e:
print(e)
sys.exit(1)
|
