def post(self, request, *args, **kwargs):
form = MessageForm( request.POST )
if form.is_valid():
data = form.cleaned_data
users_t = data["users"].split(",")
message = Message(
author = request.user,
text = data["text"],
title = data["title"],
date = datetime.now()
)
message.save()
message.users.add( request.user )
for user in users_t:
user = user.strip()
try:
if user != request.user.username:
message.users.add( User.objects.get( username = user ) )
except User.DoesNotExist as e:
print user + " does not exist."
message.save()
return HttpResponseRedirect("/mailbox/list.html" )
else:
data = {
"MEDIA_URL" : MEDIA_URL,
"user" : request.user,
"form" : form
}
data.update( csrf( request ) )
return render_to_response( self.template_name, data )
def home(request):
form = MessageForm(request.POST or None)
if request.method == "POST" and form.is_valid():
temp = form.save(commit=False)
parent = form['parent'].value()
if parent == '':
# Set a blank path then save it to get an ID
temp.path = []
temp.save()
temp.path = [temp.id]
else:
# Get the parent node
node = Message.objects.get(id=parent)
temp.depth = node.depth + 1
temp.path = node.path
# Store parents path then apply comment ID
temp.save()
temp.path.append(temp.id)
# Final save for parents and children
temp.save()
# Retrieve all comments and sort them by path
message_tree = Message.objects.all().order_by('path')
return render(request, "core/messages.html", locals())
def index(request):
post = EssayPost.objects.latest("time_stamp")
form = MessageForm()
if request.method == "POST":
form = MessageForm(request.POST)
if form.is_valid():
form.save()
context = {"post": post, "form": form}
return render(request, "index.html", context)
def send_message(recipient):
user = User.query.filter_by(username=recipient).first_or_404()
form = MessageForm()
if form.validate_on_submit():
msg = Message(author=current_user,
recipient=user,
body=form.message.data)
db.session.add(msg)
user.add_notification('unread_message_count', user.new_messages())
db.session.commit()
flash(_('Your message has been sent.'))
return redirect(url_for('user', username=recipient))
return render_template('send_message.html',
title=_('Send Message'),
form=form,
recipient=recipient)
def index(request):
post = EssayPost.objects.latest('time_stamp')
form = MessageForm()
if request.method == 'POST':
form = MessageForm(request.POST)
if form.is_valid():
form.save()
context = {'post': post, 'form': form}
return render(request, 'index.html', context)
