130 lines
4.5 KiB
Python
130 lines
4.5 KiB
Python
import argparse, sys, os, time
|
|
import ConfigParser
|
|
|
|
class Config(object):
|
|
def __init__(self):
|
|
self.version = "0.1"
|
|
self.parser = self.createArguments()
|
|
argv = sys.argv[:] # Copy command line arguments
|
|
argv = self.parseConfig(argv) # Add arguments from config file
|
|
self.parseCommandline(argv) # Parse argv
|
|
self.setAttributes()
|
|
|
|
|
|
def __str__(self):
|
|
return str(self.arguments).replace("Namespace", "Config") # Using argparse str output
|
|
|
|
|
|
# Create command line arguments
|
|
def createArguments(self):
|
|
# Platform specific
|
|
if sys.platform.startswith("win"):
|
|
upnpc = "tools\\upnpc\\upnpc-static.exe"
|
|
coffeescript = "type %s | tools\\coffee\\coffee.cmd"
|
|
else:
|
|
upnpc = None
|
|
coffeescript = None
|
|
|
|
# Create parser
|
|
parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
|
subparsers = parser.add_subparsers(title="Action to perform", dest="action")
|
|
|
|
# Main
|
|
action = subparsers.add_parser("main", help='Start UiServer and FileServer (default)')
|
|
|
|
# SiteCreate
|
|
action = subparsers.add_parser("siteCreate", help='Create a new site')
|
|
|
|
# SiteSign
|
|
action = subparsers.add_parser("siteSign", help='Update and sign content.json: address [privatekey]')
|
|
action.add_argument('address', help='Site to sign')
|
|
action.add_argument('privatekey', help='Private key (default: ask on execute)', nargs='?')
|
|
|
|
# SitePublish
|
|
action = subparsers.add_parser("sitePublish", help='Publish site to other peers: address')
|
|
action.add_argument('address', help='Site to publish')
|
|
|
|
# SiteVerify
|
|
action = subparsers.add_parser("siteVerify", help='Verify site files using md5: address')
|
|
action.add_argument('address', help='Site to verify')
|
|
|
|
|
|
# Config parameters
|
|
parser.add_argument('--debug', help='Debug mode', action='store_true')
|
|
|
|
parser.add_argument('--ui_ip', help='Web interface bind address', default="127.0.0.1", metavar='host')
|
|
parser.add_argument('--ui_port', help='Web interface bind port', default=43110, metavar='port')
|
|
parser.add_argument('--ui_restrict', help='Restrict web access', default=False, metavar='ip')
|
|
parser.add_argument('--homepage', help='Web interface Homepage', default='1EU1tbG9oC1A8jz2ouVwGZyQ5asrNsE4Vr', metavar='address')
|
|
|
|
parser.add_argument('--fileserver_ip', help='FileServer bind address', default="*", metavar='host')
|
|
parser.add_argument('--fileserver_port',help='FileServer bind port', default=15441, metavar='port')
|
|
|
|
parser.add_argument('--ip_external', help='External ip (tested on start if None)', metavar='ip')
|
|
parser.add_argument('--upnpc', help='MiniUPnP binary for open port on router', default=upnpc, metavar='executable_path')
|
|
|
|
parser.add_argument('--coffeescript_compiler', help='Coffeescript compiler for developing', default=coffeescript, metavar='executable_path')
|
|
|
|
parser.add_argument('--version', action='version', version='ZeroNet %s' % self.version)
|
|
|
|
return parser
|
|
|
|
|
|
# Find arguments specificed for current action
|
|
def getActionArguments(self):
|
|
back = {}
|
|
arguments = self.parser._subparsers._group_actions[0].choices[self.action]._actions[1:] # First is --version
|
|
for argument in arguments:
|
|
back[argument.dest] = getattr(self, argument.dest)
|
|
return back
|
|
|
|
|
|
|
|
# Try to find action from sys.argv
|
|
def getAction(self, argv):
|
|
actions = [action.choices.keys() for action in self.parser._actions if action.dest == "action"][0] # Valid actions
|
|
found_action = False
|
|
for action in actions: # See if any in sys.argv
|
|
if action in argv:
|
|
found_action = action
|
|
break
|
|
return found_action
|
|
|
|
|
|
# Parse command line arguments
|
|
def parseCommandline(self, argv):
|
|
# Find out if action is specificed on start
|
|
action = self.getAction(argv)
|
|
if len(argv) == 1 or not action: # If no action specificed set the main action
|
|
argv.append("main")
|
|
if "zeronet.py" in argv[0]:
|
|
self.arguments = self.parser.parse_args(argv[1:])
|
|
else: # Silent errors if not started with zeronet.py
|
|
self.arguments = self.parser.parse_args(argv[1:])
|
|
|
|
|
|
# Parse config file
|
|
def parseConfig(self, argv):
|
|
if os.path.isfile("zeronet.conf"):
|
|
config = ConfigParser.ConfigParser(allow_no_value=True)
|
|
config.read('zeronet.conf')
|
|
for section in config.sections():
|
|
for key, val in config.items(section):
|
|
if section != "global": # If not global prefix key with section
|
|
key = section+"_"+key
|
|
if val: argv.insert(1, val)
|
|
argv.insert(1, "--%s" % key)
|
|
return argv
|
|
|
|
|
|
|
|
# Expose arguments as class attributes
|
|
def setAttributes(self):
|
|
# Set attributes from arguments
|
|
args = vars(self.arguments)
|
|
for key, val in args.items():
|
|
setattr(self, key, val)
|
|
|
|
|
|
config = Config()
|