diff --git a/files/classes/mod_logs.py b/files/classes/mod_logs.py index d394b90817..d56fd8f682 100644 --- a/files/classes/mod_logs.py +++ b/files/classes/mod_logs.py @@ -24,6 +24,7 @@ class ModAction(Base): user = relationship("User", primaryjoin="User.id==ModAction.user_id") target_user = relationship("User", primaryjoin="User.id==ModAction.target_user_id") target_post = relationship("Post") + target_comment = relationship("Comment") def __init__(self, *args, **kwargs): if "created_utc" not in kwargs: kwargs["created_utc"] = int(time.time()) @@ -41,7 +42,7 @@ class ModAction(Base): def note(self): if self.kind=="ban_user": if self.target_post: return f'for post' - elif self.target_comment_id: return f'for comment' + if self.target_comment: return f'for comment' else: return self._note else: return self._note or ""