Adding conditions for command line arguments

paster.py

@@ -1,40 +1,114 @@
 #!/usr/bin/python3
 
-import os,sys,socket,secrets
+import os,socket,secrets
+import sys,getopt,re
+
 
 # Config
+def main(theargs):
+    # where to serve this page.
+    # 127.0.0.1 / localhost => only accessible locally, on this machine
+    # 0.0.0.0 => accessible to the internet, by any host 
+    hostname = "0.0.0.0"
 
 
-# where to serve this page.
+    # which port number to bind to
-# 127.0.0.1 / localhost => only accessible locally, on this machine
+    # choose a port number > 1000, to prevent interference with system processes
-# 0.0.0.0 => accessible to the internet, by any host 
+    port = "1234"
-hostname = "0.0.0.0"
 
 
-# which port number to bind to
+    # maximum number of clients this can handle at any given time
-# choose a port number > 1000, to prevent interference with system processes
+    queue_depth = 10
-port = "1234"
 
 
-# maximum number of clients this can handle at any given time
+    # directory path
-queue_depth = 10
+    # default is $HOME i.e. /home/username/sockbin
+    output_directory = os.path.expanduser("~") + "sockbin"
 
 
-# directory path
+    # specify the length of generated filename. multiples of 2 only 
-# default is $HOME i.e. /home/username/sockbin
+    slug_size = 8
-output_directory = os.path.expanduser("~") + "sockbin"
 
 
-# specify the length of generated filename. multiples of 2 only 
+    # amount of data transferred at a time, in bytes
-slug_size = 8
+    # set between 4096 and 64000. 
+    buffer_size = 32768
+
+    # path for log file
+    log_file = "/tmp/socklog.txt"
 
 
-# amount of data transferred at a time, in bytes
+    helpmessage = """Welcome to SockBin, the command line pastebin !
-# set between 4096 and 64000. 
+    Released under GNU GPL.
-buffer_size = 32768
+
+    -n --hostname\t\tSet the hostname to listen for. 0.0.0.0 by default.
+    -p --port\t\tExternel port number to listen on. 8888 by default
+    -q --queue_depth\tMax number of simultaneous connections to accept
+    -o --output_directory\tFile storage location. $HOME/sockbin by default
+    -s --slug_size\tLength of url to generate.
+    -b --buffer_size\tPacket size in bytes. 
+    -l --log_file\t\tPath to log file.
+    -h --help\t\tDisplay this message
+
+    """
+
+    try:
+        opts, args = getopt.getopt(theargs, "n:p:q:o:s:b:l:", ["hostname=","port=", "queue_depth=", "output_directory=", "slug_size=","buffer_size=", "log_file="])
+        for opt, arg in opts:
+            if opt in ['-n', '--hostname']:
+                if arg.find('/') != -1:
+                    print("Incorrect hostname. / not allowed. Run with -h for options")
+                    sys.exit()
+                else:
+                    hostname = arg
+            elif opt in ['-p', "--port"]:
+                if re.match("^ *\d[\d ]*$", arg) is None:
+                    print("Incorrect port format. Choose a number between 0 and 64738. Warning: ports below 1000 could interfere with system processes.")
+                    sys.exit()
+                elif int(arg) > 64738:
+                    print("Incorrect port format. Choose a number between 0 and 64738. Warning: ports below 1000 could interfere with system processes.")
+                    sys.exit()
+                elif int(arg) < 1000:
+                    choice = input("Warning: ports below 1000 could interfere with system processes. Do you want to choose a different port ? (y/n): ")
+
+                    if choice.lower() == "y":
+                        arg = input("Please enter your new port number: ")
+                        if re.match("^ *\d[\d ]*$", arg) is None:
+                            print("Incorrect port format. Choose a number between 1000 and 64738")
+                            sys.exit()
+                        else:
+                            port = arg
+                    elif choice.lower() == "n":
+                        pass
+
+
+
+                port = arg
+            elif opt in ['-q', "--queue_depth"]:
+                queue_depth = arg
+            elif opt in ['-o', "--output_directory"]:
+                output_directory = arg
+            elif opt in ['-s', "--slug_size"]:
+                slug_size = arg
+            elif opt in ['-b', "--buffer_size"]:
+                buffer_size = arg
+            elif opt in ['-l', "--log_file"]:
+                log_file = arg
+            
+        print({
+            "hostname":hostname,
+            "port":port,
+            "queue":queue_depth,
+            "output":output_directory,
+            "slug":slug_size,
+            "buff":buffer_size,
+            "log":log_file,
+        })
+
+    except getopt.GetoptError as err:
+        print(err)
+        print("Run with -h or --help to see the various options")
 
-# path for log file
-log_file = "/tmp/socklog.txt"
 
 
 
@@ -76,24 +150,6 @@ def server():
 
 
 
-def main(argv):
-   inputfile = ''
-   outputfile = ''
-   try:
-      opts, args = getopt.getopt(argv,"hi:o:",["ifile=","ofile="])
-   except getopt.GetoptError:
-      print 'test.py -i <inputfile> -o <outputfile>'
-      sys.exit(2)
-   for opt, arg in opts:
-      if opt == '-h':
-         print 'test.py -i <inputfile> -o <outputfile>'
-         sys.exit()
-      elif opt in ("-i", "--ifile"):
-         inputfile = arg
-      elif opt in ("-o", "--ofile"):
-         outputfile = arg
-   print 'Input file is "', inputfile
-   print 'Output file is "', outputfile
 
 if __name__ == "__main__":
-   main(sys.argv[1:])
+    main(sys.argv[1:])