Adds delete, edit, and hide operations
This commit is contained in:
parent
8138e71668
commit
a22b4a9162
1 changed files with 47 additions and 11 deletions
|
@ -203,6 +203,7 @@ def create_comment(comment: str = None, claim_id: str = None,
|
||||||
|
|
||||||
timestamp = int(time.time())
|
timestamp = int(time.time())
|
||||||
comment_id = create_comment_id(comment, channel_id, timestamp)
|
comment_id = create_comment_id(comment, channel_id, timestamp)
|
||||||
|
with database_proxy.atomic():
|
||||||
new_comment = Comment.create(
|
new_comment = Comment.create(
|
||||||
claim_id=claim_id,
|
claim_id=claim_id,
|
||||||
comment_id=comment_id,
|
comment_id=comment_id,
|
||||||
|
@ -216,6 +217,41 @@ def create_comment(comment: str = None, claim_id: str = None,
|
||||||
return get_comment(new_comment.comment_id)
|
return get_comment(new_comment.comment_id)
|
||||||
|
|
||||||
|
|
||||||
|
def delete_comment(comment_id: str) -> bool:
|
||||||
|
try:
|
||||||
|
comment: Comment = Comment.get_by_id(comment_id)
|
||||||
|
except DoesNotExist as e:
|
||||||
|
raise ValueError from e
|
||||||
|
else:
|
||||||
|
with database_proxy.atomic():
|
||||||
|
return 0 < comment.delete_instance(True, delete_nullable=True)
|
||||||
|
|
||||||
|
|
||||||
|
def edit_comment(comment_id: str, new_comment: str, new_sig: str, new_ts: str) -> bool:
|
||||||
|
try:
|
||||||
|
comment: Comment = Comment.get_by_id(comment_id)
|
||||||
|
except DoesNotExist as e:
|
||||||
|
raise ValueError from e
|
||||||
|
else:
|
||||||
|
with database_proxy.atomic():
|
||||||
|
comment.comment = new_comment
|
||||||
|
comment.signature = new_sig
|
||||||
|
comment.signing_ts = new_ts
|
||||||
|
|
||||||
|
# todo: add a 'last-modified' timestamp
|
||||||
|
comment.timestamp = int(time.time())
|
||||||
|
return comment.save() > 0
|
||||||
|
|
||||||
|
|
||||||
|
def set_hidden_flag(comment_ids: typing.List[str], hidden=True) -> bool:
|
||||||
|
# sets `is_hidden` flag for all `comment_ids` to the `hidden` param
|
||||||
|
with database_proxy.atomic():
|
||||||
|
update = (Comment
|
||||||
|
.update(is_hidden=hidden)
|
||||||
|
.where(Comment.comment_id.in_(comment_ids)))
|
||||||
|
return update.execute() > 0
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
logger = logging.getLogger('peewee')
|
logger = logging.getLogger('peewee')
|
||||||
logger.addHandler(logging.StreamHandler())
|
logger.addHandler(logging.StreamHandler())
|
||||||
|
|
Loading…
Reference in a new issue