예제 #1
0
파일: test.py 프로젝트: strcrzy/postal
def test_router_two_routes_two_predicates():
    emails = []
    router = Router()
    receipts = []
    updates = []
    for filename in glob('./tests/fixtures/emails/*'):
        with open(filename) as f:
            emails.append(parse_email(f.read()))

    @router.match(from_=".*[email protected].*",
                  subject=r"^Update on your donation to (?P<receiver>.*)$")
    def process_update(msg, receiver=None):
        assert(receiver == 'Maung Win')
        updates.append(msg)

    @router.match(from_=".*[email protected].*",
                  subject=r"^Receipt for your donation to (?P<receiver>.*)$")
    def process_receipt(msg, receiver=None):
        assert(receiver == 'Tu Tu')
        receipts.append(msg)
    assert(router.routes)
    print router.routes
    assert(len(router.routes) is 2)
    for email in emails:
        router.route(email)
    print(receipts)
    assert(len(receipts) is 1)
    receipt = receipts[0]
    assert(receipt['Subject'] == 'Receipt for your donation to Tu Tu')
    assert(len(updates) is 1)
    update = updates[0]
    assert(update['Subject'] == 'Update on your donation to Maung Win')
예제 #2
0
파일: imap.py 프로젝트: strcrzy/postal
 def get_mail(self, delete=True):
     """
     fetches emails from the mailbox. 
     keyword arguments:
     delete - when True, deletes the messages after fetching them.
              defaults to False
     """
     conn = imaplib.IMAP4_SSL(self.host)
     conn.login(self.user, self.password)
     conn.select(self.mailbox)
     search_result, search_data = conn.uid("search", None, "ALL")
     uids = search_data[0].split()
     message_count = len(uids)
     fetch_email = lambda uid: conn.uid('fetch', uid, '(RFC822)')[1][0][1]
     emails = [parse_email(fetch_email(uid)) for uid in uids]
     if(message_count and delete):
         delete_mail(uids)
     conn.expunge()
     conn.close()
     conn.logout()
     return emails
예제 #3
0
파일: test.py 프로젝트: strcrzy/postal
def test_router_single_route_simple_match():
    emails = []
    router = Router()
    matched = []
    for filename in glob('./tests/fixtures/emails/*'):
        with open(filename) as f:
            emails.append(parse_email(f.read()))

    @router.match(from_=".*[email protected].*")
    def process(msg, receiver=None):
        assert(msg)
        matched.append(msg)

    assert(router.routes)
    assert(len(router.routes) is 1)
    for email in emails:
        router.route(email)
    assert(len(matched) is 2)
    receipt = matched[0]
    assert(receipt['Subject'] == 'Receipt for your donation to Tu Tu')
    update = matched[1]
    assert(update['Subject'] == 'Update on your donation to Maung Win')