Adding conditions for command line arguments
This commit is contained in:
Vividh Mariy
134
paster.py
134
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.
|
||||
# 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"
|
||||
# which port number to bind to
|
||||
# choose a port number > 1000, to prevent interference with system processes
|
||||
port = "1234"
|
||||
|
||||
|
||||
# which port number to bind to
|
||||
# choose a port number > 1000, to prevent interference with system processes
|
||||
port = "1234"
|
||||
# maximum number of clients this can handle at any given time
|
||||
queue_depth = 10
|
||||
|
||||
|
||||
# maximum number of clients this can handle at any given time
|
||||
queue_depth = 10
|
||||
# directory path
|
||||
# default is $HOME i.e. /home/username/sockbin
|
||||
output_directory = os.path.expanduser("~") + "sockbin"
|
||||
|
||||
|
||||
# directory path
|
||||
# default is $HOME i.e. /home/username/sockbin
|
||||
output_directory = os.path.expanduser("~") + "sockbin"
|
||||
# specify the length of generated filename. multiples of 2 only
|
||||
slug_size = 8
|
||||
|
||||
|
||||
# specify the length of generated filename. multiples of 2 only
|
||||
slug_size = 8
|
||||
# amount of data transferred at a time, in bytes
|
||||
# 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
|
||||
# set between 4096 and 64000.
|
||||
buffer_size = 32768
|
||||
helpmessage = """Welcome to SockBin, the command line pastebin !
|
||||
Released under GNU GPL.
|
||||
|
||||
-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:])
|
||||
Reference in New Issue
Block a user