bitbot-3.11-fork/start.py

57 lines
1.8 KiB
Python
Executable file

#!/usr/bin/env python3
import argparse, sys, time
import IRCBot, Config, Database
def bool_input(s):
result = input("%s (Y/n): " % s)
return not result or result[0].lower() in ["", "y"]
arg_parser = argparse.ArgumentParser(
description="Python3 event-driven asynchronous modular IRC bot")
arg_parser.add_argument("--config", "-c", default="bot.conf",
help="Location of the JSON config file")
arg_parser.add_argument("--database", "-d", default="bot.db",
help="Location of the sqlite3 database file")
arg_parser.add_argument("--verbose", "-v", action="store_true")
args = arg_parser.parse_args()
bot = IRCBot.Bot()
database = Database.Database(bot, args.database)
config_object = Config.Config(bot, args.config)
bot.database = database
bot.config_object = config_object
bot.args = args
bot.modules.load_modules()
server_details = database.servers.get_all()
servers = []
for server_detail in server_details:
server = bot.add_server(*server_detail)
if not server == None:
servers.append(server)
if len(servers):
bot.events.on("boot").on("done").call()
for server in servers:
if not bot.connect(server):
sys.stderr.write("failed to connect to '%s', exiting\r\n" % (
str(server)))
sys.exit(1)
bot.run()
else:
try:
if bool_input("no servers found, add one?"):
hostname = input("hostname: ")
port = int(input("port: "))
tls = bool_input("tls?")
password = input("password?: ")
ipv4 = bool_input("ipv4?")
nickname = input("nickname: ")
username = input("username: ")
realname = input("realname: ")
database.add_server(hostname, port, password, ipv4,
tls, nickname, username, realname)
except KeyboardInterrupt:
print()
pass