def main(argv=None):
global homedir
# Step One: Parse The Command Line
try:
parser = optparse.OptionParser("%prog [options]")
parser.add_option("-c", "--config", metavar="filename", help="use a specific configuration file")
parser.add_option("-d", "--fork", action="store_true", dest="deamonize", help="Deamonize willie")
parser.add_option("-q", "--quit", action="store_true", dest="quit", help="Gracefully quit Willie")
parser.add_option("-k", "--kill", action="store_true", dest="kill", help="Kill Willie")
parser.add_option(
"--exit-on-error",
action="store_true",
dest="exit_on_error",
help=("Exit immediately on every error instead of " "trying to recover"),
)
parser.add_option("-l", "--list", action="store_true", dest="list_configs", help="List all config files found")
parser.add_option(
"-m",
"--migrate",
action="store_true",
dest="migrate_configs",
help="Migrate config files to the new format",
)
parser.add_option("--quiet", action="store_true", dest="quiet", help="Supress all output")
parser.add_option(
"-w", "--configure-all", action="store_true", dest="wizard", help="Run the configuration wizard."
)
parser.add_option(
"--configure-modules",
action="store_true",
dest="mod_wizard",
help=("Run the configuration wizard, but only for the " "module configuration options."),
)
parser.add_option(
"--configure-database",
action="store_true",
dest="db_wizard",
help=("Run the configuration wizard, but only for the " "database configuration options."),
)
opts, args = parser.parse_args(argv)
if opts.wizard:
wizard("all", opts.config)
return
elif opts.mod_wizard:
wizard("mod", opts.config)
return
elif opts.db_wizard:
wizard("db", opts.config)
return
check_python_version()
if opts.list_configs is not None:
configs = enumerate_configs()
print "Config files in ~/.willie:"
if len(configs[0]) is 0:
print "\tNone found"
else:
for config in configs:
print "\t%s" % config
print "-------------------------"
return
config_name = opts.config or "default"
configpath = find_config(config_name)
if not os.path.isfile(configpath):
print "Welcome to Willie!\nI can't seem to find the configuration file, so let's generate it!\n"
if not configpath.endswith(".cfg"):
configpath = configpath + ".cfg"
create_config(configpath)
configpath = find_config(config_name)
try:
config_module = Config(configpath)
except ConfigurationError as e:
stderr(e)
sys.exit(2)
if config_module.core.not_configured:
stderr("Bot is not configured, can't start")
# exit with code 2 to prevent auto restart on fail by systemd
sys.exit(2)
if not config_module.has_option("core", "homedir"):
config_module.dotdir = homedir
config_module.homedir = homedir
else:
homedir = config_module.core.homedir
config_module.dotdir = config_module.core.homedir
if not config_module.core.logdir:
config_module.core.logdir = os.path.join(homedir, "logs")
logfile = os.path.os.path.join(config_module.logdir, "stdio.log")
if not os.path.isdir(config_module.logdir):
os.mkdir(config_module.logdir)
if opts.exit_on_error:
config_module.exit_on_error = True
else:
config_module.exit_on_error = False
if opts.quiet is None:
opts.quiet = False
sys.stderr = tools.OutputRedirect(logfile, True, opts.quiet)
sys.stdout = tools.OutputRedirect(logfile, False, opts.quiet)
# Handle --quit, --kill and saving the PID to file
pid_dir = config_module.core.pid_dir or homedir
if opts.config is None:
pid_file_path = os.path.join(pid_dir, "willie.pid")
else:
basename = os.path.basename(opts.config)
if basename.endswith(".cfg"):
basename = basename[:-4]
pid_file_path = os.path.join(pid_dir, "willie-%s.pid" % basename)
if os.path.isfile(pid_file_path):
pid_file = open(pid_file_path, "r")
old_pid = int(pid_file.read())
pid_file.close()
if tools.check_pid(old_pid):
if opts.quit is None and opts.kill is None:
stderr("There's already a Willie instance running with this config file")
stderr("Try using the --quit or the --kill options")
sys.exit(1)
elif opts.kill:
stderr("Killing the willie")
os.kill(old_pid, signal.SIGKILL)
sys.exit(0)
elif opts.quit:
stderr("Signaling Willie to stop gracefully")
if hasattr(signal, "SIGUSR1"):
os.kill(old_pid, signal.SIGUSR1)
else:
os.kill(old_pid, signal.SIGTERM)
sys.exit(0)
elif not tools.check_pid(old_pid) and (opts.kill or opts.quit):
stderr("Willie is not running!")
sys.exit(1)
elif opts.quit is not None or opts.kill is not None:
stderr("Willie is not running!")
sys.exit(1)
if opts.deamonize is not None:
child_pid = os.fork()
if child_pid is not 0:
sys.exit()
pid_file = open(pid_file_path, "w")
pid_file.write(str(os.getpid()))
pid_file.close()
config_module.pid_file_path = pid_file_path
# Step Five: Initialise And Run willie
run(config_module)
except KeyboardInterrupt:
print "\n\nInterrupted"
os._exit(1)
def main(argv=None):
global homedir
# Step One: Parse The Command Line
try:
parser = optparse.OptionParser('%prog [options]')
parser.add_option('-c', '--config', metavar='filename',
help='use a specific configuration file')
parser.add_option("-d", '--fork', action="store_true",
dest="deamonize", help="Deamonize willie")
parser.add_option("-q", '--quit', action="store_true", dest="quit",
help="Gracefully quit Willie")
parser.add_option("-k", '--kill', action="store_true", dest="kill",
help="Kill Willie")
parser.add_option('--exit-on-error', action="store_true", dest="exit_on_error",
help="Exit immediately on every error instead of trying to recover")
parser.add_option("-l", '--list', action="store_true",
dest="list_configs", help="List all config files found")
parser.add_option("-m", '--migrate', action="store_true",
dest="migrate_configs",
help="Migrate config files to the new format")
parser.add_option('--quiet', action="store_true", dest="quiet",
help="Supress all output")
parser.add_option('-w', '--configure-all', action='store_true',
dest='wizard', help='Run the configuration wizard.')
parser.add_option('--configure-modules', action='store_true',
dest='mod_wizard', help='Run the configuration wizard, but only for the module configuration options.')
parser.add_option('--configure-database', action='store_true',
dest='db_wizard', help='Run the configuration wizard, but only for the database configuration options.')
opts, args = parser.parse_args(argv)
if opts.wizard:
wizard('all', opts.config)
return
elif opts.mod_wizard:
wizard('mod', opts.config)
return
elif opts.db_wizard:
wizard('db', opts.config)
return
check_python_version()
if opts.list_configs is not None:
configs = enumerate_configs()
print 'Config files in ~/.willie:'
if len(configs[0]) is 0:
print '\tNone found'
else:
for config in configs:
print '\t%s' % config
print '-------------------------'
return
config_name = opts.config or 'default'
configpath = find_config(config_name)
if not os.path.isfile(configpath):
print "Welcome to Willie!\nI can't seem to find the configuration file, so let's generate it!\n"
if not configpath.endswith('.cfg'):
configpath = configpath + '.cfg'
create_config(configpath)
configpath = find_config(config_name)
try:
config_module = Config(configpath)
except ConfigurationError as e:
stderr(e)
sys.exit(2)
if config_module.core.not_configured:
stderr('Bot is not configured, can\'t start')
# exit with code 2 to prevent auto restart on fail by systemd
sys.exit(2)
if not config_module.has_option('core', 'homedir'):
config_module.dotdir = homedir
config_module.homedir = homedir
else:
homedir = config_module.core.homedir
config_module.dotdir = config_module.core.homedir
if not config_module.core.logdir:
config_module.core.logdir = os.path.join(homedir, 'logs')
logfile = os.path.os.path.join(config_module.logdir, 'stdio.log')
if not os.path.isdir(config_module.logdir):
os.mkdir(config_module.logdir)
if opts.exit_on_error:
config_module.exit_on_error = True
else:
config_module.exit_on_error = False
if opts.quiet is None:
opts.quiet = False
sys.stderr = tools.OutputRedirect(logfile, True, opts.quiet)
sys.stdout = tools.OutputRedirect(logfile, False, opts.quiet)
#Handle --quit, --kill and saving the PID to file
pid_dir = config_module.core.pid_dir or homedir
if opts.config is None:
pid_file_path = os.path.join(pid_dir, 'willie.pid')
else:
basename = os.path.basename(opts.config)
if basename.endswith('.cfg'):
basename = basename[:-4]
pid_file_path = os.path.join(pid_dir, 'willie-%s.pid' % basename)
if os.path.isfile(pid_file_path):
pid_file = open(pid_file_path, 'r')
old_pid = int(pid_file.read())
pid_file.close()
if tools.check_pid(old_pid):
if opts.quit is None and opts.kill is None:
stderr('There\'s already a Willie instance running with this config file')
stderr('Try using the --quit or the --kill options')
sys.exit(1)
elif opts.kill:
stderr('Killing the willie')
os.kill(old_pid, signal.SIGKILL)
sys.exit(0)
elif opts.quit:
stderr('Signaling Willie to stop gracefully')
if hasattr(signal, 'SIGUSR1'):
os.kill(old_pid, signal.SIGUSR1)
else:
os.kill(old_pid, signal.SIGTERM)
sys.exit(0)
elif not tools.check_pid(old_pid) and (opts.kill or opts.quit):
stderr('Willie is not running!')
sys.exit(1)
elif opts.quit is not None or opts.kill is not None:
stderr('Willie is not running!')
sys.exit(1)
if opts.deamonize is not None:
child_pid = os.fork()
if child_pid is not 0:
sys.exit()
pid_file = open(pid_file_path, 'w')
pid_file.write(str(os.getpid()))
pid_file.close()
config_module.pid_file_path = pid_file_path
# Step Five: Initialise And Run willie
run(config_module)
except KeyboardInterrupt:
print "\n\nInterrupted"
os._exit(1)
def main(argv=None):
global homedir
# Step One: Parse The Command Line
try:
parser = optparse.OptionParser('%prog [options]')
parser.add_option('-c', '--config', metavar='filename',
help='use a specific configuration file')
parser.add_option("-d", '--fork', action="store_true",
dest="deamonize", help="Deamonize willie")
parser.add_option("-q", '--quit', action="store_true", dest="quit",
help="Gracefully quit Willie")
parser.add_option("-k", '--kill', action="store_true", dest="kill",
help="Kill Willie")
parser.add_option('--exit-on-error', action="store_true", dest="exit_on_error",
help="Exit immediately on every error instead of trying to recover")
parser.add_option("-l", '--list', action="store_true",
dest="list_configs", help="List all config files found")
parser.add_option("-m", '--migrate', action="store_true",
dest="migrate_configs",
help="Migrate config files to the new format")
parser.add_option('--quiet', action="store_true", dest="quiet",
help="Supress all output")
parser.add_option('-w', '--configure-all', action='store_true',
dest='wizard', help='Run the configuration wizard.')
parser.add_option('--configure-modules', action='store_true',
dest='mod_wizard', help='Run the configuration wizard, but only for the module configuration options.')
parser.add_option('--configure-database', action='store_true',
dest='db_wizard', help='Run the configuration wizard, but only for the database configuration options.')
opts, args = parser.parse_args(argv)
if opts.wizard:
wizard('all', opts.config)
return
elif opts.mod_wizard:
wizard('mod', opts.config)
return
elif opts.db_wizard:
wizard('db', opts.config)
return
check_python_version()
if opts.list_configs is not None:
configs = enumerate_configs()
print 'Config files in ~/.willie:'
if len(configs[0]) is 0:
print '\tNone found'
else:
for config in configs:
print '\t%s' % config
print '-------------------------'
return
config_name = opts.config or 'default'
configpath = find_config(config_name)
if not os.path.isfile(configpath):
print "Welcome to Willie!\nI can't seem to find the configuration file, so let's generate it!\n"
if not configpath.endswith('.cfg'):
configpath = configpath + '.cfg'
create_config(configpath)
configpath = find_config(config_name)
try:
config_module = Config(configpath)
except ConfigurationError as e:
stderr(e)
sys.exit(2)
if config_module.core.not_configured:
stderr('Bot is not configured, can\'t start')
# exit with code 2 to prevent auto restart on fail by systemd
sys.exit(2)
if not config_module.has_option('core', 'homedir'):
config_module.dotdir = homedir
config_module.homedir = homedir
else:
homedir = config_module.core.homedir
config_module.dotdir = config_module.core.homedir
if not config_module.core.logdir:
config_module.core.logdir = os.path.join(homedir, 'logs')
logfile = os.path.os.path.join(config_module.logdir, 'stdio.log')
if not os.path.isdir(config_module.logdir):
os.mkdir(config_module.logdir)
if opts.exit_on_error:
config_module.exit_on_error = True
else:
config_module.exit_on_error = False
if opts.quiet is None:
opts.quiet = False
sys.stderr = tools.OutputRedirect(logfile, True, opts.quiet)
sys.stdout = tools.OutputRedirect(logfile, False, opts.quiet)
#Handle --quit, --kill and saving the PID to file
pid_dir = config_module.core.pid_dir or homedir
if opts.config is None:
pid_file_path = os.path.join(pid_dir, 'willie.pid')
else:
basename = os.path.basename(opts.config)
if basename.endswith('.cfg'):
basename = basename[:-4]
pid_file_path = os.path.join(pid_dir, 'willie-%s.pid' % basename)
if os.path.isfile(pid_file_path):
pid_file = open(pid_file_path, 'r')
old_pid = int(pid_file.read())
pid_file.close()
if tools.check_pid(old_pid):
if opts.quit is None and opts.kill is None:
stderr('There\'s already a Willie instance running with this config file')
stderr('Try using the --quit or the --kill options')
sys.exit(1)
elif opts.kill:
stderr('Killing the willie')
os.kill(old_pid, signal.SIGKILL)
sys.exit(0)
elif opts.quit:
stderr('Signaling Willie to stop gracefully')
if hasattr(signal, 'SIGUSR1'):
os.kill(old_pid, signal.SIGUSR1)
else:
os.kill(old_pid, signal.SIGTERM)
sys.exit(0)
elif not tools.check_pid(old_pid) and (opts.kill or opts.quit):
stderr('Willie is not running!')
sys.exit(1)
elif opts.quit is not None or opts.kill is not None:
stderr('Willie is not running!')
sys.exit(1)
if opts.deamonize is not None:
child_pid = os.fork()
if child_pid is not 0:
sys.exit()
pid_file = open(pid_file_path, 'w')
pid_file.write(str(os.getpid()))
pid_file.close()
config_module.pid_file_path = pid_file_path
# Step Five: Initialise And Run willie
run(config_module)
except KeyboardInterrupt:
print "\n\nInterrupted"
os._exit(1)
def main(argv=None):
# Step One: Parse The Command Line
try:
parser = optparse.OptionParser('%prog [options]')
parser.add_option('-c', '--config', metavar='filename', help='use a specific configuration file')
parser.add_option("-d", '--fork', action="store_true", dest="deamonize", help="Deamonize willie")
parser.add_option("-q", '--quit', action="store_true", dest="quit", help="Gracefully quit Willie")
parser.add_option("-k", '--kill', action="store_true", dest="kill", help="Kill Willie")
parser.add_option("-l", '--list', action="store_true", dest="list_configs", help="List all config files found")
parser.add_option("-m", '--migrate', action="store_true", dest="migrate_configs", help="Migrate config files to the new format")
parser.add_option('--quiet', action="store_true", dest="quiet", help="Supress all output")
parser.add_option('-w', '--configure-all', action='store_true', dest='wizard', help='Run the configuration wizard.')
parser.add_option('--configure-modules', action='store_true', dest='mod_wizard', help='Run the configuration wizard, but only for the module configuration options.')
parser.add_option('--configure-database', action='store_true', dest='db_wizard', help='Run the configuration wizard, but only for the database configuration options.')
opts, args = parser.parse_args(argv)
if opts.wizard:
wizard('all', opts.config)
return
elif opts.mod_wizard:
wizard('mod', opts.config)
return
elif opts.db_wizard:
wizard('db', opts.config)
return
check_python_version()
if opts.list_configs is not None:
configs = enumerate_configs()
stdout('Config files in ~/.willie:')
if len(configs[0]) is 0:
stdout('\tNone found')
else:
for config in configs[0]:
stdout('\t%s' % config)
stdout('-------------------------')
stdout('Config files in ~/.jenni:')
if len(configs[1]) is 0:
stdout('\tNone found')
else:
for config in configs[1]:
stdout('\t%s' % config)
stdout('-------------------------')
stdout('Config files in ~/.phenny:')
if len(configs[2]) is 0:
stdout('\tNone found')
else:
for config in configs[2]:
stdout('\t%s' % config)
stdout('-------------------------')
return
config_name = opts.config or 'default'
if opts.migrate_configs is not None:
configpath = find_config(config_name, '.py')
new_configpath = configpath[:-2]+'cfg'
if os.path.isfile(new_configpath):
valid_answer = False
while not valid_answer:
answer = raw_input('Warning, new config file already exists. Overwrite? [y/n]')
if answer is 'n' or answer == 'no':
return
elif answer == 'y' or answer == 'yes':
valid_answer = True
old_cfg = imp.load_source('Config', configpath)
new_cfg = Config(new_configpath, load=False)
new_cfg.add_section('core')
for attrib in dir(old_cfg):
if not attrib.startswith('_'):
value = getattr(old_cfg, attrib)
if value is None:
continue #Skip NoneTypes
if type(value) is list: #Parse lists
parsed_value = ','.join(value)
else:
parsed_value = str(value)
if attrib == 'password':
attrib = 'nickserv_password'
if attrib == 'serverpass':
attrib = 'server_password'
setattr(new_cfg.core, attrib, parsed_value)
new_cfg.save()
print 'Configuration migrated sucessfully, starting Willie'
configpath = find_config(config_name)
if not os.path.isfile(configpath):
stdout("Welcome to Willie!\nI can't seem to find the configuration file, so let's generate it!\n")
if not configpath.endswith('.cfg'):
configpath = configpath + '.cfg'
create_config(configpath)
configpath = find_config(config_name)
try:
config_module = Config(configpath)
except ConfigurationError as e:
stderr(e)
sys.exit(1)
config_module.dotdir = dotdir
if not config_module.core.logdir:
config_module.core.logdir = os.path.join(dotdir, 'logs')
logfile = os.path.os.path.join(config_module.logdir, 'stdio.log')
if not os.path.isdir(config_module.logdir):
os.mkdir(config_module.logdir)
if opts.quiet is None:
opts.quiet = False
sys.stderr = tools.OutputRedirect(logfile, True, opts.quiet)
sys.stdout = tools.OutputRedirect(logfile, False, opts.quiet)
#Handle --quit, --kill and saving the PID to file
if opts.config is None:
pid_file_path = os.path.join(dotdir, '.pid-default')
else:
pid_file_path = os.path.join(dotdir, '.pid-%s' % opts.config)
if os.path.isfile(pid_file_path):
pid_file = open(pid_file_path, 'r')
old_pid = int(pid_file.read())
pid_file.close()
if tools.check_pid(old_pid):
if opts.quit is None and opts.kill is None:
stderr('There\'s already a Willie instance running with this config file')
stderr('Try using the --quit or the --kill options')
sys.exit(1)
elif opts.kill:
stderr('Killing the willie')
os.kill(old_pid, signal.SIGKILL)
sys.exit(0)
elif opts.quit:
stderr('Singaling Willie to stop gracefully')
os.kill(old_pid, signal.SIGUSR1)
sys.exit(0)
elif not tools.check_pid(old_pid) and (opts.kill or opts.quit):
stderr('Willie is not running!')
sys.exit(1)
elif opts.quit is not None or opts.kill is not None:
stderr('Willie is not running!')
sys.exit(1)
if opts.deamonize is not None:
child_pid = os.fork()
if child_pid is not 0:
sys.exit()
pid_file = open(pid_file_path, 'w')
pid_file.write(str(os.getpid()))
pid_file.close()
config_module.pid_file_path = pid_file_path
# Step Five: Initialise And Run willie
run(config_module)
except KeyboardInterrupt:
stdout("\n\nInterrupted")
os._exit(1)
def db():
config = Config('', False)
config.core.db_filename = db_filename
db = WillieDB(config)
# TODO add tests to ensure db creation works properly, too.
return db
def main(argv=None):
global homedir
# Step One: Parse The Command Line
try:
parser = argparse.ArgumentParser(description='Willie IRC Bot',
usage='%(prog)s [options]')
parser.add_argument('-c', '--config', metavar='filename',
help='use a specific configuration file')
parser.add_argument("-d", '--fork', action="store_true",
dest="deamonize", help="Deamonize willie")
parser.add_argument("-q", '--quit', action="store_true", dest="quit",
help="Gracefully quit Willie")
parser.add_argument("-k", '--kill', action="store_true", dest="kill",
help="Kill Willie")
parser.add_argument('--exit-on-error', action="store_true",
dest="exit_on_error", help=(
"Exit immediately on every error instead of "
"trying to recover"))
parser.add_argument("-l", '--list', action="store_true",
dest="list_configs",
help="List all config files found")
parser.add_argument("-m", '--migrate', action="store_true",
dest="migrate_configs",
help="Migrate config files to the new format")
parser.add_argument('--quiet', action="store_true", dest="quiet",
help="Supress all output")
parser.add_argument('-w', '--configure-all', action='store_true',
dest='wizard', help='Run the configuration wizard.')
parser.add_argument('--configure-modules', action='store_true',
dest='mod_wizard', help=(
'Run the configuration wizard, but only for the '
'module configuration options.'))
parser.add_argument('--allow-root', action="store_true", dest="root",
help="Allow Willie to run as root or administrator.")
parser.add_argument('-v', '--version', action="store_true",
dest="version", help="Show version number and exit")
opts = parser.parse_args()
# Step Two: "Do not run as root" checks.
if not opts.root:
try:
# Linux/Mac
if os.getuid() == 0 or os.geteuid() == 0:
stderr('Error: Do not run Willie with root privileges.')
sys.exit(1)
except AttributeError:
# Windows
if os.environ.get("USERNAME") == "Administrator":
stderr('Error: Do not run Willie as Administrator.')
sys.exit(1)
if opts.version:
py_ver = '%s.%s.%s' % (sys.version_info.major,
sys.version_info.minor,
sys.version_info.micro)
print('Willie %s (running on python %s)' % (__version__, py_ver))
print('http://willie.dftba.net/')
return
elif opts.wizard:
wizard('all', opts.config)
return
elif opts.mod_wizard:
wizard('mod', opts.config)
return
if opts.list_configs:
configs = enumerate_configs()
print('Config files in ~/.willie:')
if len(configs) is 0:
print('\tNone found')
else:
for config in configs:
print('\t%s' % config)
print('-------------------------')
return
config_name = opts.config or 'default'
configpath = find_config(config_name)
if not os.path.isfile(configpath):
print("Welcome to Willie!\nI can't seem to find the configuration file, so let's generate it!\n")
if not configpath.endswith('.cfg'):
configpath = configpath + '.cfg'
create_config(configpath)
configpath = find_config(config_name)
try:
config_module = Config(configpath)
except ConfigurationError as e:
stderr(e)
sys.exit(2)
if config_module.core.not_configured:
stderr('Bot is not configured, can\'t start')
# exit with code 2 to prevent auto restart on fail by systemd
sys.exit(2)
if not config_module.has_option('core', 'homedir'):
config_module.dotdir = homedir
config_module.homedir = homedir
else:
homedir = config_module.core.homedir
config_module.dotdir = config_module.core.homedir
if not config_module.core.logdir:
config_module.core.logdir = os.path.join(homedir, 'logs')
logfile = os.path.os.path.join(config_module.logdir, 'stdio.log')
if not os.path.isdir(config_module.logdir):
os.mkdir(config_module.logdir)
config_module.exit_on_error = opts.exit_on_error
config_module._is_deamonized = opts.deamonize
sys.stderr = tools.OutputRedirect(logfile, True, opts.quiet)
sys.stdout = tools.OutputRedirect(logfile, False, opts.quiet)
# Handle --quit, --kill and saving the PID to file
pid_dir = config_module.core.pid_dir or homedir
if opts.config is None:
pid_file_path = os.path.join(pid_dir, 'willie.pid')
else:
basename = os.path.basename(opts.config)
if basename.endswith('.cfg'):
basename = basename[:-4]
pid_file_path = os.path.join(pid_dir, 'willie-%s.pid' % basename)
if os.path.isfile(pid_file_path):
with open(pid_file_path, 'r') as pid_file:
try:
old_pid = int(pid_file.read())
except ValueError:
old_pid = None
if old_pid is not None and tools.check_pid(old_pid):
if not opts.quit and not opts.kill:
stderr('There\'s already a Willie instance running with this config file')
stderr('Try using the --quit or the --kill options')
sys.exit(1)
elif opts.kill:
stderr('Killing the willie')
os.kill(old_pid, signal.SIGKILL)
sys.exit(0)
elif opts.quit:
stderr('Signaling Willie to stop gracefully')
if hasattr(signal, 'SIGUSR1'):
os.kill(old_pid, signal.SIGUSR1)
else:
os.kill(old_pid, signal.SIGTERM)
sys.exit(0)
elif old_pid is None or (not tools.check_pid(old_pid)
and (opts.kill or opts.quit)):
stderr('Willie is not running!')
sys.exit(1)
elif opts.quit or opts.kill:
stderr('Willie is not running!')
sys.exit(1)
if opts.deamonize:
child_pid = os.fork()
if child_pid is not 0:
sys.exit()
with open(pid_file_path, 'w') as pid_file:
pid_file.write(str(os.getpid()))
config_module.pid_file_path = pid_file_path
# Step Five: Initialise And Run willie
run(config_module)
except KeyboardInterrupt:
print("\n\nInterrupted")
os._exit(1)
def main(argv=None):
global homedir
# Step One: Parse The Command Line
try:
parser = argparse.ArgumentParser(description='Willie IRC Bot',
usage='%(prog)s [options]')
parser.add_argument('-c', '--config', metavar='filename',
help='use a specific configuration file')
parser.add_argument("-d", '--fork', action="store_true",
dest="deamonize", help="Deamonize willie")
parser.add_argument("-q", '--quit', action="store_true", dest="quit",
help="Gracefully quit Willie")
parser.add_argument("-k", '--kill', action="store_true", dest="kill",
help="Kill Willie")
parser.add_argument('--exit-on-error', action="store_true",
dest="exit_on_error", help=(
"Exit immediately on every error instead of "
"trying to recover"))
parser.add_argument("-l", '--list', action="store_true",
dest="list_configs",
help="List all config files found")
parser.add_argument("-m", '--migrate', action="store_true",
dest="migrate_configs",
help="Migrate config files to the new format")
parser.add_argument('--quiet', action="store_true", dest="quiet",
help="Supress all output")
parser.add_argument('-w', '--configure-all', action='store_true',
dest='wizard', help='Run the configuration wizard.')
parser.add_argument('--configure-modules', action='store_true',
dest='mod_wizard', help=(
'Run the configuration wizard, but only for the '
'module configuration options.'))
parser.add_argument('-v', '--version', action="store_true",
dest="version", help="Show version number and exit")
opts = parser.parse_args()
# Step Two: "Do not run as root" checks.
try:
# Linux/Mac
if os.getuid() == 0 or os.geteuid() == 0:
stderr('Error: Do not run Willie with root privileges.')
sys.exit(1)
except AttributeError:
# Windows
if os.environ.get("USERNAME") == "Administrator":
stderr('Error: Do not run Willie as Administrator.')
sys.exit(1)
if opts.version:
py_ver = '%s.%s.%s' % (sys.version_info.major,
sys.version_info.minor,
sys.version_info.micro)
print('Willie %s (running on python %s)' % (__version__, py_ver))
print('http://willie.dftba.net/')
return
elif opts.wizard:
wizard('all', opts.config)
return
elif opts.mod_wizard:
wizard('mod', opts.config)
return
if opts.list_configs:
configs = enumerate_configs()
print('Config files in ~/.willie:')
if len(configs) is 0:
print('\tNone found')
else:
for config in configs:
print('\t%s' % config)
print('-------------------------')
return
config_name = opts.config or 'default'
configpath = find_config(config_name)
if not os.path.isfile(configpath):
print("Welcome to Willie!\nI can't seem to find the configuration file, so let's generate it!\n")
if not configpath.endswith('.cfg'):
configpath = configpath + '.cfg'
create_config(configpath)
configpath = find_config(config_name)
try:
config_module = Config(configpath)
except ConfigurationError as e:
stderr(e)
sys.exit(2)
if config_module.core.not_configured:
stderr('Bot is not configured, can\'t start')
# exit with code 2 to prevent auto restart on fail by systemd
sys.exit(2)
if not config_module.has_option('core', 'homedir'):
config_module.dotdir = homedir
config_module.homedir = homedir
else:
homedir = config_module.core.homedir
config_module.dotdir = config_module.core.homedir
if not config_module.core.logdir:
config_module.core.logdir = os.path.join(homedir, 'logs')
logfile = os.path.os.path.join(config_module.logdir, 'stdio.log')
if not os.path.isdir(config_module.logdir):
os.mkdir(config_module.logdir)
config_module.exit_on_error = opts.exit_on_error
config_module._is_deamonized = opts.deamonize
sys.stderr = tools.OutputRedirect(logfile, True, opts.quiet)
sys.stdout = tools.OutputRedirect(logfile, False, opts.quiet)
# Handle --quit, --kill and saving the PID to file
pid_dir = config_module.core.pid_dir or homedir
if opts.config is None:
pid_file_path = os.path.join(pid_dir, 'willie.pid')
else:
basename = os.path.basename(opts.config)
if basename.endswith('.cfg'):
basename = basename[:-4]
pid_file_path = os.path.join(pid_dir, 'willie-%s.pid' % basename)
if os.path.isfile(pid_file_path):
with open(pid_file_path, 'r') as pid_file:
try:
old_pid = int(pid_file.read())
except ValueError:
old_pid = None
if old_pid is not None and tools.check_pid(old_pid):
if not opts.quit and not opts.kill:
stderr('There\'s already a Willie instance running with this config file')
stderr('Try using the --quit or the --kill options')
sys.exit(1)
elif opts.kill:
stderr('Killing the willie')
os.kill(old_pid, signal.SIGKILL)
sys.exit(0)
elif opts.quit:
stderr('Signaling Willie to stop gracefully')
if hasattr(signal, 'SIGUSR1'):
os.kill(old_pid, signal.SIGUSR1)
else:
os.kill(old_pid, signal.SIGTERM)
sys.exit(0)
elif not tools.check_pid(old_pid) and (opts.kill or opts.quit):
stderr('Willie is not running!')
sys.exit(1)
elif opts.quit or opts.kill:
stderr('Willie is not running!')
sys.exit(1)
if opts.deamonize:
child_pid = os.fork()
if child_pid is not 0:
sys.exit()
with open(pid_file_path, 'w') as pid_file:
pid_file.write(str(os.getpid()))
config_module.pid_file_path = pid_file_path
# Step Five: Initialise And Run willie
run(config_module)
except KeyboardInterrupt:
print("\n\nInterrupted")
os._exit(1)
