diff options
author | Federico Caselli <cfederico87@gmail.com> | 2022-11-03 20:52:21 +0100 |
---|---|---|
committer | Federico Caselli <cfederico87@gmail.com> | 2022-11-16 23:03:04 +0100 |
commit | 4eb4ceca36c7ce931ea65ac06d6ed08bf459fc66 (patch) | |
tree | 4970cff3f78489a4a0066cd27fd4bae682402957 /lib/sqlalchemy/orm/mapper.py | |
parent | 3fc6c40ea77c971d3067dab0fdf57a5b5313b69b (diff) | |
download | sqlalchemy-4eb4ceca36c7ce931ea65ac06d6ed08bf459fc66.tar.gz |
Try running pyupgrade on the code
command run is "pyupgrade --py37-plus --keep-runtime-typing --keep-percent-format <files...>"
pyupgrade will change assert_ to assertTrue. That was reverted since assertTrue does not
exists in sqlalchemy fixtures
Change-Id: Ie1ed2675c7b11d893d78e028aad0d1576baebb55
Diffstat (limited to 'lib/sqlalchemy/orm/mapper.py')
-rw-r--r-- | lib/sqlalchemy/orm/mapper.py | 119 |
1 files changed, 49 insertions, 70 deletions
diff --git a/lib/sqlalchemy/orm/mapper.py b/lib/sqlalchemy/orm/mapper.py index 5f7ff43e4..d15c882c4 100644 --- a/lib/sqlalchemy/orm/mapper.py +++ b/lib/sqlalchemy/orm/mapper.py @@ -143,8 +143,7 @@ def _all_registries() -> Set[registry]: def _unconfigured_mappers() -> Iterator[Mapper[Any]]: for reg in _all_registries(): - for mapper in reg._mappers_to_configure(): - yield mapper + yield from reg._mappers_to_configure() _already_compiling = False @@ -905,8 +904,8 @@ class Mapper( with_polymorphic: Optional[ Tuple[ - Union[Literal["*"], Sequence[Union["Mapper[Any]", Type[Any]]]], - Optional["FromClause"], + Union[Literal["*"], Sequence[Union[Mapper[Any], Type[Any]]]], + Optional[FromClause], ] ] @@ -2518,105 +2517,85 @@ class Mapper( @HasMemoized_ro_memoized_attribute def _insert_cols_evaluating_none(self): - return dict( - ( - table, - frozenset( - col for col in columns if col.type.should_evaluate_none - ), + return { + table: frozenset( + col for col in columns if col.type.should_evaluate_none ) for table, columns in self._cols_by_table.items() - ) + } @HasMemoized.memoized_attribute def _insert_cols_as_none(self): - return dict( - ( - table, - frozenset( - col.key - for col in columns - if not col.primary_key - and not col.server_default - and not col.default - and not col.type.should_evaluate_none - ), + return { + table: frozenset( + col.key + for col in columns + if not col.primary_key + and not col.server_default + and not col.default + and not col.type.should_evaluate_none ) for table, columns in self._cols_by_table.items() - ) + } @HasMemoized.memoized_attribute def _propkey_to_col(self): - return dict( - ( - table, - dict( - (self._columntoproperty[col].key, col) for col in columns - ), - ) + return { + table: {self._columntoproperty[col].key: col for col in columns} for table, columns in self._cols_by_table.items() - ) + } @HasMemoized.memoized_attribute def _pk_keys_by_table(self): - return dict( - (table, frozenset([col.key for col in pks])) + return { + table: frozenset([col.key for col in pks]) for table, pks in self._pks_by_table.items() - ) + } @HasMemoized.memoized_attribute def _pk_attr_keys_by_table(self): - return dict( - ( - table, - frozenset([self._columntoproperty[col].key for col in pks]), - ) + return { + table: frozenset([self._columntoproperty[col].key for col in pks]) for table, pks in self._pks_by_table.items() - ) + } @HasMemoized.memoized_attribute def _server_default_cols( self, ) -> Mapping[FromClause, FrozenSet[Column[Any]]]: - return dict( - ( - table, - frozenset( - [ - col - for col in cast("Iterable[Column[Any]]", columns) - if col.server_default is not None - or ( - col.default is not None - and col.default.is_clause_element - ) - ] - ), + return { + table: frozenset( + [ + col + for col in cast("Iterable[Column[Any]]", columns) + if col.server_default is not None + or ( + col.default is not None + and col.default.is_clause_element + ) + ] ) for table, columns in self._cols_by_table.items() - ) + } @HasMemoized.memoized_attribute def _server_onupdate_default_cols( self, ) -> Mapping[FromClause, FrozenSet[Column[Any]]]: - return dict( - ( - table, - frozenset( - [ - col - for col in cast("Iterable[Column[Any]]", columns) - if col.server_onupdate is not None - or ( - col.onupdate is not None - and col.onupdate.is_clause_element - ) - ] - ), + return { + table: frozenset( + [ + col + for col in cast("Iterable[Column[Any]]", columns) + if col.server_onupdate is not None + or ( + col.onupdate is not None + and col.onupdate.is_clause_element + ) + ] ) for table, columns in self._cols_by_table.items() - ) + } @HasMemoized.memoized_attribute def _server_default_col_keys(self) -> Mapping[FromClause, FrozenSet[str]]: |