From 86001a19e9eba7bfb6ac58b046c72977fc6161e1 Mon Sep 17 00:00:00 2001 From: TrisNol Date: Fri, 11 Aug 2023 15:59:16 +0200 Subject: [PATCH] refactor: Add missing type declaration to ORM entities --- .../utils/postgres/entities.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/aki_prj23_transparenzregister/utils/postgres/entities.py b/src/aki_prj23_transparenzregister/utils/postgres/entities.py index e249f77..6d12edf 100644 --- a/src/aki_prj23_transparenzregister/utils/postgres/entities.py +++ b/src/aki_prj23_transparenzregister/utils/postgres/entities.py @@ -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")