Skip to content

Commit

Permalink
added --password -k command line arg
Browse files Browse the repository at this point in the history
  • Loading branch information
RaSan147 committed Mar 4, 2023
1 parent d50e37e commit a7d687e
Show file tree
Hide file tree
Showing 3 changed files with 55 additions and 25 deletions.
12 changes: 11 additions & 1 deletion dev_src/local_server 2.py → dev_src/local_server_pyrobox.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,16 @@
true = T = True
false = F = False


config.parser.add_argument('--password', '-k',
default=config.PASSWORD,
type=str,
help='Upload Password (default: %(default)s)')


args = config.parser.parse_known_args()[0]
config.PASSWORD = args.password

config.disabled_func.update({
"send2trash": False,
"natsort": False,
Expand Down Expand Up @@ -866,7 +876,7 @@ def fetch_url(url, file = None):

with open(file, 'wb') as f:
f.write(data)
return True
return data
except Exception:
traceback.print_exc()
return None
Expand Down
File renamed without changes.
68 changes: 44 additions & 24 deletions dev_src/pyroboxCore.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,9 @@
from queue import Queue
from typing import Union

import argparse


logging.basicConfig(level=logging.DEBUG, format='%(levelname)s: %(message)s')

logger = logging.getLogger(__name__)
Expand Down Expand Up @@ -81,6 +84,10 @@ def __init__(self):
# ASSET MAPPING
self.file_list = {}

# COMMANDLINE ARGUMENTS PARSER
self.parser = argparse.ArgumentParser(add_help=False)


def clear_temp(self):
for i in self.temp_file:
try:
Expand All @@ -107,6 +114,42 @@ def get_default_dir(self):

def address(self):
return "http://%s:%i"%(self.IP, self.port)

def parse_default_args(self, port = None, directory = None, bind = None, ):
if port is None:
port = self.port
if directory is None:
directory = self.ftp_dir
if bind is None:
bind = None

parser = self.parser

parser.add_argument('--bind', '-b',
metavar='ADDRESS', default=bind,
help='Specify alternate bind address '
'[default: all interfaces]')
parser.add_argument('--directory', '-d', default=directory,
help='Specify alternative directory '
'[default: current directory]')
parser.add_argument('port', action='store',
default=port, type=int,
nargs='?',
help='Specify alternate port [default: 8000]')
parser.add_argument('--version', '-v', action='version',
version=__version__)

self.parser.add_argument('-h', '--help', action='help',
default='==SUPPRESS==',
help=('show this help message and exit'))

args = parser.parse_known_args()[0]

return args







Expand Down Expand Up @@ -1631,32 +1674,9 @@ def finish_request(self, request, client_address):


def run(port = None, directory = None, bind = None, arg_parse= True, handler = SimpleHTTPRequestHandler):
if port is None:
port = config.port
if directory is None:
directory = config.ftp_dir

if arg_parse:
import argparse



parser = argparse.ArgumentParser()

parser.add_argument('--bind', '-b', metavar='ADDRESS',
help='Specify alternate bind address '
'[default: all interfaces]')
parser.add_argument('--directory', '-d', default=directory,
help='Specify alternative directory '
'[default:current directory]')
parser.add_argument('port', action='store',
default=port, type=int,
nargs='?',
help='Specify alternate port [default: 8000]')
parser.add_argument('--version', '-v', action='version',
version=__version__)

args = parser.parse_args()
args = config.parse_default_args(port=port, directory=directory, bind=bind)

port = args.port
directory = args.directory
Expand Down

0 comments on commit a7d687e

Please sign in to comment.