chat_liza_me.py 1.3 KB

1234567891011121314151617181920212223242526272829303132
  1. from flask import Blueprint, render_template, request, redirect, abort
  2. from ..models.massage import Massage
  3. from ..extensions import db
  4. from flask_login import current_user
  5. liza_me = Blueprint('liza/liza_me', __name__)
  6. @liza_me.route('/liza/chat_liza_me', methods=['POST', 'GET'])
  7. def chat_liza_me():
  8. if current_user.is_authenticated:
  9. if current_user.name in ['me', 'liza']:
  10. if request.method == "POST":
  11. mess = request.form['mess']
  12. if len(mess)>0:
  13. message = Massage( massage = mess, author_of_massage = 'liza')
  14. try:
  15. db.session.add(message)
  16. db.session.commit()
  17. return redirect('/liza/chat_liza_me')
  18. except Exception as e:
  19. print(str(e))
  20. else:
  21. messages = Massage.query.all()
  22. last_message = messages[-1].massage
  23. return render_template('chats/lizame.html', messages = messages, last_message=last_message)
  24. else:
  25. messages = Massage.query.all()
  26. last_message = messages[-1].massage
  27. return render_template('chats/lizame.html', messages = messages, last_message=last_message)
  28. else: abort(403)