Wammu/MailWriter.py

   1 # -*- coding: UTF-8 -*-
   2 # vim: expandtab sw=4 ts=4 sts=4:
   3 '''
   4 Wammu - Phone manager
   5 Module for writing SMS to Email.
   6 '''
   7 __author__ = 'Michal Čihař'
   8 __email__ = 'michal@cihar.com'
   9 __license__ = '''
  10 Copyright © 2003 - 2008 Michal Čihař
  11
  12 This program is free software; you can redistribute it and/or modify it
  13 under the terms of the GNU General Public License version 2 as published by
  14 the Free Software Foundation.
  15
  16 This program is distributed in the hope that it will be useful, but WITHOUT
  17 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  18 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  19 more details.
  20
  21 You should have received a copy of the GNU General Public License along with
  22 this program; if not, write to the Free Software Foundation, Inc.,
  23 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
  24 '''
  25
  26 from Wammu.Utils import SearchNumber
  27 from Wammu.MessageDisplay import SmsTextFormat
  28 from email.MIMEAudio import MIMEAudio
  29 from email.MIMEImage import MIMEImage
  30 from email.MIMEText import MIMEText
  31 from email.MIMEMultipart import MIMEMultipart
  32 import email.Utils
  33 import md5
  34 import time
  35 import tempfile
  36 import os
  37 import wx
  38 import Wammu.Data
  39 if Wammu.gammu_error == None:
  40     import gammu
  41
  42 HEADER_FORMAT = 'X-Wammu-%s'
  43 CID_FORMAT = '%d*sms@wammu.sms'
  44 PARTS_TO_HEADER = [
  45         'Folder',
  46         'Memory',
  47         'Location',
  48         'Name',
  49         'Type',
  50         'State',
  51         'Class',
  52         'MessageReference',
  53         ]
  54
  55
  56 def XPMToPNG(image):
  57     '''
  58     Convert XPM data to PNG image.
  59
  60     @todo: I'd like to avoid creating temporary file, but even PIL doesn't
  61     seem to provide such functionality.
  62     '''
  63     handle, name = tempfile.mkstemp()
  64     os.close(handle)
  65     bitmap = wx.BitmapFromXPMData(image)
  66     bitmap.SaveFile(name, wx.BITMAP_TYPE_PNG)
  67     f = file(name)
  68     data = f.read()
  69     f.close()
  70     os.unlink(name)
  71     return data
  72
  73 def RingtoneToMIDI(data):
  74     '''
  75     Converts ringtone to MIDI data.
  76
  77     @todo: I'd like to avoid creating temporary file, but Gammmu doesn't
  78     provide such functionality.
  79     '''
  80     handle, name = tempfile.mkstemp()
  81     f = os.fdopen(handle, 'w+')
  82     gammu.SaveRingtone(f, data, 'mid')
  83     f.seek(0)
  84     data = f.read()
  85     f.close()
  86     os.unlink(name)
  87     return data
  88
  89 def DateToString(date):
  90     '''
  91     Convert date to RFC 2822 format.
  92     '''
  93     return email.Utils.formatdate(time.mktime(date.timetuple()), True)
  94
  95 def SMSToMail(cfg, sms, lookuplist = None, mailbox = False):
  96     '''
  97     Converts SMS to formated mail. It will contain all images and sounds from
  98     message and predefined animations. Also text formatting is preserved in
  99     HTML.
 100     '''
 101     msg = MIMEMultipart('related', None, None, type='text/html')
 102     prepend = ''
 103     name = ''
 104     if lookuplist != None:
 105         i = SearchNumber(lookuplist, sms['Number'])
 106         if i != -1:
 107             msg.add_header(HEADER_FORMAT % 'ContactID', str(i))
 108             name = '%s ' % lookuplist[i]['Name']
 109
 110     for header in PARTS_TO_HEADER:
 111         msg.add_header(HEADER_FORMAT % header, unicode(sms['SMS'][0][header]))
 112     msg.add_header(HEADER_FORMAT % 'SMSC', sms['SMS'][0]['SMSC']['Number'])
 113     if sms['SMS'][0]['SMSCDateTime'] is not None:
 114         msg.add_header(HEADER_FORMAT % 'SMSCDate',
 115                 DateToString(sms['SMS'][0]['SMSCDateTime']))
 116
 117     remote = '%s<%s@wammu.sms>' % (name, sms['Number'])
 118     local = cfg.Read('/MesageExport/From')
 119     if sms['SMS'][0]['Type'] == 'Submit':
 120         msg['To'] = remote
 121         msg['From'] = local
 122     else:
 123         msg['To'] = local
 124         msg['From'] = remote
 125         prepend = ('Received: from %s via GSM\n' %
 126                 unicode(sms['SMS'][0]['SMSC']['Number'])) + prepend
 127
 128     if len(sms['Name']) > 0 :
 129         msg['Subject'] = SmsTextFormat(cfg, sms['Name'], False)
 130     else:
 131         msg['Subject'] = SmsTextFormat(cfg, sms['Text'], False)[:50] + '...'
 132
 133     if sms['DateTime'] is not None:
 134         msg['Date'] = DateToString(sms['DateTime'])
 135
 136     if sms.has_key('SMSInfo'):
 137         text = ''
 138         cid = 0
 139         for i in sms['SMSInfo']['Entries']:
 140             if i['ID'] in Wammu.Data.SMSIDs['PredefinedAnimation']:
 141                 if i['Number'] > len(Wammu.Data.PredefinedAnimations):
 142                     sub = MIMEImage(XPMToPNG(Wammu.Data.UnknownPredefined))
 143                 else:
 144                     img = Wammu.Data.PredefinedAnimations[i['Number']][1]
 145                     xpm = XPMToPNG(img)
 146                     sub = MIMEImage(xpm)
 147                 sub.add_header('Content-ID', '<%s>' % CID_FORMAT % cid)
 148                 sub.add_header('Content-Disposition', 'inline')
 149                 msg.attach(sub)
 150                 text = text + '<img src="cid:%s">' % (CID_FORMAT % cid)
 151                 cid = cid + 1
 152
 153             # FIXME: need sounds
 154             if 0 and i['ID'] in Wammu.Data.SMSIDs['PredefinedSound']:
 155                 if i['Number'] >= len(Wammu.Data.PredefinedSounds):
 156                     sub = ''
 157                 else:
 158                     sub = ''
 159                 sub.add_header('Content-Disposition', 'attachment')
 160                 msg.attach(sub)
 161
 162             if i['ID'] in Wammu.Data.SMSIDs['Sound']:
 163                 sub = MIMEAudio(RingtoneToMIDI(i['Ringtone']), 'midi')
 164                 sub.add_header('Content-Disposition', 'attachment')
 165                 msg.attach(sub)
 166
 167             if i['ID'] in Wammu.Data.SMSIDs['Text']:
 168                 fmt = '%s'
 169                 for format_data in Wammu.Data.TextFormats:
 170                     for name, dummy, style in format_data[1:]:
 171                         if i.has_key(name) and i[name]:
 172                             fmt = style % fmt
 173                 text = text + (fmt % SmsTextFormat(cfg, i['Buffer']))
 174
 175             if i['ID'] in Wammu.Data.SMSIDs['Bitmap']:
 176                 for bitmap in i['Bitmap']:
 177                     sub = MIMEImage(XPMToPNG(bitmap['XPM']))
 178                     sub.add_header('Content-ID', '<%s>' % CID_FORMAT % cid)
 179                     sub.add_header('Content-Disposition', 'inline')
 180                     msg.attach(sub)
 181                     text = text + '<img src="cid:%s">' % (CID_FORMAT % cid)
 182                     cid = cid + 1
 183
 184             if i['ID'] in Wammu.Data.SMSIDs['Animation']:
 185                 for bitmap in i['Bitmap']:
 186                     sub = MIMEImage(XPMToPNG(bitmap['XPM']))
 187                     sub.add_header('Content-ID', '<%s>' % CID_FORMAT % cid)
 188                     sub.add_header('Content-Disposition', 'inline')
 189                     msg.attach(sub)
 190                     text = text + '<img src="cid:%s">' % (CID_FORMAT % cid)
 191                     cid = cid + 1
 192
 193     else:
 194         text = SmsTextFormat(cfg, sms['Text'])
 195
 196     html = '<html><head></head><body>%s</body></html>'
 197     sub = MIMEText(html % text.encode('utf-8'), 'html', 'utf-8')
 198     msg.attach(sub)
 199
 200     if sms['DateTime'] is not None:
 201         filename = '%s-%s-%s.eml' % (
 202                 sms['SMS'][0]['Type'],
 203                 sms['DateTime'].strftime("%Y%m%d%H%M%S"),
 204                 md5.new(sms['Text'].encode('utf-8')).hexdigest())
 205     else:
 206         filename = '%s-%s.eml' % (
 207                 sms['SMS'][0]['Type'],
 208                 md5.new(sms['Text'].encode('utf-8')).hexdigest())
 209
 210     # Add message ID
 211     msg.add_header('Message-ID', '<%s@%s>' % (filename[:-4], sms['Number']))
 212
 213     if mailbox:
 214         if sms['DateTime'] is None:
 215             timestamp = time.asctime(time.localtime(None))
 216         else:
 217             timestamp = time.asctime(sms['DateTime'].timetuple())
 218         prepend = ('From wammu@wammu.sms %s\n' % timestamp) + prepend
 219
 220     return filename, prepend + msg.as_string()
 221