Add in missing memoized decorator calls (#133)

FasterSpeeding · web-flow · commit 25be9018bb9b · 2021-07-22T22:20:22.000+02:00
* Add in missing memoized decorator calls

* Prob safer to assume that's just a normal property
diff --git a/sqlalchemy-stubs/dialects/mysql/dml.pyi b/sqlalchemy-stubs/dialects/mysql/dml.pyi
@@ -1,12 +1,14 @@
 from typing import Any
 
+from ... import util as util
 from ...sql.dml import Insert as StandardInsert
 from ...sql.elements import ClauseElement
 
 class Insert(StandardInsert):
     stringify_dialect: str = ...
     @property
     def inserted(self): ...
+    @util.memoized_property
     def inserted_alias(self): ...
     def on_duplicate_key_update(self, *args: Any, **kw: Any) -> None: ...
 
diff --git a/sqlalchemy-stubs/dialects/mysql/mysqlconnector.pyi b/sqlalchemy-stubs/dialects/mysql/mysqlconnector.pyi
@@ -30,6 +30,7 @@ class MySQLDialect_mysqlconnector(MySQLDialect):
     def __init__(self, *arg: Any, **kw: Any) -> None: ...
     @property
     def description_encoding(self): ...
+    @util.memoized_property
     def supports_unicode_statements(self): ...
     @classmethod
     def dbapi(cls): ...
diff --git a/sqlalchemy-stubs/dialects/mysql/mysqldb.pyi b/sqlalchemy-stubs/dialects/mysql/mysqldb.pyi
@@ -27,6 +27,7 @@ class MySQLDialect_mysqldb(MySQLDialect):
     statement_compiler: Any = ...
     preparer: Any = ...
     def __init__(self, **kwargs: Any) -> None: ...
+    @util.memoized_property
     def supports_server_side_cursors(self): ...
     @classmethod
     def dbapi(cls): ...
diff --git a/sqlalchemy-stubs/dialects/mysql/pymysql.pyi b/sqlalchemy-stubs/dialects/mysql/pymysql.pyi
@@ -10,6 +10,7 @@ class MySQLDialect_pymysql(MySQLDialect_mysqldb):
     description_encoding: Any = ...
     supports_unicode_statements: bool = ...
     supports_unicode_binds: bool = ...
+    @langhelpers.memoized_property
     def supports_server_side_cursors(self): ...
     @classmethod
     def dbapi(cls): ...
diff --git a/sqlalchemy-stubs/dialects/postgresql/dml.pyi b/sqlalchemy-stubs/dialects/postgresql/dml.pyi
@@ -1,11 +1,13 @@
 from typing import Any
 from typing import Optional
 
+from ... import util as util
 from ...sql.dml import Insert as StandardInsert
 from ...sql.elements import ClauseElement
 
 class Insert(StandardInsert):
     stringify_dialect: str = ...
+    @util.memoized_property
     def excluded(self): ...
     def on_conflict_do_update(
         self,
diff --git a/sqlalchemy-stubs/dialects/postgresql/pypostgresql.pyi b/sqlalchemy-stubs/dialects/postgresql/pypostgresql.pyi
@@ -24,6 +24,7 @@ class PGDialect_pypostgresql(PGDialect):
     colspecs: Any = ...
     @classmethod
     def dbapi(cls): ...
+    @util.memoized_property
     def dbapi_exception_translation_map(self): ...
     def create_connect_args(self, url: Any): ...
     def is_disconnect(self, e: Any, connection: Any, cursor: Any): ...
diff --git a/sqlalchemy-stubs/dialects/sqlite/dml.pyi b/sqlalchemy-stubs/dialects/sqlite/dml.pyi
@@ -1,11 +1,13 @@
 from typing import Any
 from typing import Optional
 
+from ... import util as util
 from ...sql.dml import Insert as StandardInsert
 from ...sql.elements import ClauseElement
 
 class Insert(StandardInsert):
     stringify_dialect: str = ...
+    @util.memoized_property
     def excluded(self): ...
     def on_conflict_do_update(
         self,
diff --git a/sqlalchemy-stubs/engine/cursor.pyi b/sqlalchemy-stubs/engine/cursor.pyi
@@ -14,6 +14,7 @@ from .result import Result
 from .result import ResultMetaData
 from .result import Row
 from .row import LegacyRow
+from .. import util as util
 
 _TCursorResult = TypeVar("_TCursorResult", bound=CursorResult)
 
@@ -176,6 +177,7 @@ class BaseCursorResult:
     def prefetch_cols(self) -> bool: ...
     def supports_sane_rowcount(self) -> bool: ...
     def supports_sane_multi_rowcount(self) -> bool: ...
+    @util.memoized_property
     def rowcount(self) -> int: ...
     @property
     def lastrowid(self) -> Any: ...
diff --git a/sqlalchemy-stubs/ext/associationproxy.pyi b/sqlalchemy-stubs/ext/associationproxy.pyi
@@ -68,6 +68,7 @@ class AssociationProxyInstance:
     def local_attr(self): ...
     @property
     def attr(self): ...
+    @util.memoized_property
     def scalar(self): ...
     @property
     def info(self): ...
diff --git a/sqlalchemy-stubs/orm/attributes.pyi b/sqlalchemy-stubs/orm/attributes.pyi
@@ -70,8 +70,11 @@ class QueryableAttribute(
     ) -> None: ...
     def __reduce__(self): ...
     def get_history(self, instance: Any, passive: Any = ...): ...
+    @util.memoized_property
     def info(self): ...
+    @util.memoized_property
     def parent(self): ...
+    @util.memoized_property
     def expression(self): ...
     def __clause_element__(self): ...
     def adapt_to_entity(self, adapt_to_entity: Any): ...
@@ -82,6 +85,7 @@ class QueryableAttribute(
     def reverse_operate(self, op: Any, other: Any, **kwargs: Any): ...
     def hasparent(self, state: Any, optimistic: bool = ...): ...
     def __getattr__(self, key: Any): ...
+    @util.memoized_property
     def property(self): ...
 
 _T = TypeVar("_T")
diff --git a/sqlalchemy-stubs/orm/base.pyi b/sqlalchemy-stubs/orm/base.pyi
@@ -61,6 +61,7 @@ class InspectionAttr:
     extension_type: Any = ...
 
 class InspectionAttrInfo(InspectionAttr):
+    @util.memoized_property
     def info(self): ...
 
 class _MappedAttribute: ...
diff --git a/sqlalchemy-stubs/orm/descriptor_props.pyi b/sqlalchemy-stubs/orm/descriptor_props.pyi
@@ -32,6 +32,7 @@ class CompositeProperty(DescriptorProperty[_T]):
     def __init__(self, class_: Any, *attrs: Any, **kwargs: Any) -> None: ...
     def instrument_class(self, mapper: Any) -> None: ...
     def do_init(self) -> None: ...
+    @util.memoized_property
     def props(self): ...
     @property
     def columns(self): ...
@@ -44,8 +45,10 @@ class CompositeProperty(DescriptorProperty[_T]):
         ): ...
     class Comparator(PropComparator):
         __hash__: Any = ...
+        @util.memoized_property
         def clauses(self): ...
         def __clause_element__(self): ...
+        @util.memoized_property
         def expression(self): ...
         def __eq__(self, other: Any) -> Any: ...
         def __ne__(self, other: Any) -> Any: ...
diff --git a/sqlalchemy-stubs/orm/instrumentation.pyi b/sqlalchemy-stubs/orm/instrumentation.pyi
@@ -17,7 +17,6 @@ class ClassManager(HasMemoized, dict):
     expired_attribute_loader: Any = ...
     init_method: Any = ...
     factory: Any = ...
-    mapper: Any = ...
     declarative_scan: Any = ...
     registry: Any = ...
     @property
@@ -34,6 +33,7 @@ class ClassManager(HasMemoized, dict):
     def __eq__(self, other: Any) -> Any: ...
     @property
     def is_mapped(self): ...
+    @util.memoized_property
     def mapper(self) -> None: ...
     def manage(self) -> None: ...
     def manager_getter(self): ...
diff --git a/sqlalchemy-stubs/orm/mapper.pyi b/sqlalchemy-stubs/orm/mapper.pyi
@@ -120,22 +120,31 @@ class Mapper(
     @property
     def iterate_properties(self): ...
     with_polymorphic_mappers: Any = ...
+    @HasMemoized.memoized_instancemethod
     def __clause_element__(self): ...
+    @util.memoized_property
     def select_identity_token(self): ...
     @property
     def selectable(self): ...
+    @HasMemoized.memoized_attribute
     def attrs(self): ...
+    @HasMemoized.memoized_attribute
     def all_orm_descriptors(self): ...
+    @HasMemoized.memoized_attribute
     def synonyms(self): ...
     @property
     def entity_namespace(self): ...
+    @HasMemoized.memoized_attribute
     def column_attrs(self): ...
+    @HasMemoized.memoized_attribute
     def relationships(self): ...
+    @HasMemoized.memoized_attribute
     def composites(self): ...
     def common_parent(self, other: Any): ...
     def is_sibling(self, other: Any): ...
     def isa(self, other: Any): ...
     def iterate_to_root(self) -> None: ...
+    @HasMemoized.memoized_attribute
     def self_and_descendants(self): ...
     def polymorphic_iterator(self): ...
     def primary_mapper(self): ...
diff --git a/sqlalchemy-stubs/orm/path_registry.pyi b/sqlalchemy-stubs/orm/path_registry.pyi
@@ -73,7 +73,9 @@ class PropRegistry(PathRegistry):
     path: Any = ...
     natural_path: Any = ...
     def __init__(self, parent: Any, prop: Any) -> None: ...
+    @util.memoized_property
     def has_entity(self): ...
+    @util.memoized_property
     def entity(self): ...
     @property
     def mapper(self): ...
diff --git a/sqlalchemy-stubs/orm/relationships.pyi b/sqlalchemy-stubs/orm/relationships.pyi
@@ -145,7 +145,9 @@ class RelationshipProperty(StrategizedProperty[_T]):
             extra_criteria: Any = ...,
         ) -> None: ...
         def adapt_to_entity(self, adapt_to_entity: Any): ...
+        @util.memoized_property
         def entity(self): ...
+        @util.memoized_property
         def mapper(self): ...
         def __clause_element__(self): ...
         def of_type(self, cls: Any): ...
@@ -157,6 +159,7 @@ class RelationshipProperty(StrategizedProperty[_T]):
         def has(self, criterion: Optional[Any] = ..., **kwargs: Any): ...
         def contains(self, other: Any, **kwargs: Any): ...
         def __ne__(self, other: Any) -> Any: ...
+        @util.memoized_property
         def property(self): ...
     def merge(
         self,
@@ -177,7 +180,9 @@ class RelationshipProperty(StrategizedProperty[_T]):
         visited_states: Any,
         halt_on: Optional[Any] = ...,
     ) -> None: ...
+    @util.memoized_property
     def entity(self) -> Union[AliasedInsp, Mapper]: ...
+    @util.memoized_property
     def mapper(self) -> Mapper: ...
     def do_init(self) -> None: ...
     @property
@@ -223,9 +228,13 @@ class JoinCondition:
     def primaryjoin_minus_local(self): ...
     @property
     def secondaryjoin_minus_local(self): ...
+    @util.memoized_property
     def primaryjoin_reverse_remote(self): ...
+    @util.memoized_property
     def remote_columns(self): ...
+    @util.memoized_property
     def local_columns(self): ...
+    @util.memoized_property
     def foreign_key_columns(self): ...
     def join_targets(
         self,
diff --git a/sqlalchemy-stubs/orm/state.pyi b/sqlalchemy-stubs/orm/state.pyi
@@ -31,6 +31,7 @@ class InstanceState(interfaces.InspectionAttrInfo):
     committed_state: Any = ...
     expired_attributes: Any = ...
     def __init__(self, obj: Any, manager: Any) -> None: ...
+    @util.memoized_property
     def attrs(self): ...
     @property
     def transient(self): ...
@@ -52,7 +53,9 @@ class InstanceState(interfaces.InspectionAttrInfo):
     def identity(self): ...
     @property
     def identity_key(self): ...
+    @util.memoized_property
     def parents(self): ...
+    @util.memoized_property
     def mapper(self): ...
     @property
     def has_identity(self): ...
diff --git a/sqlalchemy-stubs/pool/base.pyi b/sqlalchemy-stubs/pool/base.pyi
@@ -44,7 +44,9 @@ class _ConnectionRecord:
     fairy_ref: Any = ...
     starttime: Any = ...
     connection: Any = ...
+    @util.memoized_property
     def info(self): ...
+    @util.memoized_property
     def record_info(self): ...
     @classmethod
     def checkout(cls, pool: Any): ...
@@ -64,6 +66,7 @@ class _ConnectionFairy:
     ) -> None: ...
     @property
     def is_valid(self): ...
+    @util.memoized_property
     def info(self): ...
     @property
     def record_info(self): ...
diff --git a/sqlalchemy-stubs/pool/impl.pyi b/sqlalchemy-stubs/pool/impl.pyi
@@ -48,6 +48,7 @@ class SingletonThreadPool(Pool):
     def connect(self): ...
 
 class StaticPool(Pool):
+    @util.memoized_property
     def connection(self): ...
     def status(self): ...
     def dispose(self) -> None: ...
diff --git a/sqlalchemy-stubs/util/_concurrency_py3k.pyi b/sqlalchemy-stubs/util/_concurrency_py3k.pyi
@@ -1,5 +1,7 @@
 import sys
 
+from .. import util as util
+
 if sys.version_info >= (3, 0):
     import asyncio as asyncio  # noqa
     from typing import Any
@@ -18,7 +20,8 @@ if sys.version_info >= (3, 0):
         **kwargs: Any,
     ) -> _T: ...
     class AsyncAdaptedLock:
-        mutex: Any = ...
+        @util.memoized_property
+        def mutex(self) -> Any: ...
         def __init__(self) -> None: ...
         def __enter__(self: _AAL) -> _AAL: ...
         def __exit__(self, *arg: Any, **kw: Any) -> None: ...
