|
@@ -4,18 +4,29 @@ import os
|
|
|
|
|
|
from rollbot import Rollbot
|
|
|
from command_system import RollbotMessage
|
|
|
-from config import DB_FILE
|
|
|
+from config import BOTS_LOOKUP, DB_FILE, get_config
|
|
|
|
|
|
try:
|
|
|
os.remove(DB_FILE)
|
|
|
except FileNotFoundError:
|
|
|
pass
|
|
|
-rollbot = Rollbot(logging.getLogger(__name__))
|
|
|
+
|
|
|
+import db
|
|
|
+
|
|
|
+rollbot = Rollbot(
|
|
|
+ plugins=get_config("plugins"),
|
|
|
+ aliases=get_config("aliases"),
|
|
|
+ responses=get_config("responses"),
|
|
|
+ lookup=BOTS_LOOKUP,
|
|
|
+ sleep_time=float(get_config("sleep_time")),
|
|
|
+ callback=None,
|
|
|
+ session_factory=db.session_scope
|
|
|
+)
|
|
|
+db.init_db()
|
|
|
|
|
|
|
|
|
def test_drive(msg, from_admin=True):
|
|
|
- rmsg = RollbotMessage("mock", None, None, "test_group", None, msg)
|
|
|
- rmsg.from_admin = from_admin
|
|
|
+ rmsg = RollbotMessage("mock", None, None, "test_group", None, msg, from_admin)
|
|
|
r =rollbot.run_command(rmsg)
|
|
|
print(msg, ":", r.txt, ",", r.failure)
|
|
|
|