72 lines
2.6 KiB
Python
72 lines
2.6 KiB
Python
|
from os import _exit
|
|||
|
|
from random import randint
|
|||
|
|
|
|||
|
|
import config as cfg
|
|||
|
|
from irc import IRC
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Ramen:
|
|||
|
|
|
|||
|
|
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
|
|||
|
|
|
|||
|
|
def attach(self, socket):
|
|||
|
|
self.sock = socket
|
|||
|
|
|
|||
|
|
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:
|
|||
|
|
# 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)
|
|||
|
|
|
|||
|
|
# 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):
|
|||
|
|
self.handle_water()
|
|||
|
|
elif self.msg["req"] == ("botsnack " + self.bot_nick):
|
|||
|
|
self.handle_botsnack()
|
|||
|
|
|
|||
|
|
def handle_admin_req(self, req, admin_user):
|
|||
|
|
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)
|
|||
|
|
|
|||
|
|
def handle_rollcall(self):
|
|||
|
|
resp = (
|
|||
|
|
"一、二、三、らーめん缶！ "
|
|||
|
|
"Hello, I am a ramen vending machine. "
|
|||
|
|
"Please type a code for service: "
|
|||
|
|
"!help "
|
|||
|
|
"Support: +81 012-700-1MIO どうぞめしあがれ。"
|
|||
|
|
)
|
|||
|
|
self.irc.send("PRIVMSG", resp, recvr=self.msg["req_chan"], \
|
|||
|
|
sock=self.sock)
|
|||
|
|
|
|||
|
|
def handle_water(self):
|
|||
|
|
resp = [
|
|||
|
|
("\x01ACTION happily pours the hot liquid into a bowl of noodles "
|
|||
|
|
"and offers it to ") + self.msg["nick"] + "\x01",
|
|||
|
|
"（ ^_^）o自自o（^_^ ）Cheers!",
|
|||
|
|
"Water Level [/////////] 200% - Thanks! (^▽^)"
|
|||
|
|
]
|
|||
|
|
self.irc.send("PRIVMSG", resp[randint(0, len(resp)-1)], \
|
|||
|
|
recvr=self.msg["req_chan"], sock=self.sock)
|
|||
|
|
|
|||
|
|
def handle_botsnack(self):
|
|||
|
|
self.irc.send("PRIVMSG", "Ramen time anytime! ヽ(´▽`)/", \
|
|||
|
|
recvr=self.msg["req_chan"], sock=self.sock)
|