gplugs.mail

gozerbots mail plugin.

exception gplugs.mail.Mailservernotset

Bases: exceptions.Exception

exception to raise if mail server is not set

gplugs.mail.domail(mailto, txt, fromm=None, mailserver=None, subject=None)

sent the mail

gplugs.mail.getemail(username)

get email from the userstate.

gplugs.mail.handle_mail(bot, ievent)

mail result from pipeline to the user giving the command

gplugs.mail.handle_mailtime(bot, ievent)

mail log since a given time

gplugs.mail.handle_re(bot, ievent)

mail log since last spoken

CODE

# plugs/mail.py
#
#

""" gozerbots mail plugin. """

__depend__ = ['log', ]
__gendocfirst__ = ['mail-set', ]
__status__ = "seem"

gozerbot imports

from gozerbot.persist.persiststate import UserState
from gozerbot.commands import cmnds
from gozerbot.examples import examples
from gozerbot.generic import waitforqueue, rlog, hourmin, strtotime
from gozerbot.users import users
from gozerbot.config import config
from gozerbot.plugins import plugins
from gozerbot.plughelp import plughelp
from gozerbot.aliases import aliasset
from gozerbot.tests import tests
from gplugs.log import logs

basic imports

import smtplib
import random
import time
import types

plughelp

plughelp.add('mail', 'mail related commands')

defines

rendezvous = {}

getemail function

def getemail(username):
    """ get email from the userstate. """
    try:
        state = UserState(username)
        if state: return state['email']
    except KeyError: return

exceptions

class Mailservernotset(Exception):

    """ exception to raise if mail server is not set """

    def __str__(self):
        return "config['mailserver'] is not set"

domail function

def domail(mailto, txt, fromm=None, mailserver=None, subject=None):
    """ sent the mail """
    if not txt: rlog(10, 'mail', 'no text to send') ; return
    if fromm: fromaddr = fromm
    else:
        fromaddr = config['mailfrom']
        if not fromaddr: fromaddr = 'gozerbot@gozerbot.org'
    if not mailserver: mailserver = config['mailserver']
    if not mailserver: raise Mailservernotset
    if type(txt) != types.ListType: txt = [txt, ]
    msg = ("From: %s\r\nTo: %s\r\nSubject: %s\r\n\r\n" % (fromaddr, mailto, subject))
    for i in txt: msg += "%s\r\n" % i
    server = smtplib.SMTP(mailserver)
    server.sendmail(fromaddr, mailto, msg)

mail command

def handle_mail(bot, ievent):
    """ mail result from pipeline to the user giving the command """
    if not ievent.inqueue: ievent.reply('use mail in a pipeline') ; return
    ievent.reply('waiting for input to mail')
    result = waitforqueue(ievent.inqueue, 10)
    if not result: ievent.reply('no data to mail') ; return
    username = users.getname(ievent.userhost)
    email = getemail(username)
    if email:
        try:
            sub = "output of %s" % ievent.origtxt
            domail(email, result, subject=sub)
        except Exception, ex: ievent.reply("can't send email: %s" % str(ex)) ; return
        ievent.reply('%s lines sent' % len(result))
    else: ievent.reply("can't get email of %s" % username)

cmnds.add('mail', handle_mail, ['MAIL', 'OPER'], threaded=True)
examples.add('mail', 'mail pipelined data to user giving the command', 'todo | mail')
tests.add('exceptions | mail')

re command

def handle_re(bot, ievent):
    """ mail log since last spoken """
    if not logs: ievent.reply('log plugin is not enabled') ; return
    if ievent.channel not in logs.loglist: ievent.reply('logging is not enabled in %s' % ievent.channel) ; return
    lastlist = logs.lastspokelist(ievent.channel, ievent.userhost, 100)
    lasttime = time.time()
    gotcha = None
    for i in lastlist[::-1]:
        delta = lasttime - i
        if delta > 600:
            gotcha = i
            break
        else: lasttime = i
    if gotcha:
        result = logs.fromtimewithbot(ievent.channel, gotcha)
        if result:
            username = users.getname(ievent.userhost)
            email = getemail(username)
            if email:
                try:
                    res = []
                    for i in result:
                        if i[2] == 'bot':
                            txt = i[4]
                        else:
                            nr = i[4].find(' ')
                            txt = i[4][nr:].strip()
                        res.append("[%s] <%s> %s" % (hourmin(float(i[1])), i[2], txt))
                    domail(email, res, subject="log of %s" % ievent.channel)
                except Exception, ex:
                    ievent.reply("can't send email: %s" % str(ex))
                    return
                ievent.reply('%s lines sent' % len(result))
                return
            else:
                ievent.reply("can't get email of %s" % username)
                return
    ievent.reply('no data found')

cmnds.add('re', handle_re, ['MAIL', 'OPER'])
examples.add('re', 'mail the log since last spoken word', 're')
tests.add('re', 'lines sent')

mailtime command

def handle_mailtime(bot, ievent):
    """ mail log since a given time """
    if not logs:
        ievent.reply('log plugin is not enabled')
        return
    if ievent.channel not in logs.loglist:
        ievent.reply('logging is not enabled in %s' % ievent.channel)
        return
    fromtime = strtotime(ievent.rest)
    if not fromtime:
        ievent.reply("can't detect time")
        return
    result = logs.fromtimewithbot(ievent.channel, fromtime)
    if result:
        username = users.getname(ievent.userhost)
        email = getemail(username)
        if email:
            try:
                res = []
                for i in result:
                    if i[2] == 'bot': txt = i[4]
                    else: nr = i[4].find(' ') ; txt = i[4][nr:].strip()
                    res.append("[%s] <%s> %s" % (hourmin(float(i[1])), i[2], txt))
                domail(email, res, subject="log of %s" % ievent.channel)
            except Exception, ex:
                ievent.reply("can't send email: %s" % str(ex))
                return
            ievent.reply('%s lines sent' % len(result))
            return
        else:
            ievent.reply("can't get email of %s" % username)
            return
    ievent.reply('no data found')

cmnds.add('mail-time', handle_mailtime, ['MAIL', 'OPER'])
examples.add('mail-time', 'mail the log since given time', 'mail-time 12:00')
tests.add('mail-time 9:00')