30 lines
955 B
Python
30 lines
955 B
Python
|
import datetime
|
||
|
|
from szurubooru import db, errors
|
||
|
|
from szurubooru.func import users, posts
|
||
|
|
|
||
|
|
class CommentNotFoundError(errors.NotFoundError): pass
|
||
|
|
class EmptyCommentTextError(errors.ValidationError): pass
|
||
|
|
|
||
|
|
def serialize_comment(comment, authenticated_user):
|
||
|
|
return {
|
||
|
|
'id': comment.comment_id,
|
||
|
|
'user': users.serialize_user(comment.user, authenticated_user),
|
||
|
|
'post': posts.serialize_post(comment.post, authenticated_user),
|
||
|
|
'text': comment.text,
|
||
|
|
'creationTime': comment.creation_time,
|
||
|
|
'lastEditTime': comment.last_edit_time,
|
||
|
|
}
|
||
|
|
|
||
|
|
def create_comment(user, post, text):
|
||
|
|
comment = db.Comment()
|
||
|
|
comment.user = user
|
||
|
|
comment.post = post
|
||
|
|
update_comment_text(comment, text)
|
||
|
|
comment.creation_time = datetime.datetime.now()
|
||
|
|
return comment
|
||
|
|
|
||
|
|
def update_comment_text(comment, text):
|
||
|
|
if not text:
|
||
|
|
raise EmptyCommentTextError('Comment text cannot be empty.')
|
||
|
|
comment.text = text
|