summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/orm/mapper.py
diff options
context:
space:
mode:
authorFederico Caselli <cfederico87@gmail.com>2022-11-03 20:52:21 +0100
committerFederico Caselli <cfederico87@gmail.com>2022-11-16 23:03:04 +0100
commit4eb4ceca36c7ce931ea65ac06d6ed08bf459fc66 (patch)
tree4970cff3f78489a4a0066cd27fd4bae682402957 /lib/sqlalchemy/orm/mapper.py
parent3fc6c40ea77c971d3067dab0fdf57a5b5313b69b (diff)
downloadsqlalchemy-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.py119
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]]: