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')
