refactor: Add missing type declaration to ORM entities

This commit is contained in:
TrisNol 2023-08-11 15:59:16 +02:00
parent db2da67ee4
commit 86001a19e9

View File

@ -101,7 +101,7 @@ class Sentiment(Base): # type: ignore
company_hr = Column(Integer)
company_court = Column(Integer)
date = Column(DateTime(), default=datetime.now)
sentiment_type = Column(Enum(SentimentTypeEnum), nullable=False)
sentiment_type: SentimentTypeEnum = Column(Enum(SentimentTypeEnum), nullable=False)
value = Column(Float(), nullable=False)
source = Column(String(100))
@ -150,7 +150,7 @@ class PersonRelation(Base): # type: ignore
person_id = mapped_column(ForeignKey("person.id"))
date_from = Column(DateTime(), default=datetime.now)
date_to = Column(DateTime(), default=datetime.now)
relation = Column(Enum(RelationTypeEnum), nullable=False)
relation: RelationTypeEnum = Column(Enum(RelationTypeEnum), nullable=False)
# company = relationship("Company")
# person = relationship("Person", foreign_keys=[person_id])
@ -182,7 +182,9 @@ class CompanyRelation(Base): # type: ignore
company2_id = Column(Integer, nullable=False)
date_from = Column(DateTime(), default=datetime.now)
date_to = Column(DateTime(), default=datetime.now)
relation = Column(Enum(RelationTypeCompanyEnum), nullable=False)
relation: RelationTypeCompanyEnum = Column(
Enum(RelationTypeCompanyEnum), nullable=False
)
# company = relationship("Company")