Fix errors reported by mypy

Signed-off-by: Jan Lindemann <jan@janware.com>
This commit is contained in:
Jan Lindemann 2025-03-01 07:58:19 +01:00
commit 841b5a3391
2 changed files with 22 additions and 12 deletions

View file

@ -29,8 +29,8 @@ class CompositeForeignKey: # export
assert(len(self.__child_col_set) == len(self.__parent_col_set))
self.__len = len(self.__child_col_set)
self.__column_relations: Optional[list[SingleForeignKey]] = None
self.__parent_columns_by_child_column: Optional[dict[str, Column]] = None
self.__child_columns_by_parent_column: Optional[dict[str, Column]] = None
self.__parent_columns_by_child_column: Optional[dict[str, Any]] = None
self.__child_columns_by_parent_column: Optional[dict[str, Any]] = None
def __table_rel_str(self):
return f'{self.__child_table.name} => {self.__parent_table.name}'
@ -72,21 +72,21 @@ class CompositeForeignKey: # export
def parent_columns(self) -> ColumnSet:
return self.__parent_col_set
def parent_column(self, child_column):
def parent_column(self, child_column) -> Any:
child_column_name = child_column if isinstance(child_column, str) else child_column.name
if self.__parent_columns_by_child_column is None:
d: dict[str, Column] = {}
d: dict[str, Any] = {}
assert(len(self.__child_col_set) == len(self.__parent_col_set))
for i in range(0, len(self.__child_col_set)):
d[self.__child_col_set[i].name] = self.__parent_col_set[i]
self.__parent_columns_by_child_column = d
return self.__parent_columns_by_child_column[child_column]
def child_column(self, parent_column):
def child_column(self, parent_column) -> Any:
slog(WARNING, f'{self}: Looking for child column belonging to parent column "{parent_column}"')
parent_column_name = parent_column if isinstance(parent_column, str) else parent_column.name
if self.__child_columns_by_parent_column is None:
d: dict[str, Column] = {}
d: dict[str, Any] = {}
assert(len(self.__parent_col_set) == len(self.__child_col_set))
for i in range(0, len(self.__parent_col_set)):
d[self.__parent_col_set[i].name] = self.__child_col_set[i]