Add EasyMessage and reply interface
This commit is contained in:
parent
8369689a71
commit
301f426cf9
25
xbotlib.py
25
xbotlib.py
@ -8,6 +8,25 @@ from pathlib import Path
|
|||||||
from slixmpp import ClientXMPP
|
from slixmpp import ClientXMPP
|
||||||
|
|
||||||
|
|
||||||
|
class EasyMessage:
|
||||||
|
"""A simple message interface."""
|
||||||
|
|
||||||
|
def __init__(self, message):
|
||||||
|
self.message = message
|
||||||
|
|
||||||
|
@property
|
||||||
|
def body(self):
|
||||||
|
return self.message["body"]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def sender(self):
|
||||||
|
return self.message["from"]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def receiver(self):
|
||||||
|
return self.message["to"]
|
||||||
|
|
||||||
|
|
||||||
class Bot(ClientXMPP):
|
class Bot(ClientXMPP):
|
||||||
CONFIG_FILE = "bot.conf"
|
CONFIG_FILE = "bot.conf"
|
||||||
|
|
||||||
@ -63,7 +82,7 @@ class Bot(ClientXMPP):
|
|||||||
def message(self, message):
|
def message(self, message):
|
||||||
"""Handle message event."""
|
"""Handle message event."""
|
||||||
if message["type"] in ("chat", "normal"):
|
if message["type"] in ("chat", "normal"):
|
||||||
self.react(message)
|
self.react(EasyMessage(message))
|
||||||
|
|
||||||
def session_start(self, event):
|
def session_start(self, event):
|
||||||
"""Handle session_start event."""
|
"""Handle session_start event."""
|
||||||
@ -84,3 +103,7 @@ class Bot(ClientXMPP):
|
|||||||
"""Run the bot."""
|
"""Run the bot."""
|
||||||
self.connect()
|
self.connect()
|
||||||
self.process()
|
self.process()
|
||||||
|
|
||||||
|
def reply(self, to, body, type="chat"):
|
||||||
|
"""Send a message."""
|
||||||
|
self.send_message(mto=to, mbody=body, mtype=type)
|
||||||
|
Reference in New Issue
Block a user