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
|
#!/usr/bin/env python
# mbot - a mail handling robot
#
# Author: Dimitri Fontaine <dim@tapoueh.org>
# Author: Christophe Truffier <toffe@nah-ko.org>
#
# This code is licensed under the GPL.
# Get yourself a version here : http://www.gnu.org/copyleft/gpl.html
# $Id: NewsHandler.py,v 1.33 2003/12/22 15:05:19 nah-ko Exp $
import MailHandler
import sys, os, email, re, Image
import ConfigParser
from MailHandler import MailHandler
# This class does not provides SGBD specific code, see MyNewsHandler
# and PgNewsHandler
class NewsHandler(MailHandler):
""" Manage adding news in a data base """
def read_conf(self, ConfObj):
''' Getting config options for this handler '''
self.log.notice("[NewsHandler]: read_conf")
MailHandler.read_conf(self, ConfObj,
['host', 'db', 'db_user', 'db_pass',
'photo_tbl', 'photo_tblsq',
'news_tbl', 'news_tblsq',
'site', 'attach_path', 'tnx', 'tny'])
def getsite(self, e_mail):
""" Get the site information """
import rfc822
self.log.notice("[NewsHandler]: getsite")
(name, sender) = rfc822.parseaddr(e_mail)
self.log.debug("[NewsHandler]: getsite -> sender='%s'" % sender)
(user, dom) = sender.split('@')
self.log.debug("[NewsHandler]: getsite -> dom='%s'" % dom)
try:
dico_site = eval(self.site)
site = dico_site[dom]
except KeyError, SyntaxError:
site = 'test'
self.log.debug("[NewsHandler]: getsite -> site='%s'" % site)
return site
def add_news(self, text, img_id=0):
self.log.notice("[NewsHandler]: add_news")
date = self.date
sender = self.sender
subject = re.escape(self.params)
SITE = self.getsite(self.dest)
if img_id == 0:
myquery = """
INSERT INTO %s (site,date,de,sujet,message)
VALUES ('%s','%s','%s','%s','%s')
""" % (self.news_tbl, SITE, date, sender, subject, re.escape(text))
else:
myquery = """
INSERT INTO %s (site,date,de,sujet,message,id_img)
VALUES ('%s','%s','%s','%s','%s','%d')
""" % (self.news_tbl, SITE,
date, sender, subject, re.escape(text), img_id)
self.log.debug("[NewsHandler]: myquery = %s" % myquery)
self.id = self.execQuery(myquery)
self.log.debug("[NewsHandler]: add_news -> id='%d'" % self.id)
return self.id
def handle(self, body):
"Get news from text and attachment if present"
self.log.notice("[NewsHandler]")
result = "Automatic response to: " + self.params + "\n"
id_img = id_news = 0
# Handle text part of a news
self.log.notice("[NewsHandler]: Use [%s] section" % self.section)
if type(body) == type(""):
id_news = self.add_news(body)
result = result + " #%d added" % id_news
self.log.debug("[NewsHandler]: result='%s' for '%s'" \
% (result, self.section))
# Handle image part of a news
else:
# This part is not image but html
if type(body) == type(""):
self.log.debug("[NewsHandler]: text part")
if id_img != 0:
id_news = self.add_news(body, id_img)
else:
id_news = self.add_news(body)
# This part is not a text one
else:
maintype, subtype = body.get_content_type().split('/',1)
self.log.debug("[NewsHandler]: maintype='%s'" \
% maintype + \
" subtype='%s'" \
% subtype)
# We insert an image in the data base
if maintype == "image":
self.log.debug("[NewsHandler]: image part")
file = self.attach_path + body.get_filename()
TNfile = self.attach_path + "TN_" + body.get_filename()
# We first save the image
f = open(file, "w")
f.write(body.get_payload(decode=1))
f.close()
self.log.debug("[NewsHandler]: image saved to '%s'" \
% self.attach_path)
# Then a thumbnail
img = Image.open(file)
img.thumbnail((int(self.tnX), int(self.tnY)))
img.save(TNfile, img.format)
f = open(TNfile, "r")
TNdata = f.read()
f.close()
self.log.debug("[NewsHandler]: thumbnail created to '%s'" \
% self.attach_path)
filesize = os.stat(file).st_size
# Now we add the images in the data base
id_img = self.add_img(body.get_filename(),
body.get_content_type(),
body.get_payload(decode=1),
TNdata, filesize)
self.log.debug("[NewsHandler]: image #%d " % id_img + \
"added to DB '%s'" % self.db)
# And we clean the temporary created files
os.remove(file)
os.remove(TNfile)
self.log.debug("[NewsHandler]: '%s' and '%s' " \
% (file, TNfile) + \
"removed")
result = result + "Image #%d added" % id_img
return [('text/plain', result)]
|
