diff --git a/models.py b/models.py index 79cd27b..5673525 100644 --- a/models.py +++ b/models.py @@ -2,7 +2,7 @@ from __future__ import annotations from datetime import datetime -from eloquent import Model # type: ignore +from eloquent import Model, QueryBuilder # type: ignore from .config import DB @@ -32,11 +32,17 @@ class Answer (Model): __timestamps__ = False @property - def query ( + def query_rel ( self, ) -> Query: return self.belongs_to (Query) + @property + def answered_flags ( + self, + ) -> list[AnsweredFlag]: + return self.has_many (AnsweredFlag) + def answered ( self, platform: int, @@ -74,7 +80,7 @@ class Query (Model): def answer_histories ( self, ) -> list[Answer]: - return [x.answer for x in self.has_many (QueryAnswerHistory).get ()] + return [x.answer for x in self.has_many (QueryAnswerHistory)] class QueryAnswerHistory (Model): @@ -85,7 +91,7 @@ class QueryAnswerHistory (Model): __timestamps__ = False @property - def query ( + def query_rel ( self, ) -> Query: return self.belongs_to (Query)