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