|
|
|
@ -1,49 +1,56 @@
|
|
|
|
|
from os import _exit
|
|
|
|
|
from sys import exit
|
|
|
|
|
from random import randint
|
|
|
|
|
from time import sleep
|
|
|
|
|
|
|
|
|
|
import config as cfg
|
|
|
|
|
from irc import IRC
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class Ramen:
|
|
|
|
|
"""Functions for ramenkan."""
|
|
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
self.irc = IRC()
|
|
|
|
|
self.irc.is_debug = cfg.debug
|
|
|
|
|
self.bot_nick = cfg.bot_nick
|
|
|
|
|
self.admin_user = cfg.admin_user
|
|
|
|
|
self.admin_code = cfg.admin_code
|
|
|
|
|
self.req_prefix = cfg.req_prefix
|
|
|
|
|
irc = IRC()
|
|
|
|
|
irc.debug = cfg.debug
|
|
|
|
|
|
|
|
|
|
def attach(self, socket):
|
|
|
|
|
self.sock = socket
|
|
|
|
|
def run(self):
|
|
|
|
|
"""Connect to the server and channels, initialise the listener in the
|
|
|
|
|
application loop."""
|
|
|
|
|
self.irc.connect(cfg.server, cfg.bot_nick)
|
|
|
|
|
self.irc.join_channels(cfg.channels)
|
|
|
|
|
while 1:
|
|
|
|
|
sleep(2)
|
|
|
|
|
data = self.irc.receive(debug=cfg.debug)
|
|
|
|
|
self.irc.keep_alive(data)
|
|
|
|
|
self.msg = self.irc.parse_msg(data, cfg.req_prefix)
|
|
|
|
|
for chan in cfg.channels:
|
|
|
|
|
self.handle(chan)
|
|
|
|
|
|
|
|
|
|
def handle(self, channel, data):
|
|
|
|
|
"""Listen for requests in a channel and match responses."""
|
|
|
|
|
self.msg = self.irc.parse_msg(data, self.req_prefix)
|
|
|
|
|
|
|
|
|
|
if self.msg["req_chan"] == self.bot_nick:
|
|
|
|
|
def handle(self, channel):
|
|
|
|
|
"""Listen for requests in a channel and pass them to handler
|
|
|
|
|
functions."""
|
|
|
|
|
if self.msg["req_chan"] == cfg.bot_nick:
|
|
|
|
|
# Respond to some commands only from admin user
|
|
|
|
|
if self.msg["user"].lower() == self.admin_user.lower() and \
|
|
|
|
|
self.admin_code in self.msg["req"]:
|
|
|
|
|
self.handle_admin_req(self.msg["req"], self.admin_user)
|
|
|
|
|
if self.msg["user"].lower() == cfg.admin_user.lower() and \
|
|
|
|
|
cfg.admin_code in self.msg["req"]:
|
|
|
|
|
self.handle_admin_req(self.msg["req"], cfg.admin_user)
|
|
|
|
|
|
|
|
|
|
# Respond only in the channel the request was made
|
|
|
|
|
if channel == self.msg["req_chan"]:
|
|
|
|
|
# General commands
|
|
|
|
|
if self.msg["req"] == "rollcall" or self.msg["req"] == "help":
|
|
|
|
|
self.handle_rollcall()
|
|
|
|
|
elif self.msg["req"] == ("water " + self.bot_nick):
|
|
|
|
|
elif self.msg["req"] == ("water " + cfg.bot_nick):
|
|
|
|
|
self.handle_water()
|
|
|
|
|
elif self.msg["req"] == ("botsnack " + self.bot_nick):
|
|
|
|
|
elif self.msg["req"] == ("botsnack " + cfg.bot_nick):
|
|
|
|
|
self.handle_botsnack()
|
|
|
|
|
|
|
|
|
|
def handle_admin_req(self, req, admin_user):
|
|
|
|
|
"""Perform admin functions."""
|
|
|
|
|
if "exit" in req:
|
|
|
|
|
self.irc.send("PRIVMSG", "Okay, okay, I'll leave. (´・ω・`)", \
|
|
|
|
|
recvr=admin_user, sock=self.sock)
|
|
|
|
|
self.irc.send("QUIT", ":noodling off", sock=self.sock)
|
|
|
|
|
_exit(0)
|
|
|
|
|
recvr=admin_user)
|
|
|
|
|
self.irc.send("QUIT", ":noodling off")
|
|
|
|
|
exit("Shutting down ...")
|
|
|
|
|
|
|
|
|
|
def handle_rollcall(self):
|
|
|
|
|
resp = (
|
|
|
|
@ -53,8 +60,7 @@ class Ramen:
|
|
|
|
|
"!help "
|
|
|
|
|
"Support: +81 012-700-1MIO どうぞめしあがれ。"
|
|
|
|
|
)
|
|
|
|
|
self.irc.send("PRIVMSG", resp, recvr=self.msg["req_chan"], \
|
|
|
|
|
sock=self.sock)
|
|
|
|
|
self.irc.send("PRIVMSG", resp, recvr=self.msg["req_chan"])
|
|
|
|
|
|
|
|
|
|
def handle_water(self):
|
|
|
|
|
resp = [
|
|
|
|
@ -64,8 +70,12 @@ class Ramen:
|
|
|
|
|
"Water Level [/////////] 200% - Thanks! (^▽^)"
|
|
|
|
|
]
|
|
|
|
|
self.irc.send("PRIVMSG", resp[randint(0, len(resp)-1)], \
|
|
|
|
|
recvr=self.msg["req_chan"], sock=self.sock)
|
|
|
|
|
recvr=self.msg["req_chan"])
|
|
|
|
|
|
|
|
|
|
def handle_botsnack(self):
|
|
|
|
|
self.irc.send("PRIVMSG", "Ramen time anytime! ヽ(´▽`)/", \
|
|
|
|
|
recvr=self.msg["req_chan"], sock=self.sock)
|
|
|
|
|
recvr=self.msg["req_chan"])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
app = Ramen()
|
|
|
|
|
app.run()
|
|
|
|
|