mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 13:34:28 +00:00
105 lines
3.2 KiB
Python
105 lines
3.2 KiB
Python
import re
|
|
import datetime
|
|
from sqlalchemy import Column, \
|
|
Integer, \
|
|
Text, \
|
|
Boolean, \
|
|
DateTime, \
|
|
ForeignKey, \
|
|
String
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.ext.declarative import declared_attr
|
|
|
|
|
|
class Diario:
|
|
__tablename__ = "diario"
|
|
|
|
@declared_attr
|
|
def diario_id(self):
|
|
return Column(Integer, primary_key=True)
|
|
|
|
@declared_attr
|
|
def creator_id(self):
|
|
return Column(Integer, ForeignKey("users.uid"))
|
|
|
|
@declared_attr
|
|
def quoted_account_id(self):
|
|
return Column(Integer, ForeignKey("users.uid"))
|
|
|
|
@declared_attr
|
|
def quoted(self):
|
|
return Column(String)
|
|
|
|
@declared_attr
|
|
def text(self):
|
|
return Column(Text)
|
|
|
|
@declared_attr
|
|
def context(self):
|
|
return Column(Text)
|
|
|
|
@declared_attr
|
|
def timestamp(self) -> datetime.datetime:
|
|
return Column(DateTime, nullable=False)
|
|
|
|
@declared_attr
|
|
def media_url(self):
|
|
return Column(String)
|
|
|
|
@declared_attr
|
|
def spoiler(self):
|
|
return Column(Boolean, default=False)
|
|
|
|
@declared_attr
|
|
def creator(self):
|
|
return relationship("User", foreign_keys=self.creator_id, backref="diario_created")
|
|
|
|
@declared_attr
|
|
def quoted_account(self):
|
|
return relationship("User", foreign_keys=self.quoted_account_id, backref="diario_quoted")
|
|
|
|
def json(self) -> dict:
|
|
return {
|
|
"diario_id": self.diario_id,
|
|
"creator": self.creator.json() if self.creator else None,
|
|
"quoted_account": self.quoted_account.json() if self.quoted_account else None,
|
|
"quoted": self.quoted,
|
|
"text": self.text,
|
|
"context": self.context,
|
|
"timestamp": self.timestamp.isoformat(),
|
|
"media_url": self.media_url,
|
|
"spoiler": self.spoiler
|
|
}
|
|
|
|
def __repr__(self):
|
|
return f"<Diario diario_id={self.diario_id}" \
|
|
f" creator_id={self.creator_id}" \
|
|
f" quoted_account_id={self.quoted_account_id}" \
|
|
f" quoted={self.quoted}" \
|
|
f" text={self.text}" \
|
|
f" context={self.context}" \
|
|
f" timestamp={self.timestamp}" \
|
|
f" media_url={self.media_url}" \
|
|
f" spoiler={self.spoiler}>"
|
|
|
|
def __str__(self):
|
|
text = f"Riga #{self.diario_id}"
|
|
text += f" (salvata da {str(self.creator)}"
|
|
text += f" il {self.timestamp.strftime('%Y-%m-%d %H:%M')}):\n"
|
|
if self.media_url is not None:
|
|
text += f"{self.media_url}\n"
|
|
if self.text is not None:
|
|
if self.spoiler:
|
|
hidden = re.sub(r"\w", "█", self.text)
|
|
text += f"\"{hidden}\"\n"
|
|
else:
|
|
text += f"[b]\"{self.text}\"[/b]\n"
|
|
if self.quoted_account is not None:
|
|
text += f" —{str(self.quoted_account)}"
|
|
elif self.quoted is not None:
|
|
text += f" —{self.quoted}"
|
|
else:
|
|
text += f" —Anonimo"
|
|
if self.context:
|
|
text += f", [i]{self.context}[/i]"
|
|
return text
|