summaryrefslogtreecommitdiff
path: root/jobs/EmailChecker.py
blob: 11f1f869d93a6e0782a97d400a2a4921cd3bbc6c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#!/usr/bin/env python

from __future__ import absolute_import
from builtins import str
import os
import base64
import datetime 
import logging

import imaplib

from . import JobBase

class EmailChecker(JobBase.JobBase):
    def executeEvery(self):
        return JobBase.JobFrequency.HOUR
    def notifyOnFailureEvery(self):
        return JobBase.JobFailureNotificationFrequency.EVERYTIME
    def numberFailuresBeforeNotification(self):
        return JobBase.JobFailureCountMinimumBeforeNotification.ONE
    def execute(self):
        if self.config.getboolean('email', 'nomail'):
            logging.debug("Pretending the EmailChecker job succeeded.")
            return True

        USER = self.config.get('email', 'user')
        PASS = self.config.get('email', 'pass')

    	logdetails = ""
        
        #Generate a random subject
        subj = base64.b64encode(os.urandom(20))
        logdetails += "Target subject is " + subj + "\n\n"
        
        if not self.sendEmail(subj, "", USER):
            return False
        
        M = imaplib.IMAP4_SSL(self.config.get('email', 'imapserver'))
        M.login(USER, PASS)
        
        #If we have set up a filter to auto-delete messages from ourself
        if self.config.getboolean('email', 'ideletesentmessagesautomatically'):
            logdetails += "Switching to trash\n"
            M.select("[Gmail]/Trash")
        
        criteria = '(FROM "'+USER+'" SINCE "'+(datetime.date.today() - datetime.timedelta(hours=24)).strftime("%d-%b-%Y")+'")'
        logdetails += "Criteria: " + criteria + "\n"
        typ, data = M.search(None, criteria)
        
        foundSubject = False
        for num in data[0].split():
            logdetails += "Found IMAP item" + str(num) + "\n"
            typ, data = M.fetch(num, '(BODY.PEEK[HEADER.FIELDS (Subject)])')
            logdetails += "IMAP details: " + str(data) + "\n"
            if subj in data[0][1]:
                logdetails += "Found the target subject!!\n"
                foundSubject = True
        M.close()
        M.logout()

        self.logdetails = logdetails
        return foundSubject
    def onFailure(self):
        return self.sendEmail("Email Fetch Failure", self.logdetails)