diff options
| -rwxr-xr-x | autopeer_shell.py | 30 |
1 files changed, 24 insertions, 6 deletions
diff --git a/autopeer_shell.py b/autopeer_shell.py index 1670eb3..c891e63 100755 --- a/autopeer_shell.py +++ b/autopeer_shell.py @@ -1,9 +1,8 @@ #!/usr/bin/python3 #TODO: capture ^C during commands -#TODO: cronjob to generate config files from database -import cmd, functools, getpass, os, re, sys, time +import cmd, functools, getpass, os, re, subprocess, sys, time try: import readline except: pass @@ -12,9 +11,24 @@ from config import * from lib_autopeer import * def as_from_user(which=None): - # TODO: get list of user's maintained ASes - # from their MNTNER - user_ases=[4242422452] + if which is not None: + try: + which = int(which) + except: + raise Exception('your AS number has to be a number, silly') + + user_ases=[] + if not re.match(r'^[a-zA-Z0-9-]+$', MNTNER): + raise Exception('your mntner name has bad characters') + filenames = subprocess.run(['grep', '-l', '-r', '-P', fr'^\s*mnt-by:\s*{MNTNER}\s*$', '/opt/autopeer/dn42-registry/data/aut-num'], capture_output=True, text=True).stdout + for filename in filenames.split('\n'): + if not len(filename): + continue + if (mo := re.match(r'^.*/?AS(\d+)$', filename)): + user_ases.append(int(mo.group(1))) + else: + print(f"oops, something went wrong getting your ASes, specifically: {filename}", file=sys.stderr) + if which is None: return user_ases[0] elif which in user_ases: @@ -45,7 +59,11 @@ SELECTED_ASN = as_from_user() class AutopeerShell(cmd.Cmd): def preloop(self): self.intro = f'Welcome to the autopeer shell. Type help or ? to list commands.\nSelected AS: {SELECTED_ASN}' - self.prompt = f'{getpass.getuser()}@{socket.gethostname()}> ' + self.prompt = f'{getpass.getuser()}@{socket.gethostname()}:{SELECTED_ASN}> ' + + def postcmd(self, stop, line): + self.prompt = f'{getpass.getuser()}@{socket.gethostname()}:{SELECTED_ASN}> ' + return stop @parse(0) def do_ls(self): |