asterisk-config/whois.py

141 lines
4.3 KiB
Python
Raw Normal View History

2021-11-19 15:52:42 +08:00
#!/usr/bin/python
import socket
import subprocess
from pathlib import Path
from sys import argv
import re
import shutil
from time import time
WHOIS_SERVER = ('172.20.129.8', 43)
ASTROOT = Path("/var/lib/asterisk/sounds")
ERRSOUND = ASTROOT / "en" / "something-terribly-wrong.gsm"
ROOTDIR = Path("/var/tmp")
OUTDIR = ROOTDIR / "ast-dynamic"
OUTDIR.mkdir(exist_ok=True)
2021-11-23 13:36:15 +08:00
def cleanup(*files):
2021-11-19 15:52:42 +08:00
try:
2021-11-23 13:36:15 +08:00
for f in files:
if f.exists():
f.unlink()
2021-11-19 15:52:42 +08:00
now = time()
for f in OUTDIR.iterdir():
if f.is_file() and now - f.stat().st_mtime > 60.0:
f.unlink()
except Exception:
raise
2021-11-23 13:36:15 +08:00
def espeak(to_speak, outfile):
2021-11-19 15:52:42 +08:00
p1 = subprocess.run(
["espeak-ng", "-v", "en-us", "--stdin", "-p", "80", "--stdout"],
input=str(to_speak).encode("utf-8"),
capture_output=True,
check=True,
timeout=10
)
p2 = subprocess.run(['ffmpeg', '-hide_banner', '-i', "-",
2021-11-23 13:36:15 +08:00
"-vn", "-ac", "1", "-ar", "8000", f"{outfile}"],
2021-11-19 15:52:42 +08:00
input=p1.stdout,
capture_output=True,
check=True,
timeout=10
)
def whois(arg):
try:
i = int(arg)
except ValueError:
raise ValueError(f"invalid autonomous system number {','.join(list(arg))}.")
else:
if 0 <= i <= 9999:
arg = f"AS424242{i:04d}"
elif 20000 <= i <= 29999:
arg = f"AS42424{i:05d}"
elif 70000 <= i <= 79999:
arg = f"AS42012{i:05d}"
else:
arg = f"AS{i}"
with socket.socket(family=socket.AF_INET, type=socket.SOCK_STREAM) as s:
s.settimeout(5)
s.connect(WHOIS_SERVER)
s.send((f"{arg}\r\n").encode("utf-8"))
r = bytes()
while len(_r := s.recv(1024)) != 0:
r += _r
r = r.decode('utf-8')
lines = r.split('\n')
target = None
for (i, line) in enumerate(lines[::-1]):
if line:
target = False
elif not line and target is False:
target = -i
break
lines = lines[target:]
def multiline_process():
attr = ""
for i, line in enumerate(lines):
if not line:
continue
if line.startswith('%'):
continue
if m := re.match(r'^([\S]+):.*$', line):
attr = m.groups()[0]
assert attr
else:
assert attr
lines[i] = f"{attr}:{lines[i]}"
try:
multiline_process()
except Exception as e:
raise
whois_filter = lambda x: any([x.startswith(field) for field in {'remarks', 'nserver', 'descr', 'ds-rdata', 'mp-import', 'mp-export'}])
lines = [l for l in lines if l and not whois_filter(l)]
lines_new = list()
repl = {
"aut-num": "Autonomous System Number",
"as-name": "Autonomous System Name",
"admin-c": "Administrative Contact",
"tech-c": "Technical Contact",
"mnt-by": "Maintained by",
}
for l in lines:
if not l.strip():
continue
elif l.startswith('%'):
continue
elif l.startswith("aut-num:"):
_p, _s = l.split()
_s = ",".join(list(_s)).replace("A,S,", "")
l = f"{_p} {_s}"
for fr, to in repl.items():
if l.startswith(f"{fr}:"):
l = l.replace(f"{fr}:", f"{to}:", 1)
lines_new.append(l)
if not lines_new:
lines_new.append(f"There is no such Autonomous System {','.join(list(arg)).replace('A,S,', '')}")
else:
lines_new.insert(0, "Found the following who is record from the dn42 registry.")
return ".\n".join(lines_new)+"."
def main():
# argv[1]: as number
# argv[2]: unique identifier
2021-11-23 13:36:15 +08:00
unid = argv[2]
outfile = OUTDIR / f"{unid}.wav"
errfile = OUTDIR / f"{unid}{ERRSOUND.suffix}"
cleanup(outfile, errfile)
2021-11-19 15:52:42 +08:00
try:
ret = whois(argv[1])
except Exception as e:
ret = f"error, {e}"
try:
2021-11-23 13:36:15 +08:00
espeak(ret, outfile)
2021-11-19 15:52:42 +08:00
except Exception:
2021-11-23 13:36:15 +08:00
shutil.copy(ERRSOUND, errfile)
2021-11-19 15:52:42 +08:00
if __name__ == "__main__":
main()