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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
#!/usr/bin/env python3
"""
simple client for que.run
"""
import argparse
import configparser
import os
import subprocess
import sys
import time
import urllib.parse
import urllib.request as request
MAX_TIMEOUT = 99999999 # basically never timeout
def auth():
conf_file = os.path.expanduser("~/.config/que.conf")
if not os.path.exists(conf_file):
sys.exit("you need a ~/.config/que.conf")
cfg = configparser.ConfigParser()
cfg.read(conf_file)
return cfg
def send(args):
"Send a message to the que."
ns = args.target.split("/")[0]
key = auth()[ns]["key"]
data = args.infile.read().encode("utf-8").strip()
req = request.Request(f"{args.host}/{args.target}")
req.add_header("User-AgenT", "Que/Client")
if key:
req.add_header("Authorization", key)
if args.serve:
while not time.sleep(1):
with request.urlopen(req, data=data, timeout=MAX_TIMEOUT) as req:
pass
else:
with request.urlopen(req, data=data, timeout=MAX_TIMEOUT) as req:
pass
def recv(args):
"Receive a message from the que."
ns = args.target.split("/")[0]
key = auth()[ns]["key"]
def _recv(_req):
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
)
params = urllib.parse.urlencode({"poll": args.poll})
req = request.Request(f"{args.host}/{args.target}?{params}")
req.add_header("User-Agent", "Que/Client")
if key:
req.add_header("Authorization", key)
with request.urlopen(req) as _req:
if args.poll:
while not time.sleep(1):
_recv(_req)
else:
_recv(_req)
def get_args():
cli = argparse.ArgumentParser(description=__doc__)
cli.add_argument(
"--host", default="http://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=" ".join(
[
"when polling, run this shell command after each response,",
"presumably for side effects,"
"replacing '\que' with the target and '\msg' with the body of the response",
]
),
)
cli.add_argument(
"--serve",
default=False,
action="store_true",
help=" ".join(
[
"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",
)
return cli.parse_args()
if __name__ == "__main__":
try:
args = get_args()
if args.infile:
send(args)
else:
recv(args)
except KeyboardInterrupt:
sys.exit(0)
except urllib.error.HTTPError as e:
print(e)
sys.exit(1)
|
