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: