e040ed6708
This has exposed an interesting amount of bugs and stopped throwing type errors every 5 seconds It's worth noting that not all models are fully typed, that is, we have `Mapped[Any]` in some places where a narrower type would be nice. Upgrading to SQLA2 we don't *need* this, but it is helpful and makes error checking reliable.
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
from sqlalchemy import *
|
|
from sqlalchemy.orm import relationship
|
|
from files.classes.base import CreatedDateTimeBase
|
|
from files.helpers.config.const import *
|
|
from enum import Enum
|
|
from sqlalchemy import Enum as EnumType
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
class UserTag(Enum):
|
|
Quality = 0
|
|
Good = 1
|
|
Comment = 2
|
|
Warning = 3
|
|
Tempban = 4
|
|
Permban = 5
|
|
Spam = 6
|
|
Bot = 7
|
|
|
|
class UserNote(CreatedDateTimeBase):
|
|
__tablename__ = "usernotes"
|
|
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
author_id: Mapped[int] = mapped_column(Integer, ForeignKey("users.id"), nullable=False)
|
|
reference_user: Mapped[int] = mapped_column(Integer, ForeignKey("users.id", ondelete='CASCADE'), nullable=False)
|
|
reference_comment: Mapped[int | None] = mapped_column(Integer, ForeignKey("comments.id", ondelete='SET NULL'))
|
|
reference_post: Mapped[int | None] = mapped_column(Integer, ForeignKey("submissions.id", ondelete='SET NULL'))
|
|
note: Mapped[str] = mapped_column(String, nullable=False)
|
|
tag: Mapped[str] = mapped_column(EnumType(UserTag), nullable=False)
|
|
|
|
author = relationship("User", foreign_keys='UserNote.author_id')
|
|
user = relationship("User", foreign_keys='UserNote.reference_user', back_populates="notes")
|
|
|
|
comment = relationship("Comment", back_populates="notes")
|
|
post = relationship("Submission", back_populates="notes")
|
|
|
|
def __repr__(self):
|
|
return f"<{self.__class__.__name__}(id={self.id})>"
|
|
|
|
def json(self):
|
|
reference = None
|
|
|
|
if self.comment:
|
|
reference = self.comment.permalink
|
|
elif self.post:
|
|
reference = self.post.permalink
|
|
|
|
data = {'id': self.id,
|
|
'author_name': self.author.username,
|
|
'author_id': self.author.id,
|
|
'user_name': self.user.username,
|
|
'user_id': self.user.id,
|
|
'created': self.created_utc,
|
|
'reference': reference,
|
|
'note': self.note,
|
|
'tag': self.tag.value
|
|
}
|
|
|
|
return data
|