def delete_post(self, post_id):
from app.posts.models import Post
try:
post = Post.get(Post.id == post_id)
if self.owns_post(post):
post.delete_instance()
return "Done"
return "You do not have correct permissions."
except:
return "Error"
def edit_post(self, post_id, content):
from app.posts.models import Post
try:
post = Post.get(Post.id == post_id)
if self.owns_post(post):
post.content = content
post.save()
return "Done"
return "Do not have correct permissions."
except:
return "Error"
def comment(self, post_id, comment_text):
from app.posts.models import Post, Comment
post = Post.get(Post.id == post_id)
user = User.get(User.id == post.user)
ids = [friend.id for friend in user.get_friends()]
ids.append(user.id)
print(ids)
print(comment_text)
try:
if self.id in ids:
Comment.create(
user=self,
post=post,
content=comment_text
)
return "Commented."
return "You do not have correct permissions"
except:
return "Error commenting."
