{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "import configparser\n",
    "import imaplib\n",
    "import email\n",
    "import io\n",
    "import zipfile\n",
    "import xml.etree.ElementTree\n",
    "import psycopg2\n",
    "import re\n",
    "import datetime"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def fetch_msg(num):\n",
    "    return mailbox.fetch(num, '(RFC822)')[1][0][1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 36,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def xml_of_part(part):\n",
    "    with zipfile.ZipFile(io.BytesIO(part.get_payload(decode=True))) as zf:\n",
    "            fn = zf.infolist()[0].filename\n",
    "            contents = zf.read(fn).decode('utf-8')\n",
    "            return xml.etree.ElementTree.fromstring(contents)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 42,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def xml_of(message):\n",
    "    reports = []\n",
    "    if message.is_multipart():\n",
    "        for p in message.get_payload():\n",
    "            if 'zip' in p.get_content_type():\n",
    "                reports += [xml_of_part(p)]\n",
    "    else:\n",
    "        reports = [xml_of_part(message)]\n",
    "    return reports"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 35,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def extract_report(msg):\n",
    "    pmsg = email.message_from_bytes(msg)\n",
    "    return xml_of(pmsg)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['dmarc.ini']"
      ]
     },
     "execution_count": 38,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "config = configparser.ConfigParser()\n",
    "config.read('dmarc.ini')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 39,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "('OK', [b'174'])"
      ]
     },
     "execution_count": 39,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "mailbox = imaplib.IMAP4(host=config['imap']['server'], \n",
    "                      port=config['imap']['port'])\n",
    "mailbox.starttls()\n",
    "mailbox.login(config['imap']['username'], config['imap']['password'])\n",
    "mailbox.select('INBOX', readonly=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 40,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "('OK',\n",
       " [b'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 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174'])"
      ]
     },
     "execution_count": 40,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "resp, nums = mailbox.search(None, \"ALL\")\n",
    "resp, nums"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 44,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "b'1'\n",
      "b'1' [<Element 'feedback' at 0x7f9ea4237278>]\n",
      "b'2'\n",
      "b'2' [<Element 'feedback' at 0x7f9ea4274138>]\n",
      "b'3'\n",
      "b'3' [<Element 'feedback' at 0x7f9ea4211368>]\n",
      "b'4'\n",
      "b'4' [<Element 'feedback' at 0x7f9ea427ae08>]\n",
      "b'5'\n",
      "b'5' [<Element 'feedback' at 0x7f9ea420d4a8>]\n",
      "b'6'\n",
      "b'6' [<Element 'feedback' at 0x7f9ea42743b8>]\n",
      "b'7'\n",
      "b'7' [<Element 'feedback' at 0x7f9ea4211868>]\n",
      "b'8'\n",
      "b'8' [<Element 'feedback' at 0x7f9ea427ae08>]\n",
      "b'9'\n",
      "b'9' [<Element 'feedback' at 0x7f9ea4274138>]\n",
      "b'10'\n",
      "b'10' [<Element 'feedback' at 0x7f9ea422ff48>]\n",
      "b'11'\n",
      "b'11' [<Element 'feedback' at 0x7f9ea41c6cc8>]\n",
      "b'12'\n",
      "b'12' [<Element 'feedback' at 0x7f9ea427aa98>]\n",
      "b'13'\n",
      "b'13' [<Element 'feedback' at 0x7f9ea42743b8>]\n",
      "b'14'\n",
      "b'14' [<Element 'feedback' at 0x7f9ea427aef8>]\n",
      "b'15'\n",
      "b'15' [<Element 'feedback' at 0x7f9ea4237278>]\n",
      "b'16'\n",
      "b'16' [<Element 'feedback' at 0x7f9ea4237db8>]\n",
      "b'17'\n",
      "b'17' [<Element 'feedback' at 0x7f9ea422ff48>]\n",
      "b'18'\n",
      "b'18' [<Element 'feedback' at 0x7f9ea41c6a48>]\n",
      "b'19'\n",
      "b'19' [<Element 'feedback' at 0x7f9ea41c6cc8>]\n",
      "b'20'\n",
      "b'20' [<Element 'feedback' at 0x7f9ea41c6908>]\n",
      "b'21'\n",
      "b'21' [<Element 'feedback' at 0x7f9ea41c6c28>]\n",
      "b'22'\n",
      "b'22' [<Element 'feedback' at 0x7f9ea41c6f98>]\n",
      "b'23'\n",
      "b'23' [<Element 'feedback' at 0x7f9ea41c6ae8>]\n",
      "b'24'\n",
      "b'24' [<Element 'feedback' at 0x7f9ea41c6f48>]\n",
      "b'25'\n",
      "b'25' [<Element 'feedback' at 0x7f9ea4211228>]\n",
      "b'26'\n",
      "b'26' [<Element 'feedback' at 0x7f9ea4211188>]\n",
      "b'27'\n",
      "b'27' [<Element 'feedback' at 0x7f9ea41c6a48>]\n",
      "b'28'\n",
      "b'28' [<Element 'feedback' at 0x7f9ea41c66d8>]\n",
      "b'29'\n",
      "b'29' [<Element 'feedback' at 0x7f9ea422f598>]\n",
      "b'30'\n",
      "b'30' [<Element 'feedback' at 0x7f9ea4211548>]\n",
      "b'31'\n",
      "b'31' [<Element 'feedback' at 0x7f9ea41c6ae8>]\n",
      "b'32'\n",
      "b'32' [<Element 'feedback' at 0x7f9ea41c6ea8>]\n",
      "b'33'\n",
      "b'33' [<Element 'feedback' at 0x7f9ea41c6c28>]\n",
      "b'34'\n",
      "b'34' [<Element 'feedback' at 0x7f9ea42374f8>]\n",
      "b'35'\n",
      "b'35' [<Element 'feedback' at 0x7f9ea41c6a48>]\n",
      "b'36'\n",
      "b'36' [<Element 'feedback' at 0x7f9ea41c67c8>]\n",
      "b'37'\n",
      "b'37' [<Element 'feedback' at 0x7f9ea4211cc8>]\n",
      "b'38'\n",
      "b'38' [<Element 'feedback' at 0x7f9ea4211318>]\n",
      "b'39'\n",
      "b'39' [<Element 'feedback' at 0x7f9ea426cd18>]\n",
      "b'40'\n",
      "b'40' [<Element 'feedback' at 0x7f9ea41c64f8>]\n",
      "b'41'\n",
      "b'41' [<Element 'feedback' at 0x7f9ea42743b8>]\n",
      "b'42'\n",
      "b'42' [<Element 'feedback' at 0x7f9ea4211318>]\n",
      "b'43'\n",
      "b'43' [<Element 'feedback' at 0x7f9ea427a318>]\n",
      "b'44'\n",
      "b'44' [<Element 'feedback' at 0x7f9ea42b7c78>]\n",
      "b'45'\n",
      "b'45' [<Element 'feedback' at 0x7f9ea42b7d68>]\n",
      "b'46'\n",
      "b'46' [<Element 'feedback' at 0x7f9ea42b7bd8>]\n",
      "b'47'\n",
      "b'47' [<Element 'feedback' at 0x7f9ea42b7c28>]\n",
      "b'48'\n",
      "b'48' [<Element 'feedback' at 0x7f9ea42743b8>]\n",
      "b'49'\n",
      "b'49' [<Element 'feedback' at 0x7f9ea42b79a8>]\n",
      "b'50'\n",
      "b'50' [<Element 'feedback' at 0x7f9ea42b7958>]\n",
      "b'51'\n",
      "b'51' [<Element 'feedback' at 0x7f9ea42b7c78>]\n",
      "b'52'\n",
      "b'52' [<Element 'feedback' at 0x7f9ea46e1408>]\n",
      "b'53'\n",
      "b'53' [<Element 'feedback' at 0x7f9ea46e1b88>]\n",
      "b'54'\n",
      "b'54' [<Element 'feedback' at 0x7f9ea42b7c28>]\n",
      "b'55'\n",
      "b'55' [<Element 'feedback' at 0x7f9ea46e1778>]\n",
      "b'56'\n",
      "b'56' [<Element 'feedback' at 0x7f9ea4274f98>]\n",
      "b'57'\n",
      "b'57' [<Element 'feedback' at 0x7f9ea422ff48>]\n",
      "b'58'\n",
      "b'58' [<Element 'feedback' at 0x7f9ea42480e8>]\n",
      "b'59'\n",
      "b'59' [<Element 'feedback' at 0x7f9ea42b79a8>]\n",
      "b'60'\n",
      "b'60' [<Element 'feedback' at 0x7f9ea427a958>]\n",
      "b'61'\n",
      "b'61' [<Element 'feedback' at 0x7f9ea427aa48>]\n",
      "b'62'\n",
      "b'62' [<Element 'feedback' at 0x7f9ea41c60e8>]\n",
      "b'63'\n",
      "b'63' [<Element 'feedback' at 0x7f9ea41c6d68>]\n",
      "b'64'\n",
      "b'64' [<Element 'feedback' at 0x7f9ea427a548>]\n",
      "b'65'\n",
      "b'65' [<Element 'feedback' at 0x7f9ea427aa48>]\n",
      "b'66'\n",
      "b'66' [<Element 'feedback' at 0x7f9ea41c60e8>]\n",
      "b'67'\n",
      "b'67' [<Element 'feedback' at 0x7f9ea41c64f8>]\n",
      "b'68'\n",
      "b'68' [<Element 'feedback' at 0x7f9ea46f27c8>]\n",
      "b'69'\n",
      "b'69' [<Element 'feedback' at 0x7f9ea427a8b8>]\n",
      "b'70'\n",
      "b'70' [<Element 'feedback' at 0x7f9ea41c64a8>]\n",
      "b'71'\n",
      "b'71' [<Element 'feedback' at 0x7f9ea41c69a8>]\n",
      "b'72'\n",
      "b'72' [<Element 'feedback' at 0x7f9ea41c6a98>]\n",
      "b'73'\n",
      "b'73' [<Element 'feedback' at 0x7f9ea42110e8>]\n",
      "b'74'\n",
      "b'74' [<Element 'feedback' at 0x7f9ea41c6458>]\n",
      "b'75'\n",
      "b'75' [<Element 'feedback' at 0x7f9ea41c6f98>]\n",
      "b'76'\n",
      "b'76' [<Element 'feedback' at 0x7f9ea427a548>]\n",
      "b'77'\n",
      "b'77' [<Element 'feedback' at 0x7f9ea426cdb8>]\n",
      "b'78'\n",
      "b'78' [<Element 'feedback' at 0x7f9ea4274f98>]\n",
      "b'79'\n",
      "b'79' [<Element 'feedback' at 0x7f9ea41c6188>]\n",
      "b'80'\n",
      "b'80' [<Element 'feedback' at 0x7f9ea4211e08>]\n",
      "b'81'\n",
      "b'81' [<Element 'feedback' at 0x7f9ea4237b88>]\n",
      "b'82'\n",
      "b'82' [<Element 'feedback' at 0x7f9ea4237188>]\n",
      "b'83'\n",
      "b'83' [<Element 'feedback' at 0x7f9ea4211868>]\n",
      "b'84'\n",
      "b'84' [<Element 'feedback' at 0x7f9ea46e1408>]\n",
      "b'85'\n",
      "b'85' [<Element 'feedback' at 0x7f9ea4237b38>]\n",
      "b'86'\n",
      "b'86' [<Element 'feedback' at 0x7f9ea42379f8>]\n",
      "b'87'\n",
      "b'87' [<Element 'feedback' at 0x7f9ea46e14a8>]\n",
      "b'88'\n",
      "b'88' [<Element 'feedback' at 0x7f9ea427aae8>]\n",
      "b'89'\n",
      "b'89' [<Element 'feedback' at 0x7f9ea42743b8>]\n",
      "b'90'\n",
      "b'90' [<Element 'feedback' at 0x7f9ea426cf48>]\n",
      "b'91'\n",
      "b'91' [<Element 'feedback' at 0x7f9ebd87d958>]\n",
      "b'92'\n",
      "b'92' [<Element 'feedback' at 0x7f9ea4248048>]\n",
      "b'93'\n",
      "b'93' [<Element 'feedback' at 0x7f9ea4274ea8>]\n",
      "b'94'\n",
      "b'94' [<Element 'feedback' at 0x7f9ea4248138>]\n",
      "b'95'\n",
      "b'95' [<Element 'feedback' at 0x7f9ebd87d958>]\n",
      "b'96'\n",
      "b'96' [<Element 'feedback' at 0x7f9ea4248048>]\n",
      "b'97'\n",
      "b'97' [<Element 'feedback' at 0x7f9ea4237688>]\n",
      "b'98'\n",
      "b'98' [<Element 'feedback' at 0x7f9ea4248138>]\n",
      "b'99'\n",
      "b'99' [<Element 'feedback' at 0x7f9ebd87d958>]\n",
      "b'100'\n",
      "b'100' [<Element 'feedback' at 0x7f9ea427a278>]\n",
      "b'101'\n",
      "b'101' [<Element 'feedback' at 0x7f9ea426cf48>]\n",
      "b'102'\n",
      "b'102' [<Element 'feedback' at 0x7f9ea4bb24f8>]\n",
      "b'103'\n",
      "b'103' [<Element 'feedback' at 0x7f9ebd87d958>]\n",
      "b'104'\n",
      "b'104' [<Element 'feedback' at 0x7f9ea4211138>]\n",
      "b'105'\n",
      "b'105' [<Element 'feedback' at 0x7f9ea4274f98>]\n",
      "b'106'\n",
      "b'106' [<Element 'feedback' at 0x7f9ea4248138>]\n",
      "b'107'\n",
      "b'107' [<Element 'feedback' at 0x7f9ebd87d958>]\n",
      "b'108'\n",
      "b'108' [<Element 'feedback' at 0x7f9ea4248048>]\n",
      "b'109'\n",
      "b'109' [<Element 'feedback' at 0x7f9ea4274f98>]\n",
      "b'110'\n",
      "b'110' [<Element 'feedback' at 0x7f9ea4248138>]\n",
      "b'111'\n",
      "b'111' [<Element 'feedback' at 0x7f9ebd87d958>]\n",
      "b'112'\n",
      "b'112' [<Element 'feedback' at 0x7f9ea4248048>]\n",
      "b'113'\n",
      "b'113' [<Element 'feedback' at 0x7f9ebcb50548>]\n",
      "b'114'\n",
      "b'114' [<Element 'feedback' at 0x7f9ea4b7dea8>]\n",
      "b'115'\n",
      "b'115' [<Element 'feedback' at 0x7f9ebd87d958>]\n",
      "b'116'\n",
      "b'116' [<Element 'feedback' at 0x7f9ea42b7d68>]\n",
      "b'117'\n",
      "b'117' [<Element 'feedback' at 0x7f9ebcb50a48>]\n",
      "b'118'\n",
      "b'118' [<Element 'feedback' at 0x7f9ea427aae8>]\n",
      "b'119'\n",
      "b'119' [<Element 'feedback' at 0x7f9ebd87d958>]\n",
      "b'120'\n",
      "b'120' [<Element 'feedback' at 0x7f9ea46e1408>]\n",
      "b'121'\n",
      "b'121' [<Element 'feedback' at 0x7f9ea46e1778>]\n",
      "b'122'\n",
      "b'122' [<Element 'feedback' at 0x7f9ea46e1598>]\n",
      "b'123'\n",
      "b'123' [<Element 'feedback' at 0x7f9ea427a8b8>]\n",
      "b'124'\n",
      "b'124' [<Element 'feedback' at 0x7f9ea46e1408>]\n",
      "b'125'\n",
      "b'125' [<Element 'feedback' at 0x7f9ea427aa48>]\n",
      "b'126'\n",
      "b'126' [<Element 'feedback' at 0x7f9ea46e1598>]\n",
      "b'127'\n",
      "b'127' [<Element 'feedback' at 0x7f9ea427a8b8>]\n",
      "b'128'\n",
      "b'128' [<Element 'feedback' at 0x7f9ea46e1408>]\n",
      "b'129'\n",
      "b'129' [<Element 'feedback' at 0x7f9ea4211548>]\n",
      "b'130'\n",
      "b'130' [<Element 'feedback' at 0x7f9ea42119f8>]\n",
      "b'131'\n",
      "b'131' [<Element 'feedback' at 0x7f9ea427a8b8>]\n",
      "b'132'\n",
      "b'132' [<Element 'feedback' at 0x7f9ebd87d958>]\n",
      "b'133'\n",
      "b'133' [<Element 'feedback' at 0x7f9ea4b7d1d8>]\n",
      "b'134'\n",
      "b'134' [<Element 'feedback' at 0x7f9ebcb50c28>]\n",
      "b'135'\n",
      "b'135' [<Element 'feedback' at 0x7f9ea427a8b8>]\n",
      "b'136'\n",
      "b'136' [<Element 'feedback' at 0x7f9ea426c9a8>]\n",
      "b'137'\n",
      "b'137' [<Element 'feedback' at 0x7f9ea4b7d1d8>]\n",
      "b'138'\n",
      "b'138' [<Element 'feedback' at 0x7f9ea42b79a8>]\n",
      "b'139'\n",
      "b'139' [<Element 'feedback' at 0x7f9ea46f26d8>]\n",
      "b'140'\n",
      "b'140' [<Element 'feedback' at 0x7f9ea426c9a8>]\n",
      "b'141'\n",
      "b'141' [<Element 'feedback' at 0x7f9ea427a8b8>]\n",
      "b'142'\n",
      "b'142' [<Element 'feedback' at 0x7f9ea46e1598>]\n",
      "b'143'\n",
      "b'143' [<Element 'feedback' at 0x7f9ea4bb24f8>]\n",
      "b'144'\n",
      "b'144' [<Element 'feedback' at 0x7f9ea426c9a8>]\n",
      "b'145'\n",
      "b'145' [<Element 'feedback' at 0x7f9ea426cdb8>]\n",
      "b'146'\n",
      "b'146' [<Element 'feedback' at 0x7f9ea4bb2458>]\n",
      "b'147'\n",
      "b'147' [<Element 'feedback' at 0x7f9ea426cd18>]\n",
      "b'148'\n",
      "b'148' [<Element 'feedback' at 0x7f9ea4bb2688>]\n",
      "b'149'\n",
      "b'149' [<Element 'feedback' at 0x7f9ea426cdb8>]\n",
      "b'150'\n",
      "b'150' [<Element 'feedback' at 0x7f9ea46e1408>]\n",
      "b'151'\n",
      "b'151' [<Element 'feedback' at 0x7f9ea4211188>]\n",
      "b'152'\n",
      "b'152' [<Element 'feedback' at 0x7f9ea4bb2688>]\n",
      "b'153'\n",
      "b'153' [<Element 'feedback' at 0x7f9ea426cdb8>]\n",
      "b'154'\n",
      "b'154' [<Element 'feedback' at 0x7f9ea4274138>]\n",
      "b'155'\n",
      "b'155' [<Element 'feedback' at 0x7f9ea422ff48>]\n",
      "b'156'\n",
      "b'156' [<Element 'feedback' at 0x7f9ea4bb2688>]\n",
      "b'157'\n",
      "b'157' [<Element 'feedback' at 0x7f9ea426cdb8>]\n",
      "b'158'\n",
      "b'158' [<Element 'feedback' at 0x7f9ea427ab88>]\n",
      "b'159'\n",
      "b'159' [<Element 'feedback' at 0x7f9ea422ff48>]\n",
      "b'160'\n",
      "b'160' [<Element 'feedback' at 0x7f9ea4237638>]\n",
      "b'161'\n",
      "b'161' [<Element 'feedback' at 0x7f9ea426cdb8>]\n",
      "b'162'\n",
      "b'162' [<Element 'feedback' at 0x7f9ea42480e8>]\n",
      "b'163'\n",
      "b'163' [<Element 'feedback' at 0x7f9ea422ff48>]\n",
      "b'164'\n",
      "b'164' [<Element 'feedback' at 0x7f9ea4237638>]\n",
      "b'165'\n",
      "b'165' [<Element 'feedback' at 0x7f9ea426cdb8>]\n",
      "b'166'\n",
      "b'166' [<Element 'feedback' at 0x7f9ea427ab88>]\n",
      "b'167'\n",
      "b'167' [<Element 'feedback' at 0x7f9ea422ff48>]\n",
      "b'168'\n",
      "b'168' [<Element 'feedback' at 0x7f9ea42b79a8>]\n",
      "b'169'\n",
      "b'169' [<Element 'feedback' at 0x7f9ea426cdb8>]\n",
      "b'170'\n",
      "b'170' [<Element 'feedback' at 0x7f9ea4b7dae8>]\n",
      "b'171'\n",
      "b'171' [<Element 'feedback' at 0x7f9ea422ff48>]\n",
      "b'172'\n",
      "b'172' [<Element 'feedback' at 0x7f9ea42b79a8>]\n",
      "b'173'\n",
      "b'173' [<Element 'feedback' at 0x7f9ea426cdb8>]\n",
      "b'174'\n",
      "b'174' [<Element 'feedback' at 0x7f9ea4b7d3b8>]\n"
     ]
    }
   ],
   "source": [
    "for n in nums[0].split():\n",
    "    print(n)\n",
    "    reports = extract_report(fetch_msg(n))\n",
    "    print(n, reports)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 45,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[<Element 'feedback' at 0x7f9ea422ff48>,\n",
       " <Element 'feedback' at 0x7f9ea4b7d048>,\n",
       " <Element 'feedback' at 0x7f9ebcb50bd8>,\n",
       " <Element 'feedback' at 0x7f9ea427aef8>,\n",
       " <Element 'feedback' at 0x7f9ea42113b8>,\n",
       " <Element 'feedback' at 0x7f9ea42376d8>,\n",
       " <Element 'feedback' at 0x7f9ea46f25e8>,\n",
       " <Element 'feedback' at 0x7f9ebcb34db8>,\n",
       " <Element 'feedback' at 0x7f9ea4246098>,\n",
       " <Element 'feedback' at 0x7f9ea4246bd8>,\n",
       " <Element 'feedback' at 0x7f9ebcb497c8>,\n",
       " <Element 'feedback' at 0x7f9ea46f2048>,\n",
       " <Element 'feedback' at 0x7f9ea41ce908>,\n",
       " <Element 'feedback' at 0x7f9ea41dc4a8>,\n",
       " <Element 'feedback' at 0x7f9ea41d0048>,\n",
       " <Element 'feedback' at 0x7f9ea41d0b38>,\n",
       " <Element 'feedback' at 0x7f9ea41c26d8>,\n",
       " <Element 'feedback' at 0x7f9ea41dc598>,\n",
       " <Element 'feedback' at 0x7f9ea41c0e08>,\n",
       " <Element 'feedback' at 0x7f9ea41c0ef8>,\n",
       " <Element 'feedback' at 0x7f9ea41be188>,\n",
       " <Element 'feedback' at 0x7f9ea41bec78>,\n",
       " <Element 'feedback' at 0x7f9ea41df818>,\n",
       " <Element 'feedback' at 0x7f9ea41e43b8>,\n",
       " <Element 'feedback' at 0x7f9ea41e4ea8>,\n",
       " <Element 'feedback' at 0x7f9ea41e8a48>,\n",
       " <Element 'feedback' at 0x7f9ea41ed688>,\n",
       " <Element 'feedback' at 0x7f9ea41f11d8>,\n",
       " <Element 'feedback' at 0x7f9ea41f1368>,\n",
       " <Element 'feedback' at 0x7f9ea41ee408>,\n",
       " <Element 'feedback' at 0x7f9ea41eef48>,\n",
       " <Element 'feedback' at 0x7f9ea41f6a98>,\n",
       " <Element 'feedback' at 0x7f9ea41f9638>,\n",
       " <Element 'feedback' at 0x7f9ea417e278>,\n",
       " <Element 'feedback' at 0x7f9ea4182f48>,\n",
       " <Element 'feedback' at 0x7f9ea4182ea8>,\n",
       " <Element 'feedback' at 0x7f9ea41869a8>,\n",
       " <Element 'feedback' at 0x7f9ea4186cc8>,\n",
       " <Element 'feedback' at 0x7f9ea418a7c8>,\n",
       " <Element 'feedback' at 0x7f9ea418b408>,\n",
       " <Element 'feedback' at 0x7f9ea418bef8>,\n",
       " <Element 'feedback' at 0x7f9ea4190a98>,\n",
       " <Element 'feedback' at 0x7f9ea4195638>,\n",
       " <Element 'feedback' at 0x7f9ea4199228>,\n",
       " <Element 'feedback' at 0x7f9ea4199d68>,\n",
       " <Element 'feedback' at 0x7f9ea419e908>,\n",
       " <Element 'feedback' at 0x7f9ea419e9f8>,\n",
       " <Element 'feedback' at 0x7f9ea41a15e8>,\n",
       " <Element 'feedback' at 0x7f9ea41a2188>,\n",
       " <Element 'feedback' at 0x7f9ea41a2d18>,\n",
       " <Element 'feedback' at 0x7f9ea41a5908>,\n",
       " <Element 'feedback' at 0x7f9ea41a94a8>,\n",
       " <Element 'feedback' at 0x7f9ea41ad048>,\n",
       " <Element 'feedback' at 0x7f9ea41adb88>,\n",
       " <Element 'feedback' at 0x7f9ea41adcc8>,\n",
       " <Element 'feedback' at 0x7f9ea41b17c8>,\n",
       " <Element 'feedback' at 0x7f9ea41b4ef8>,\n",
       " <Element 'feedback' at 0x7f9ea41b1958>,\n",
       " <Element 'feedback' at 0x7f9ea41b8908>,\n",
       " <Element 'feedback' at 0x7f9ea41ba4f8>,\n",
       " <Element 'feedback' at 0x7f9ea413d098>,\n",
       " <Element 'feedback' at 0x7f9ea4142138>,\n",
       " <Element 'feedback' at 0x7f9ea4142c78>,\n",
       " <Element 'feedback' at 0x7f9ea4146868>,\n",
       " <Element 'feedback' at 0x7f9ea414b458>,\n",
       " <Element 'feedback' at 0x7f9ea41ba4a8>,\n",
       " <Element 'feedback' at 0x7f9ea414bbd8>,\n",
       " <Element 'feedback' at 0x7f9ebcb43778>,\n",
       " <Element 'feedback' at 0x7f9ea4b93368>,\n",
       " <Element 'feedback' at 0x7f9ea4b93ea8>,\n",
       " <Element 'feedback' at 0x7f9ea414fa98>,\n",
       " <Element 'feedback' at 0x7f9ea4153688>,\n",
       " <Element 'feedback' at 0x7f9ea4159188>,\n",
       " <Element 'feedback' at 0x7f9ea4159cc8>,\n",
       " <Element 'feedback' at 0x7f9ea4159db8>,\n",
       " <Element 'feedback' at 0x7f9ea415fea8>,\n",
       " <Element 'feedback' at 0x7f9ea415da48>,\n",
       " <Element 'feedback' at 0x7f9ea4161638>,\n",
       " <Element 'feedback' at 0x7f9ea4165728>,\n",
       " <Element 'feedback' at 0x7f9ea416a2c8>,\n",
       " <Element 'feedback' at 0x7f9ea416e188>,\n",
       " <Element 'feedback' at 0x7f9ea416ef48>,\n",
       " <Element 'feedback' at 0x7f9ea416e048>,\n",
       " <Element 'feedback' at 0x7f9ea4173cc8>,\n",
       " <Element 'feedback' at 0x7f9ea416de08>,\n",
       " <Element 'feedback' at 0x7f9ea4176958>,\n",
       " <Element 'feedback' at 0x7f9ea417a4f8>,\n",
       " <Element 'feedback' at 0x7f9ea40fe098>,\n",
       " <Element 'feedback' at 0x7f9ea40feb88>,\n",
       " <Element 'feedback' at 0x7f9ea4104c78>,\n",
       " <Element 'feedback' at 0x7f9ea4108868>,\n",
       " <Element 'feedback' at 0x7f9ea40feb38>,\n",
       " <Element 'feedback' at 0x7f9ea410fa98>,\n",
       " <Element 'feedback' at 0x7f9ea410b688>,\n",
       " <Element 'feedback' at 0x7f9ea41121d8>,\n",
       " <Element 'feedback' at 0x7f9ea4112d68>,\n",
       " <Element 'feedback' at 0x7f9ea41158b8>,\n",
       " <Element 'feedback' at 0x7f9ea411b458>,\n",
       " <Element 'feedback' at 0x7f9ea411e048>,\n",
       " <Element 'feedback' at 0x7f9ea411eb88>,\n",
       " <Element 'feedback' at 0x7f9ea411b408>,\n",
       " <Element 'feedback' at 0x7f9ea4123818>,\n",
       " <Element 'feedback' at 0x7f9ea411f958>,\n",
       " <Element 'feedback' at 0x7f9ea41264a8>,\n",
       " <Element 'feedback' at 0x7f9ea412a098>,\n",
       " <Element 'feedback' at 0x7f9ea412abd8>,\n",
       " <Element 'feedback' at 0x7f9ea412f728>,\n",
       " <Element 'feedback' at 0x7f9ea41352c8>,\n",
       " <Element 'feedback' at 0x7f9ea4135e58>,\n",
       " <Element 'feedback' at 0x7f9ea413aa48>,\n",
       " <Element 'feedback' at 0x7f9ea41159a8>,\n",
       " <Element 'feedback' at 0x7f9ea4132638>,\n",
       " <Element 'feedback' at 0x7f9ea40bf228>,\n",
       " <Element 'feedback' at 0x7f9ea40c1318>,\n",
       " <Element 'feedback' at 0x7f9ea40c1e58>,\n",
       " <Element 'feedback' at 0x7f9ea40c7a48>,\n",
       " <Element 'feedback' at 0x7f9ea40cb5e8>,\n",
       " <Element 'feedback' at 0x7f9ea40d0188>,\n",
       " <Element 'feedback' at 0x7f9ea40d0c78>,\n",
       " <Element 'feedback' at 0x7f9ea40d0e08>,\n",
       " <Element 'feedback' at 0x7f9ea40d6958>,\n",
       " <Element 'feedback' at 0x7f9ea40c84f8>,\n",
       " <Element 'feedback' at 0x7f9ea40cd098>,\n",
       " <Element 'feedback' at 0x7f9ea40dc228>,\n",
       " <Element 'feedback' at 0x7f9ea40dcd18>,\n",
       " <Element 'feedback' at 0x7f9ea40e08b8>,\n",
       " <Element 'feedback' at 0x7f9ea40e7408>,\n",
       " <Element 'feedback' at 0x7f9ea40eb4a8>,\n",
       " <Element 'feedback' at 0x7f9ea40eb778>,\n",
       " <Element 'feedback' at 0x7f9ea40e5868>,\n",
       " <Element 'feedback' at 0x7f9ea40f1458>,\n",
       " <Element 'feedback' at 0x7f9ea40f1f98>,\n",
       " <Element 'feedback' at 0x7f9ea40f5bd8>,\n",
       " <Element 'feedback' at 0x7f9ea40f9728>,\n",
       " <Element 'feedback' at 0x7f9ea407e818>,\n",
       " <Element 'feedback' at 0x7f9ea4081458>,\n",
       " <Element 'feedback' at 0x7f9ea40f5228>,\n",
       " <Element 'feedback' at 0x7f9ea40830e8>,\n",
       " <Element 'feedback' at 0x7f9ea4083cc8>,\n",
       " <Element 'feedback' at 0x7f9ea4085e08>,\n",
       " <Element 'feedback' at 0x7f9ea4089ef8>,\n",
       " <Element 'feedback' at 0x7f9ea408eb38>,\n",
       " <Element 'feedback' at 0x7f9ea4092688>,\n",
       " <Element 'feedback' at 0x7f9ea40967c8>,\n",
       " <Element 'feedback' at 0x7f9ea4096868>,\n",
       " <Element 'feedback' at 0x7f9ea409c958>,\n",
       " <Element 'feedback' at 0x7f9ea4099548>,\n",
       " <Element 'feedback' at 0x7f9ea409e688>,\n",
       " <Element 'feedback' at 0x7f9ea40a2188>,\n",
       " <Element 'feedback' at 0x7f9ea40a2cc8>,\n",
       " <Element 'feedback' at 0x7f9ea40a8818>,\n",
       " <Element 'feedback' at 0x7f9ea40ad3b8>,\n",
       " <Element 'feedback' at 0x7f9ea40adef8>,\n",
       " <Element 'feedback' at 0x7f9ea40b2ae8>,\n",
       " <Element 'feedback' at 0x7f9ea40b2bd8>,\n",
       " <Element 'feedback' at 0x7f9ea40a77c8>,\n",
       " <Element 'feedback' at 0x7f9ea40b6368>,\n",
       " <Element 'feedback' at 0x7f9ea40b94f8>,\n",
       " <Element 'feedback' at 0x7f9ea403e048>,\n",
       " <Element 'feedback' at 0x7f9ea403eb88>,\n",
       " <Element 'feedback' at 0x7f9ea4044c78>,\n",
       " <Element 'feedback' at 0x7f9ea40b9f98>,\n",
       " <Element 'feedback' at 0x7f9ea40bb138>,\n",
       " <Element 'feedback' at 0x7f9ea40bbc28>,\n",
       " <Element 'feedback' at 0x7f9ea404e818>,\n",
       " <Element 'feedback' at 0x7f9ea4053458>,\n",
       " <Element 'feedback' at 0x7f9ea40594f8>,\n",
       " <Element 'feedback' at 0x7f9ea405e098>,\n",
       " <Element 'feedback' at 0x7f9ea405eb88>,\n",
       " <Element 'feedback' at 0x7f9ea405ec78>,\n",
       " <Element 'feedback' at 0x7f9ea40634f8>,\n",
       " <Element 'feedback' at 0x7f9ea4137598>,\n",
       " <Element 'feedback' at 0x7f9ea40511d8>,\n",
       " <Element 'feedback' at 0x7f9ea4051d18>]"
      ]
     },
     "execution_count": 45,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "dmarc_reports = [report for report_set in [extract_report(fetch_msg(n)) for n in nums[0].split()]\n",
    "                for report in report_set]\n",
    "dmarc_reports"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 46,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "('BYE', [b'Logging out'])"
      ]
     },
     "execution_count": 46,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "mailbox.close()\n",
    "mailbox.logout()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 47,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "field_maps = {'./policy_published/adkim': {'pg_field_name': 'policy_published_adkim',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'varchar'},\n",
    " './policy_published/aspf': {'pg_field_name': 'policy_published_aspf',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'varchar'},\n",
    " './policy_published/domain': {'pg_field_name': 'policy_published_domain',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'varchar'},\n",
    " './policy_published/p': {'pg_field_name': 'policy_published_p',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'varchar'},\n",
    " './policy_published/pct': {'pg_field_name': 'policy_published_pct',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'int'},\n",
    " './record[{}]/auth_results/dkim/domain': {'pg_field_name': 'auth_results_dkim_domain',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'varchar'},\n",
    " './record[{}]/auth_results/dkim/result': {'pg_field_name': 'auth_results_dkim_result',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'varchar'},\n",
    " './record[{}]/auth_results/spf/domain': {'pg_field_name': 'auth_results_spf_domain',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'varchar'},\n",
    " './record[{}]/auth_results/spf/result': {'pg_field_name': 'auth_results_spf_result',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'varchar'},\n",
    " './record[{}]/identifiers/header_from': {'pg_field_name': 'identifiers_header_from',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'varchar'},\n",
    " './record[{}]/row/count': {'pg_field_name': 'count',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'int'},\n",
    " './record[{}]/row/policy_evaluated/disposition': {'pg_field_name': 'policy_evaluated_disposition',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'varchar'},\n",
    " './record[{}]/row/policy_evaluated/dkim': {'pg_field_name': 'policy_evaluated_dkim',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'varchar'},\n",
    " './record[{}]/row/policy_evaluated/spf': {'pg_field_name': 'policy_evaluated_spf',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'varchar'},\n",
    " './record[{}]/row/source_ip': {'pg_field_name': 'source_ip',\n",
    "  'pg_table': 'report_items',\n",
    "  'pg_type': 'inet'},\n",
    " './report_metadata/date_range/begin': {'pg_field_name': 'report_metadata_date_range_begin',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'timestamp'},\n",
    " './report_metadata/date_range/end': {'pg_field_name': 'report_metadata_date_range_end',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'timestamp'},\n",
    " './report_metadata/email': {'pg_field_name': 'report_metadata_email',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'varchar'},\n",
    " './report_metadata/org_name': {'pg_field_name': 'report_metadata_org_name',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'varchar'},\n",
    " './report_metadata/report_id': {'pg_field_name': 'report_metadata_report_id',\n",
    "  'pg_table': 'reports',\n",
    "  'pg_type': 'varchar'}}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 48,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def maybe_strip(text):\n",
    "    if text:\n",
    "        return text.strip()\n",
    "    else:\n",
    "        return ''"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 49,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def write_report(connection, cursor, report):\n",
    "    \n",
    "    field_names = []\n",
    "    values = {}\n",
    "    for f in [f for f in field_maps if field_maps[f]['pg_table'] == 'reports']:\n",
    "        field_names += [field_maps[f]['pg_field_name']]\n",
    "        if field_maps[f]['pg_type'] == 'int':\n",
    "            values[field_maps[f]['pg_field_name']] = int(report.find(f).text)\n",
    "        elif field_maps[f]['pg_type'] == 'timestamp':\n",
    "            values[field_maps[f]['pg_field_name']] = datetime.datetime.utcfromtimestamp(int(report.find(f).text))\n",
    "        elif field_maps[f]['pg_type'] == 'inet':\n",
    "            values[field_maps[f]['pg_field_name']] = maybe_strip(report.find(f).text)\n",
    "        else:\n",
    "            values[field_maps[f]['pg_field_name']] = maybe_strip(report.find(f).text)\n",
    "    insert_string = 'insert into reports (' + ', '.join(field_names) + ') '\n",
    "    insert_string += 'values (' + ', '.join('%({})s'.format(f) for f in field_names) + ');'\n",
    "    cursor.execute(insert_string, values)\n",
    "    \n",
    "    for i in range(1, len(report.findall('./record'))+1):\n",
    "        field_names = []\n",
    "        cursor.execute('select id, report_metadata_report_id from reports where report_metadata_report_id = %s;', \n",
    "            [report.find('./report_metadata/report_id').text])\n",
    "        results = cursor.fetchall()\n",
    "        if len(results) != 1:\n",
    "            raise\n",
    "        else:\n",
    "            report_id = results[0][0]\n",
    "        values = {'report_id': report_id}\n",
    "        for f in [f for f in field_maps if field_maps[f]['pg_table'] == 'report_items']:\n",
    "            field_names += [field_maps[f]['pg_field_name']]\n",
    "            if field_maps[f]['pg_type'] == 'int':\n",
    "                values[field_maps[f]['pg_field_name']] = int(report.find(f.format(i)).text)\n",
    "            elif field_maps[f]['pg_type'] == 'timestamp':\n",
    "                values[field_maps[f]['pg_field_name']] = datetime.datetime.utcfromtimestamp(int(report.find(f.format(i)).text))\n",
    "            elif field_maps[f]['pg_type'] == 'inet':\n",
    "                values[field_maps[f]['pg_field_name']] = maybe_strip(report.find(f.format(i)).text)\n",
    "            else:\n",
    "                values[field_maps[f]['pg_field_name']] = maybe_strip(report.find(f.format(i)).text)\n",
    "        insert_string = 'insert into report_items (report_id, ' + ', '.join(field_names) + ') '\n",
    "        insert_string += 'values (%(report_id)s, ' + ', '.join('%({})s'.format(f) for f in field_names) + ');'\n",
    "        cursor.execute(insert_string, values)\n",
    "    connection.commit()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 51,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "conn = psycopg2.connect(host=config['database']['server'],\n",
    "                        database=config['database']['database'], \n",
    "                        user=config['database']['username'], \n",
    "                        password=config['database']['password'])\n",
    "with conn.cursor() as cur:\n",
    "    for report in dmarc_reports:\n",
    "        write_report(conn, cur, report)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 52,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "conn.close()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.4.3+"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}