help='Specify the pid file for the daemon (default: current bot data directory)') args = vars(parser.parse_args()) # create a dictionary of args config_path = args['config'] # setup the environment to be able to import the config.py if config_path: # appends the current config in order to find config.py sys.path.insert(0, path.dirname(path.abspath(config_path))) else: config_path = execution_dir + sep + 'config.py' config = get_config(config_path) # will exit if load fails if args['list']: from errbot.main import enumerate_backends print('Available backends:') for backend_name in enumerate_backends(config): print('\t\t%s' % backend_name) sys.exit(0) # this is temporary until the modes are removed to translate the mode names to backend names classic_vs_plugin_names = {'text': 'Text', 'graphic': 'Graphic', 'hipchat': 'Hipchat', 'irc': 'IRC', 'xmpp': 'XMPP', 'slack': 'Slack', 'null': 'Null'} filtered_mode = [mname for mname in classic_vs_plugin_names.keys() if args[mname]] if args['restore']: backend = 'Null' # we don't want any backend when we restore
def main(): execution_dir = getcwd() # By default insert the execution path (useful to be able to execute err from # the source tree directly without installing it. sys.path.insert(0, execution_dir) parser = argparse.ArgumentParser(description='The main entry point of the XMPP bot err.') parser.add_argument('-c', '--config', default=None, help='Full path to your config.py (default: config.py in current working directory).') parser.add_argument('-r', '--restore', nargs='?', default=None, const='default', help='Restores a bot from backup.py. (default: backup.py from the bot data directory).') parser.add_argument('-l', '--list', action='store_true', help='Lists all the backends found.') backend_group = parser.add_mutually_exclusive_group() backend_group.add_argument('-X', '--xmpp', action='store_true', help='XMPP backend [DEFAULT]') backend_group.add_argument('-H', '--hipchat', action='store_true', help='Hipchat backend') backend_group.add_argument('-I', '--irc', action='store_true', help='IRC backend') backend_group.add_argument('-S', '--slack', action='store_true', help='Slack backend') backend_group.add_argument('-T', '--text', action='store_true', help='locale text debug backend') backend_group.add_argument('-G', '--graphic', action='store_true', help='local graphical debug mode backend') backend_group.add_argument('-N', '--null', action='store_true', help='no backend') if not ON_WINDOWS: option_group = parser.add_argument_group('arguments to run it as a Daemon') option_group.add_argument('-d', '--daemon', action='store_true', help='Detach the process from the console') option_group.add_argument('-p', '--pidfile', default=None, help='Specify the pid file for the daemon (default: current bot data directory)') args = vars(parser.parse_args()) # create a dictionary of args config_path = args['config'] # setup the environment to be able to import the config.py if config_path: # appends the current config in order to find config.py sys.path.insert(0, path.dirname(path.abspath(config_path))) else: config_path = execution_dir + sep + 'config.py' config = get_config(config_path) # will exit if load fails if args['list']: from errbot.main import enumerate_backends print('Available backends:') for backend_name in enumerate_backends(config): print('\t\t%s' % backend_name) sys.exit(0) # this is temporary until the modes are removed to translate the mode names to backend names classic_vs_plugin_names = {'text': 'Text', 'graphic': 'Graphic', 'hipchat': 'Hipchat', 'irc': 'IRC', 'xmpp': 'XMPP', 'slack': 'Slack', 'null': 'Null'} filtered_mode = [mname for mname in classic_vs_plugin_names.keys() if args[mname]] if args['restore']: backend = 'Null' # we don't want any backend when we restore elif filtered_mode: backend = classic_vs_plugin_names[filtered_mode[0]] if backend != 'Text': log.warn("""Deprecation notice: Please add BACKEND='%s' to your config.py instead of using the '--%s' command line parameter. The backend command line parameters will be removed on the next version of Err. """ % (backend, filtered_mode[0])) elif hasattr(config, 'BACKEND'): backend = config.BACKEND else: log.warn("""Deprecation notice: Err is defaulting to XMPP because you did not specify any backend. Please add BACKEND='XMPP' to your config.py if you really want that. This behaviour will be removed on the next version of Err. """) backend = 'XMPP' # default value log.info("Selected backend '%s'." % backend) # Check if at least we can start to log something before trying to start # the bot (esp. daemonize it). log.info("Checking for '%s'..." % config.BOT_DATA_DIR) if not path.exists(config.BOT_DATA_DIR): raise Exception("The data directory '%s' for the bot does not exist" % config.BOT_DATA_DIR) if not access(config.BOT_DATA_DIR, W_OK): raise Exception("The data directory '%s' should be writable for the bot" % config.BOT_DATA_DIR) if (not ON_WINDOWS) and args['daemon']: if args['text']: raise Exception('You cannot run in text and daemon mode at the same time') if args['restore']: raise Exception('You cannot restore a backup in daemon mode.') if args['pidfile']: pid = args['pidfile'] else: pid = config.BOT_DATA_DIR + sep + 'err.pid' # noinspection PyBroadException try: def action(): from errbot.main import main main(backend, logger, config) daemon = Daemonize(app="err", pid=pid, action=action) daemon.start() except Exception: log.exception('Failed to daemonize the process') exit(0) from errbot.main import main restore = args['restore'] if restore == 'default': # restore with no argument, get the default location restore = path.join(config.BOT_DATA_DIR, 'backup.py') main(backend, logger, config, restore) log.info('Process exiting')
) args = vars(parser.parse_args()) # create a dictionary of args config_path = args['config'] # setup the environment to be able to import the config.py if config_path: # appends the current config in order to find config.py sys.path.insert(0, path.dirname(path.abspath(config_path))) else: config_path = execution_dir + sep + 'config.py' config = get_config(config_path) # will exit if load fails if args['list']: from errbot.main import enumerate_backends print('Available backends:') for backend_name in enumerate_backends(config): print('\t\t%s' % backend_name) sys.exit(0) # this is temporary until the modes are removed to translate the mode names to backend names classic_vs_plugin_names = { 'text': 'Text', 'graphic': 'Graphic', 'hipchat': 'Hipchat', 'irc': 'IRC', 'xmpp': 'XMPP', 'slack': 'Slack', 'null': 'Null' } filtered_mode = [
def main(): execution_dir = getcwd() # By default insert the execution path (useful to be able to execute err from # the source tree directly without installing it. sys.path.insert(0, execution_dir) parser = argparse.ArgumentParser(description='The main entry point of the errbot.') parser.add_argument('-c', '--config', default=None, help='Full path to your config.py (default: config.py in current working directory).') mode_selection = parser.add_mutually_exclusive_group() mode_selection.add_argument('-v', '--version', action='version', version='Errbot version {}'.format(VERSION)) mode_selection.add_argument('-r', '--restore', nargs='?', default=None, const='default', help='restore a bot from backup.py (default: backup.py from the bot data directory)') mode_selection.add_argument('-l', '--list', action='store_true', help='list all available backends') mode_selection.add_argument('--new-plugin', nargs='?', default=None, const='current_dir', help='create a new plugin in the specified directory') mode_selection.add_argument('-T', '--text', dest="backend", action='store_const', const="Text", help='force local text backend') mode_selection.add_argument('-G', '--graphic', dest="backend", action='store_const', const="Graphic", help='force local graphical backend') if not ON_WINDOWS: option_group = parser.add_argument_group('optional daemonization arguments') option_group.add_argument('-d', '--daemon', action='store_true', help='Detach the process from the console') option_group.add_argument('-p', '--pidfile', default=None, help='Specify the pid file for the daemon (default: current bot data directory)') args = vars(parser.parse_args()) # create a dictionary of args # This must come BEFORE the config is loaded below, to avoid printing # logs as a side effect of config loading. if args['new_plugin']: directory = os.getcwd() if args['new_plugin'] == "current_dir" else args['new_plugin'] for handler in logging.getLogger().handlers: logger.removeHandler(handler) try: new_plugin_wizard(directory) except KeyboardInterrupt: sys.exit(1) except Exception as e: sys.stderr.write(str(e) + "\n") sys.exit(1) finally: sys.exit(0) config_path = args['config'] # setup the environment to be able to import the config.py if config_path: # appends the current config in order to find config.py sys.path.insert(0, path.dirname(path.abspath(config_path))) else: config_path = execution_dir + sep + 'config.py' config = get_config(config_path) # will exit if load fails if args['list']: from errbot.main import enumerate_backends print('Available backends:') for backend_name in enumerate_backends(config): print('\t\t%s' % backend_name) sys.exit(0) if args['restore']: backend = 'Null' # we don't want any backend when we restore elif args['backend'] is None: if not hasattr(config, 'BACKEND'): log.fatal("The BACKEND configuration option is missing in config.py") sys.exit(1) backend = config.BACKEND else: backend = args['backend'] log.info("Selected backend '%s'." % backend) # Check if at least we can start to log something before trying to start # the bot (esp. daemonize it). log.info("Checking for '%s'..." % config.BOT_DATA_DIR) if not path.exists(config.BOT_DATA_DIR): raise Exception("The data directory '%s' for the bot does not exist" % config.BOT_DATA_DIR) if not access(config.BOT_DATA_DIR, W_OK): raise Exception("The data directory '%s' should be writable for the bot" % config.BOT_DATA_DIR) if (not ON_WINDOWS) and args['daemon']: if args['text']: raise Exception('You cannot run in text and daemon mode at the same time') if args['restore']: raise Exception('You cannot restore a backup in daemon mode.') if args['pidfile']: pid = args['pidfile'] else: pid = config.BOT_DATA_DIR + sep + 'err.pid' # noinspection PyBroadException try: def action(): from errbot.main import main main(backend, logger, config) daemon = Daemonize(app="err", pid=pid, action=action, chdir=os.getcwd()) daemon.start() except Exception: log.exception('Failed to daemonize the process') exit(0) from errbot.main import main restore = args['restore'] if restore == 'default': # restore with no argument, get the default location restore = path.join(config.BOT_DATA_DIR, 'backup.py') main(backend, logger, config, restore) log.info('Process exiting')
def main(): execution_dir = getcwd() # By default insert the execution path (useful to be able to execute err from # the source tree directly without installing it. sys.path.insert(0, execution_dir) parser = argparse.ArgumentParser( description='The main entry point of the errbot.') parser.add_argument( '-c', '--config', default=None, help= 'Full path to your config.py (default: config.py in current working directory).' ) mode_selection = parser.add_mutually_exclusive_group() mode_selection.add_argument('-v', '--version', action='version', version='Errbot version {}'.format(VERSION)) mode_selection.add_argument( '-r', '--restore', nargs='?', default=None, const='default', help= 'restore a bot from backup.py (default: backup.py from the bot data directory)' ) mode_selection.add_argument('-l', '--list', action='store_true', help='list all available backends') mode_selection.add_argument( '--new-plugin', nargs='?', default=None, const='current_dir', help='create a new plugin in the specified directory') # storage manipulation mode_selection.add_argument( '--storage-set', nargs=1, help='DANGER: Delete the given storage namespace ' 'and set the python dictionary expression ' 'passed on stdin.') mode_selection.add_argument( '--storage-merge', nargs=1, help='DANGER: Merge in the python dictionary expression ' 'passed on stdin into the given storage namespace.') mode_selection.add_argument('--storage-get', nargs=1, help='Dump the given storage namespace in a ' 'format compatible for --storage-set and ' '--storage-merge.') mode_selection.add_argument('-T', '--text', dest="backend", action='store_const', const="Text", help='force local text backend') mode_selection.add_argument('-G', '--graphic', dest="backend", action='store_const', const="Graphic", help='force local graphical backend') if not ON_WINDOWS: option_group = parser.add_argument_group( 'optional daemonization arguments') option_group.add_argument('-d', '--daemon', action='store_true', help='Detach the process from the console') option_group.add_argument( '-p', '--pidfile', default=None, help= 'Specify the pid file for the daemon (default: current bot data directory)' ) args = vars(parser.parse_args()) # create a dictionary of args # This must come BEFORE the config is loaded below, to avoid printing # logs as a side effect of config loading. if args['new_plugin']: directory = os.getcwd( ) if args['new_plugin'] == "current_dir" else args['new_plugin'] for handler in logging.getLogger().handlers: logger.removeHandler(handler) try: new_plugin_wizard(directory) except KeyboardInterrupt: sys.exit(1) except Exception as e: sys.stderr.write(str(e) + "\n") sys.exit(1) finally: sys.exit(0) config_path = args['config'] # setup the environment to be able to import the config.py if config_path: # appends the current config in order to find config.py sys.path.insert(0, path.dirname(path.abspath(config_path))) else: config_path = execution_dir + sep + 'config.py' config = get_config(config_path) # will exit if load fails if args['list']: from errbot.main import enumerate_backends print('Available backends:') for backend_name in enumerate_backends(config): print('\t\t%s' % backend_name) sys.exit(0) def storage_action(namespace, fn): # Used to defer imports until it is really necessary during the loading time. from errbot.main import get_storage_plugin from errbot.storage import StoreMixin try: with StoreMixin() as sdm: sdm.open_storage(get_storage_plugin(config), namespace) fn(sdm) return 0 except Exception as e: print(str(e), file=sys.stderr) return -3 if args['storage_get']: err_value = storage_action(args['storage_get'][0], lambda sdm: print(repr(dict(sdm)))) sys.exit(err_value) if args['storage_set']: def replace(sdm): new_dict = _read_dict( ) # fail early and don't erase the storage if the input is invalid. sdm.clear() sdm.update(new_dict) err_value = storage_action(args['storage_set'][0], replace) sys.exit(err_value) if args['storage_merge']: def merge(sdm): new_dict = _read_dict() sdm.update(new_dict) err_value = storage_action(args['storage_merge'][0], merge) sys.exit(err_value) if args['restore']: backend = 'Null' # we don't want any backend when we restore elif args['backend'] is None: if not hasattr(config, 'BACKEND'): log.fatal( "The BACKEND configuration option is missing in config.py") sys.exit(1) backend = config.BACKEND else: backend = args['backend'] log.info("Selected backend '%s'." % backend) # Check if at least we can start to log something before trying to start # the bot (esp. daemonize it). log.info("Checking for '%s'..." % config.BOT_DATA_DIR) if not path.exists(config.BOT_DATA_DIR): raise Exception("The data directory '%s' for the bot does not exist" % config.BOT_DATA_DIR) if not access(config.BOT_DATA_DIR, W_OK): raise Exception( "The data directory '%s' should be writable for the bot" % config.BOT_DATA_DIR) if (not ON_WINDOWS) and args['daemon']: if args['backend'] == "Text": raise Exception( 'You cannot run in text and daemon mode at the same time') if args['restore']: raise Exception('You cannot restore a backup in daemon mode.') if args['pidfile']: pid = args['pidfile'] else: pid = config.BOT_DATA_DIR + sep + 'err.pid' # noinspection PyBroadException try: def action(): from errbot.main import main main(backend, logger, config) daemon = Daemonize(app="err", pid=pid, action=action, chdir=os.getcwd()) log.info("Daemonizing") daemon.start() except Exception: log.exception('Failed to daemonize the process') exit(0) from errbot.main import main restore = args['restore'] if restore == 'default': # restore with no argument, get the default location restore = path.join(config.BOT_DATA_DIR, 'backup.py') main(backend, logger, config, restore) log.info('Process exiting')