mirror of
https://github.com/solero/houdini.git
synced 2025-11-03 22:21:54 +00:00
bootstrap.py now takes command line arguments
This commit is contained in:
Ben
130
bootstrap.py
130
bootstrap.py
@@ -1,15 +1,131 @@
|
||||
import asyncio
|
||||
import sys
|
||||
import logging
|
||||
from houdini.houdini import HoudiniFactory
|
||||
import argparse
|
||||
import config
|
||||
|
||||
from houdini.houdini import Houdini
|
||||
from houdini import ConflictResolution
|
||||
|
||||
if __name__ == '__main__':
|
||||
if sys.platform == 'win32':
|
||||
loop = asyncio.ProactorEventLoop()
|
||||
asyncio.set_event_loop(loop)
|
||||
|
||||
logger = logging.getLogger('houdini')
|
||||
factory_instance = HoudiniFactory(server='Login')
|
||||
|
||||
parser = argparse.ArgumentParser(description='Boot a Houdini server')
|
||||
parser.add_argument('server', action='store', default='Login',
|
||||
help='Name of the server to boot')
|
||||
|
||||
parser.add_argument('-id', action='store', help='Houdini server ID')
|
||||
parser.add_argument('-a', '--address', action='store', help='Houdini server address')
|
||||
parser.add_argument('-p', '--port', action='store', help='Houdini server port', type=int)
|
||||
parser.add_argument('-c', '--capacity', action='store', help='Houdini server capacity', type=int)
|
||||
parser.add_argument('-C', '--cache-expiry', dest='cache_expiry', action='store',
|
||||
help='Cache expiry (seconds)', type=int)
|
||||
parser.add_argument('-P', '--plugins', action='store',
|
||||
nargs='*', help='Plugins to load')
|
||||
|
||||
boot_modes = parser.add_mutually_exclusive_group()
|
||||
boot_modes.add_argument('-W', '--world', action='store_true', help='Run server in world mode')
|
||||
boot_modes.add_argument('-L', '--login', action='store_true', help='Run server in login mode')
|
||||
|
||||
logging_group = parser.add_argument_group('logging')
|
||||
logging_group.add_argument('-lg', '--logging-general', action='store',
|
||||
dest='logging_general_path',
|
||||
help='General log path')
|
||||
logging_group.add_argument('-le', '--logging-error', action='store',
|
||||
dest='logging_error_path',
|
||||
help='Error log path')
|
||||
logging_group.add_argument('-ll', '--logging-level', action='store',
|
||||
dest='logging_level',
|
||||
help='Logging level')
|
||||
|
||||
database_group = parser.add_argument_group('database')
|
||||
database_group.add_argument('-da', '--database-address', action='store',
|
||||
dest='database_address',
|
||||
default=config.database['Address'],
|
||||
help='Postgresql database address')
|
||||
database_group.add_argument('-du', '--database-username', action='store',
|
||||
dest='database_username',
|
||||
default=config.database['Username'],
|
||||
help='Postgresql database username')
|
||||
database_group.add_argument('-dp', '--database-password', action='store',
|
||||
dest='database_password',
|
||||
default=config.database['Password'],
|
||||
help='Postgresql database password')
|
||||
database_group.add_argument('-dn', '--database-name', action='store',
|
||||
dest='database_name',
|
||||
default=config.database['Name'],
|
||||
help='Postgresql database name')
|
||||
|
||||
redis_group = parser.add_argument_group('redis')
|
||||
redis_group.add_argument('-ra', '--redis-address', action='store',
|
||||
dest='redis_address',
|
||||
default=config.redis['Address'],
|
||||
help='Redis server address')
|
||||
redis_group.add_argument('-rp', '--redis-port', action='store',
|
||||
dest='redis_port',
|
||||
type=int,
|
||||
default=config.redis['Port'],
|
||||
help='Redis server port')
|
||||
|
||||
command_group = parser.add_argument_group('commands')
|
||||
command_group.add_argument('-cp', '--command-prefix', action='store', dest='command_prefix',
|
||||
nargs='*',
|
||||
default=config.commands['Prefix'],
|
||||
help='Command prefixes')
|
||||
command_group.add_argument('-csd', '--command-string-delimiters', action='store', dest='command_string_delimiters',
|
||||
nargs='*',
|
||||
default=config.commands['StringDelimiters'],
|
||||
help='Command string delimiters')
|
||||
command_group.add_argument('-ccm', '--command-conflict-mode', action='store', dest='command_conflict_mode',
|
||||
default=config.commands['ConflictMode'].name,
|
||||
help='Command conflict mode', choices=['Silent', 'Append', 'Exception'])
|
||||
args = parser.parse_args()
|
||||
|
||||
database = {
|
||||
'Address': args.database_address,
|
||||
'Username': args.database_username,
|
||||
'Password': args.database_password,
|
||||
'Name': args.database_name
|
||||
}
|
||||
|
||||
redis = {
|
||||
'Address': args.redis_address,
|
||||
'Port': args.redis_port
|
||||
}
|
||||
|
||||
commands = {
|
||||
'Prefix': args.command_prefix,
|
||||
'StringDelimiters': args.command_string_delimiters,
|
||||
'ConflictMode': getattr(ConflictResolution, args.command_conflict_mode)
|
||||
}
|
||||
|
||||
server = {
|
||||
'Address': args.address or config.servers[args.server]['Address'],
|
||||
'Port': args.port or config.servers[args.server]['Port'],
|
||||
'World': True if args.world else False if args.login else None or config.servers[args.server]['World'],
|
||||
'Plugins': True if args.plugins and '*' in args.plugins
|
||||
else args.plugins or config.servers[args.server]['Plugins']
|
||||
}
|
||||
|
||||
logging = {
|
||||
'General': args.logging_general_path or config.servers[args.server]['Logging']['General'],
|
||||
'Errors': args.logging_error_path or config.servers[args.server]['Logging']['Errors'],
|
||||
'Level': args.logging_level or config.servers[args.server]['Logging']['Level']
|
||||
}
|
||||
|
||||
if server['World']:
|
||||
server.update({
|
||||
'Id': args.id or config.servers[args.server]['Id'],
|
||||
'Capacity': args.capacity or config.servers[args.server]['Capacity'],
|
||||
'CacheExpiry': args.cache_expiry or config.servers[args.server]['CacheExpiry']
|
||||
})
|
||||
|
||||
server['logging'] = logging
|
||||
|
||||
factory_instance = Houdini(args.server,
|
||||
database=database,
|
||||
redis=redis,
|
||||
commands=commands,
|
||||
server=server)
|
||||
try:
|
||||
asyncio.run(factory_instance.start())
|
||||
except KeyboardInterrupt:
|
||||
|
||||
@@ -3,6 +3,7 @@ import os
|
||||
import sys
|
||||
import pkgutil
|
||||
import importlib
|
||||
import copy
|
||||
|
||||
from houdini.spheniscidae import Spheniscidae
|
||||
from houdini.penguin import Penguin
|
||||
@@ -43,9 +44,9 @@ from houdini.plugins import PluginManager
|
||||
from houdini.commands import CommandManager
|
||||
|
||||
|
||||
class HoudiniFactory:
|
||||
class Houdini:
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
def __init__(self, server_name, **kwargs):
|
||||
self.server = None
|
||||
self.redis = None
|
||||
self.config = None
|
||||
@@ -53,7 +54,11 @@ class HoudiniFactory:
|
||||
self.db = db
|
||||
self.peers_by_ip = {}
|
||||
|
||||
self.server_name = kwargs['server']
|
||||
self.server_name = server_name
|
||||
self.database_config_override = kwargs.get('database')
|
||||
self.redis_config_override = kwargs.get('redis')
|
||||
self.commands_config_override = kwargs.get('commands')
|
||||
self.server_config_override = kwargs.get('server')
|
||||
self.server_config = None
|
||||
|
||||
self.logger = None
|
||||
@@ -91,7 +96,12 @@ class HoudiniFactory:
|
||||
async def start(self):
|
||||
self.config = config
|
||||
|
||||
self.server_config = self.config.servers[self.server_name]
|
||||
self.server_config = copy.deepcopy(self.config.servers[self.server_name])
|
||||
self.server_config.update(self.server_config_override)
|
||||
|
||||
self.config.database.update(self.database_config_override)
|
||||
self.config.redis.update(self.redis_config_override)
|
||||
self.config.commands.update(self.commands_config_override)
|
||||
|
||||
general_log_directory = os.path.dirname(self.server_config["Logging"]["General"])
|
||||
errors_log_directory = os.path.dirname(self.server_config["Logging"]["Errors"])
|
||||
|
||||
Reference in New Issue
Block a user