From c048fda55a27d83ece24cc307332f70089e4d5d1 Mon Sep 17 00:00:00 2001 From: randomuser Date: Tue, 20 Jul 2021 03:43:12 -0500 Subject: [PATCH] add perimitive argparse support --- esgd.py | 21 ++++++++++++++++----- 1 file changed, 16 insertions(+), 5 deletions(-) diff --git a/esgd.py b/esgd.py index 8bb6024..64cf15c 100644 --- a/esgd.py +++ b/esgd.py @@ -1,5 +1,6 @@ import socketserver import subprocess +import argparse import os import time import pwd @@ -186,13 +187,23 @@ class GopherHandler(socketserver.BaseRequestHandler, GopherServerLogic): self.serveFile(file[0], parsed[1], file[1]) log = Logger() -host = "localhost" -port = 70 -location = "/tmp/test" if __name__ == "__main__": - switchgrp = grp.getgrnam("nobody").gr_gid - switchusr = pwd.getpwnam("nobody").pw_uid + parse = argparse.ArgumentParser() + parse.add_argument("-u", "--user", default="nobody", help="user to change to on startup") + parse.add_argument("-g", "--group", default="nobody", help="group to change to on startup") + parse.add_argument("-s", "--host", default="localhost", help="host to host on") + parse.add_argument("-p", "--port", default=70, type=int, help="port to host on") + parse.add_argument("-d", "--dir", default="/var/gopher", help="directory to host from") + args = parse.parse_args() + log.log("arguments parsed") + + switchgrp = grp.getgrnam(args.group).gr_gid + switchusr = pwd.getpwnam(args.user).pw_uid + host = args.host + port = args.port + location = args.dir + log.log("arguments good") try: with socketserver.TCPServer((host, port), GopherHandler) as server: os.setgid(switchgrp)