def new_message():
fromAddress = request.form['fromAddress']
ccAddress = request.form['ccAddress']
subject = request.form['subject']
body = request.form['body']
reportsEnabled = True if request.form.get('reportsEnabled') else False
reportAddress = request.form['reportAddress']
# Parse .txt file of from addresses if uploaded
fromAddrList = request.files.get('fromAddressList')
if fromAddrList.stream.getvalue():
fromAddrList = fromAddrList.stream.getvalue().split('\n')
if '' in fromAddrList: fromAddrList.remove('')
fromAddress = ', '.join(fromAddrList)
try: # Save message in db
message_id = models.connection.acw.messages.insert({'created_at': datetime.datetime.now(),
'fromAddress': fromAddress, 'ccAddress': ccAddress,
'subject': subject, 'body': body, 'reportsEnabled': reportsEnabled,
'reportAddress': reportAddress, 'user': session['user'] })
# Save uploads in message attachments folder
basePath = os.path.dirname(os.path.realpath(__file__))
messageAttachmentsFolder = '{}/uploads/{}/attachments/'.format(basePath, str(message_id))
helpers.make_dir(messageAttachmentsFolder)
for attachment in request.files.getlist('attachments'):
savePath = '{}{}'.format(messageAttachmentsFolder, attachment.filename)
attachment.save(savePath)
except Exception, e:
print e
def edit_message(_id=None):
if request.method != 'POST':
messages = list(models.connection.acw.messages.find({'user':session['user']}))
message = models.connection.acw.messages.find_one({u'_id': ObjectId(_id)})
return render_template('edit_message.html', messages=messages, message=message, messageCount=len(messages), datetime=datetime.datetime.now())
else:
fromAddress = request.form['fromAddress']
ccAddress = request.form['ccAddress']
subject = request.form['subject']
body = request.form['body']
reportsEnabled = True if request.form.get('reportsEnabled') else False
reportAddress = request.form['reportAddress']
_id = request.form['_id']
# Parse .txt file of from addresses if uploaded
fromAddrList = request.files.get('fromAddressList')
if fromAddrList.stream.getvalue():
fromAddrList = fromAddrList.stream.getvalue().split('\n')
if '' in fromAddrList: fromAddrList.remove('')
fromAddress = ', '.join(fromAddrList)
try: # Save message in db
db = models.connection.acw.messages
db.update({'_id': ObjectId(_id)}, {'$set': {'fromAddress': fromAddress, 'ccAddress': ccAddress,
'subject': subject, 'body': body, 'reportsEnabled': reportsEnabled, 'reportAddress': reportAddress }})
# Save uploads in message attachments folder
basePath = os.path.dirname(os.path.realpath(__file__))
messageAttachmentsFolder = '{}/uploads/{}/attachments/'.format(basePath, message['_id'])
helpers.make_dir(messageAttachmentsFolder)
for attachment in request.files.getlist('attachments'):
savePath = '{}{}'.format(messageAttachmentsFolder, attachment.filename)
attachment.save(savePath)
except Exception, e:
print e
finally:
