Track a user's away status in IRCUser

2 files changed, 3 insertions, 0 deletions

diff --git a/IRCLineHandler.py b/IRCLineHandler.py
index 9781581d..b5a486f6 100644
--- a/IRCLineHandler.py
+++ b/IRCLineHandler.py
@@ -449,9 +449,11 @@ class LineHandler(object):
         user = event["server"].get_user(nickname)
         message = event["arbitrary"]
         if message:
+            user.away = True
             self.events.on("received.away.on").call(user=user,
                 server=event["server"], message=message)
         else:
+            user.away = False
             self.events.on("received.away.off").call(user=user,
                 server=event["server"])
 
diff --git a/IRCUser.py b/IRCUser.py
index 4229ad12..696b4a3b 100644
--- a/IRCUser.py
+++ b/IRCUser.py
@@ -12,6 +12,7 @@ class User(object):
         self.bot = bot
         self.channels = set([])
         self.identified_account = None
+        self.away = False
         self.buffer = IRCBuffer.Buffer(bot)
 
     def __repr__(self):