refactor: Remove wrong typing from entities

This commit is contained in:
TrisNol 2023-08-11 16:31:58 +02:00
parent 4e7f6bca1d
commit e7627aa2b7

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: SentimentTypeEnum = Column(Enum(SentimentTypeEnum), nullable=False)
sentiment_type = 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: RelationTypeEnum = Column(Enum(RelationTypeEnum), nullable=False)
relation = Column(Enum(RelationTypeEnum), nullable=False)
# company = relationship("Company")
# person = relationship("Person", foreign_keys=[person_id])
@ -182,9 +182,7 @@ 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: RelationTypeCompanyEnum = Column(
Enum(RelationTypeCompanyEnum), nullable=False
)
relation = Column(Enum(RelationTypeCompanyEnum), nullable=False)
# company = relationship("Company")