streamlit-nightly 1.35.1.dev20240610__py2.py3-none-any.whl → 1.35.1.dev20240612__py2.py3-none-any.whl

streamlit/runtime/caching/__init__.py

@@ -14,24 +14,26 @@
 
 from __future__ import annotations
 
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
-from google.protobuf.message import Message
-
-from streamlit.proto.Block_pb2 import Block
 from streamlit.runtime.caching.cache_data_api import (
     CACHE_DATA_MESSAGE_REPLAY_CTX,
     CacheDataAPI,
-    _data_caches,
+    get_data_cache_stats_provider,
 )
 from streamlit.runtime.caching.cache_errors import CACHE_DOCS_URL
 from streamlit.runtime.caching.cache_resource_api import (
     CACHE_RESOURCE_MESSAGE_REPLAY_CTX,
     CacheResourceAPI,
-    _resource_caches,
+    get_resource_cache_stats_provider,
 )
 from streamlit.runtime.caching.legacy_cache_api import cache as _cache
-from streamlit.runtime.state.common import WidgetMetadata
+
+if TYPE_CHECKING:
+    from google.protobuf.message import Message
+
+    from streamlit.proto.Block_pb2 import Block
+    from streamlit.runtime.state.common import WidgetMetadata
 
 
 def save_element_message(
@@ -84,12 +86,6 @@ def save_media_data(image_data: bytes | str, mimetype: str, image_id: str) -> No
     CACHE_RESOURCE_MESSAGE_REPLAY_CTX.save_image_data(image_data, mimetype, image_id)
 
 
-# Explicitly export public symbols
-from streamlit.runtime.caching.cache_data_api import get_data_cache_stats_provider
-from streamlit.runtime.caching.cache_resource_api import (
-    get_resource_cache_stats_provider,
-)
-
 # Create and export public API singletons.
 cache_data = CacheDataAPI(decorator_metric_name="cache_data")
 cache_resource = CacheResourceAPI(decorator_metric_name="cache_resource")

streamlit/runtime/caching/cache_data_api.py

@@ -19,8 +19,17 @@ from __future__ import annotations
 import pickle
 import threading
 import types
-from datetime import timedelta
-from typing import Any, Callable, Final, Literal, TypeVar, Union, cast, overload
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Callable,
+    Final,
+    Literal,
+    TypeVar,
+    Union,
+    cast,
+    overload,
+)
 
 from typing_extensions import TypeAlias
 
@@ -44,7 +53,6 @@ from streamlit.runtime.caching.cached_message_replay import (
     MultiCacheResults,
     show_widget_replay_deprecation,
 )
-from streamlit.runtime.caching.hashing import HashFuncsDict
 from streamlit.runtime.caching.storage import (
     CacheStorage,
     CacheStorageContext,
@@ -63,6 +71,11 @@ from streamlit.runtime.scriptrunner.script_run_context import get_script_run_ctx
 from streamlit.runtime.stats import CacheStat, CacheStatsProvider, group_stats
 from streamlit.time_util import time_to_seconds
 
+if TYPE_CHECKING:
+    from datetime import timedelta
+
+    from streamlit.runtime.caching.hashing import HashFuncsDict
+
 _LOGGER: Final = get_logger(__name__)
 
 CACHE_DATA_MESSAGE_REPLAY_CTX = CachedMessageReplayContext(CacheType.DATA)
@@ -344,8 +357,7 @@ class CacheDataAPI:
 
     # Bare decorator usage
     @overload
-    def __call__(self, func: F) -> F:
-        ...
+    def __call__(self, func: F) -> F: ...
 
     # Decorator with arguments
     @overload
@@ -358,8 +370,7 @@ class CacheDataAPI:
         persist: CachePersistType | bool = None,
         experimental_allow_widgets: bool = False,
         hash_funcs: HashFuncsDict | None = None,
-    ) -> Callable[[F], F]:
-        ...
+    ) -> Callable[[F], F]: ...
 
     def __call__(
         self,

streamlit/runtime/caching/cache_errors.py

@@ -14,13 +14,15 @@
 
 from __future__ import annotations
 
-import types
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from streamlit import type_util
 from streamlit.errors import MarkdownFormattedException, StreamlitAPIException
 from streamlit.runtime.caching.cache_type import CacheType, get_decorator_api_name
 
+if TYPE_CHECKING:
+    from types import FunctionType
+
 CACHE_DOCS_URL = "https://docs.streamlit.io/library/advanced-features/caching"
 
 
@@ -47,7 +49,7 @@ class UnhashableParamError(StreamlitAPIException):
     def __init__(
         self,
         cache_type: CacheType,
-        func: types.FunctionType,
+        func: FunctionType,
         arg_name: str | None,
         arg_value: Any,
         orig_exc: BaseException,
@@ -59,7 +61,7 @@ class UnhashableParamError(StreamlitAPIException):
     @staticmethod
     def _create_message(
         cache_type: CacheType,
-        func: types.FunctionType,
+        func: FunctionType,
         arg_name: str | None,
         arg_value: Any,
     ) -> str:
@@ -96,7 +98,7 @@ class CacheReplayClosureError(StreamlitAPIException):
     def __init__(
         self,
         cache_type: CacheType,
-        cached_func: types.FunctionType,
+        cached_func: FunctionType,
     ):
         func_name = get_cached_func_name_md(cached_func)
         decorator_name = get_decorator_api_name(cache_type)
@@ -118,7 +120,7 @@ How to fix this:
 
 
 class UnserializableReturnValueError(MarkdownFormattedException):
-    def __init__(self, func: types.FunctionType, return_value: types.FunctionType):
+    def __init__(self, func: FunctionType, return_value: FunctionType):
         MarkdownFormattedException.__init__(
             self,
             f"""

streamlit/runtime/caching/cache_resource_api.py

@@ -19,8 +19,7 @@ from __future__ import annotations
 import math
 import threading
 import types
-from datetime import timedelta
-from typing import Any, Callable, Final, TypeVar, cast, overload
+from typing import TYPE_CHECKING, Any, Callable, Final, TypeVar, cast, overload
 
 from cachetools import TTLCache
 from typing_extensions import TypeAlias
@@ -44,12 +43,16 @@ from streamlit.runtime.caching.cached_message_replay import (
     MultiCacheResults,
     show_widget_replay_deprecation,
 )
-from streamlit.runtime.caching.hashing import HashFuncsDict
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner.script_run_context import get_script_run_ctx
 from streamlit.runtime.stats import CacheStat, CacheStatsProvider, group_stats
 from streamlit.time_util import time_to_seconds
 
+if TYPE_CHECKING:
+    from datetime import timedelta
+
+    from streamlit.runtime.caching.hashing import HashFuncsDict
+
 _LOGGER: Final = get_logger(__name__)
 
 
@@ -223,8 +226,7 @@ class CacheResourceAPI:
 
     # Bare decorator usage
     @overload
-    def __call__(self, func: F) -> F:
-        ...
+    def __call__(self, func: F) -> F: ...
 
     # Decorator with arguments
     @overload
@@ -237,8 +239,7 @@ class CacheResourceAPI:
         validate: ValidateFunc | None = None,
         experimental_allow_widgets: bool = False,
         hash_funcs: HashFuncsDict | None = None,
-    ) -> Callable[[F], F]:
-        ...
+    ) -> Callable[[F], F]: ...
 
     def __call__(
         self,
@@ -490,11 +491,11 @@ class ResourceCache(Cache):
 
     @property
     def max_entries(self) -> float:
-        return cast(float, self._mem_cache.maxsize)
+        return self._mem_cache.maxsize
 
     @property
     def ttl_seconds(self) -> float:
-        return cast(float, self._mem_cache.ttl)
+        return self._mem_cache.ttl
 
     def read_result(self, key: str) -> CachedResult:
         """Read a value and associated messages from the cache.

streamlit/runtime/caching/cache_utils.py

@@ -21,10 +21,9 @@ import hashlib
 import inspect
 import threading
 import time
-import types
 from abc import abstractmethod
 from collections import defaultdict
-from typing import Any, Callable, Final
+from typing import TYPE_CHECKING, Any, Callable, Final
 
 from streamlit import type_util
 from streamlit.elements.spinner import spinner
@@ -38,7 +37,6 @@ from streamlit.runtime.caching.cache_errors import (
     UnserializableReturnValueError,
     get_cached_func_name_md,
 )
-from streamlit.runtime.caching.cache_type import CacheType
 from streamlit.runtime.caching.cached_message_replay import (
     CachedMessageReplayContext,
     CachedResult,
@@ -49,6 +47,11 @@ from streamlit.runtime.caching.hashing import HashFuncsDict, update_hash
 from streamlit.type_util import UNEVALUATED_DATAFRAME_TYPES
 from streamlit.util import HASHLIB_KWARGS
 
+if TYPE_CHECKING:
+    from types import FunctionType
+
+    from streamlit.runtime.caching.cache_type import CacheType
+
 _LOGGER: Final = get_logger(__name__)
 
 # The timer function we use with TTLCache. This is the default timer func, but
@@ -120,7 +123,7 @@ class CachedFuncInfo:
 
     def __init__(
         self,
-        func: types.FunctionType,
+        func: FunctionType,
         show_spinner: bool | str,
         allow_widgets: bool,
         hash_funcs: HashFuncsDict | None,
@@ -354,7 +357,7 @@ class CachedFunc:
 
 def _make_value_key(
     cache_type: CacheType,
-    func: types.FunctionType,
+    func: FunctionType,
     func_args: tuple[Any, ...],
     func_kwargs: dict[str, Any],
     hash_funcs: HashFuncsDict | None,
@@ -419,7 +422,7 @@ def _make_value_key(
     return value_key
 
 
-def _make_function_key(cache_type: CacheType, func: types.FunctionType) -> str:
+def _make_function_key(cache_type: CacheType, func: FunctionType) -> str:
     """Create the unique key for a function's cache.
 
     A function's key is stable across reruns of the app, and changes when
@@ -458,7 +461,7 @@ def _make_function_key(cache_type: CacheType, func: types.FunctionType) -> str:
     return cache_key
 
 
-def _get_positional_arg_name(func: types.FunctionType, arg_index: int) -> str | None:
+def _get_positional_arg_name(func: FunctionType, arg_index: int) -> str | None:
     """Return the name of a function's positional argument.
 
     If arg_index is out of range, or refers to a parameter that is not a

streamlit/runtime/caching/cached_message_replay.py

@@ -17,28 +17,29 @@ from __future__ import annotations
 import contextlib
 import hashlib
 import threading
-import types
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any, Iterator, Literal, Union
 
-from google.protobuf.message import Message
-
 import streamlit as st
 from streamlit import runtime, util
 from streamlit.deprecation_util import show_deprecation_warning
-from streamlit.proto.Block_pb2 import Block
 from streamlit.runtime.caching.cache_errors import CacheReplayClosureError
-from streamlit.runtime.caching.cache_type import CacheType
 from streamlit.runtime.caching.hashing import update_hash
 from streamlit.runtime.scriptrunner.script_run_context import (
     ScriptRunContext,
     get_script_run_ctx,
 )
-from streamlit.runtime.state.common import WidgetMetadata
 from streamlit.util import HASHLIB_KWARGS
 
 if TYPE_CHECKING:
+    from types import FunctionType
+
+    from google.protobuf.message import Message
+
     from streamlit.delta_generator import DeltaGenerator
+    from streamlit.proto.Block_pb2 import Block
+    from streamlit.runtime.caching.cache_type import CacheType
+    from streamlit.runtime.state.common import WidgetMetadata
 
 
 @dataclass(frozen=True)
@@ -233,7 +234,7 @@ class CachedMessageReplayContext(threading.local):
 
     @contextlib.contextmanager
     def calling_cached_function(
-        self, func: types.FunctionType, allow_widgets: bool
+        self, func: FunctionType, allow_widgets: bool
     ) -> Iterator[None]:
         """Context manager that should wrap the invocation of a cached function.
         It allows us to track any `st.foo` messages that are generated from inside the function
@@ -364,7 +365,7 @@ class CachedMessageReplayContext(threading.local):
 
 
 def replay_cached_messages(
-    result: CachedResult, cache_type: CacheType, cached_func: types.FunctionType
+    result: CachedResult, cache_type: CacheType, cached_func: FunctionType
 ) -> None:
     """Replay the st element function calls that happened when executing a
     cache-decorated function.
@@ -431,7 +432,7 @@ def _make_widget_key(widgets: list[tuple[str, Any]], cache_type: CacheType) -> s
 
 
 def show_widget_replay_deprecation(
-    decorator: Literal["cache_data", "cache_resource"]
+    decorator: Literal["cache_data", "cache_resource"],
 ) -> None:
     show_deprecation_warning(
         "The `experimental_allow_widgets` parameter is deprecated and will be removed "

streamlit/runtime/caching/hashing.py

@@ -202,9 +202,9 @@ class _HashStacks:
     """Stacks of what has been hashed, with at most 1 stack per thread."""
 
     def __init__(self):
-        self._stacks: weakref.WeakKeyDictionary[
-            threading.Thread, _HashStack
-        ] = weakref.WeakKeyDictionary()
+        self._stacks: weakref.WeakKeyDictionary[threading.Thread, _HashStack] = (
+            weakref.WeakKeyDictionary()
+        )
 
     def __repr__(self) -> str:
         return util.repr_(self)

streamlit/runtime/caching/storage/cache_storage_protocol.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-""" Declares the CacheStorageContext dataclass, which contains parameter information for
+"""Declares the CacheStorageContext dataclass, which contains parameter information for
 each function decorated by `@st.cache_data` (for example: ttl, max_entries etc.)
 
 Declares the CacheStorageManager protocol, which implementations are used

streamlit/runtime/connection_factory.py

@@ -16,8 +16,7 @@ from __future__ import annotations
 
 import os
 import re
-from datetime import timedelta
-from typing import Any, Final, Literal, TypeVar, overload
+from typing import TYPE_CHECKING, Any, Final, Literal, TypeVar, overload
 
 from streamlit.connections import (
     BaseConnection,
@@ -31,6 +30,9 @@ from streamlit.runtime.caching import cache_resource
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.secrets import secrets_singleton
 
+if TYPE_CHECKING:
+    from datetime import timedelta
+
 # NOTE: Adding support for a new first party connection requires:
 #   1. Adding the new connection name and class to this dict.
 #   2. Writing two new @overloads for connection_factory (one for the case where the

streamlit/runtime/credentials.py

@@ -20,9 +20,8 @@ import json
 import os
 import sys
 import textwrap
-from collections import namedtuple
 from datetime import datetime
-from typing import Final, NoReturn
+from typing import Final, NamedTuple, NoReturn
 from uuid import uuid4
 
 from streamlit import cli_util, env_util, file_util, util
@@ -36,13 +35,10 @@ if env_util.IS_WINDOWS:
 else:
     _CONFIG_FILE_PATH = "~/.streamlit/config.toml"
 
-_Activation = namedtuple(
-    "_Activation",
-    [
-        "email",  # str : the user's email.
-        "is_valid",  # boolean : whether the email is valid.
-    ],
-)
+
+class _Activation(NamedTuple):
+    email: str | None  # the user's email.
+    is_valid: bool  # whether the email is valid.
 
 
 def email_prompt() -> str:
@@ -55,19 +51,14 @@ def email_prompt() -> str:
     )
 
     # IMPORTANT: Break the text below at 80 chars.
-    return """
-      {0}%(welcome)s
+    return f"""
+      {"👋 " if show_emoji else ""}{cli_util.style_for_cli("Welcome to Streamlit!", bold=True)}
 
       If you’d like to receive helpful onboarding emails, news, offers, promotions,
       and the occasional swag, please enter your email address below. Otherwise,
       leave this field blank.
 
-      %(email)s""".format(
-        "👋 " if show_emoji else ""
-    ) % {
-        "welcome": cli_util.style_for_cli("Welcome to Streamlit!", bold=True),
-        "email": cli_util.style_for_cli("Email: ", fg="blue"),
-    }
+      {cli_util.style_for_cli("Email: ", fg="blue")}"""
 
 
 _TELEMETRY_HEADLESS_TEXT = """
@@ -119,10 +110,10 @@ def _send_email(email: str) -> None:
     response.raise_for_status()
 
 
-class Credentials(object):
+class Credentials:
     """Credentials class."""
 
-    _singleton: "Credentials" | None = None
+    _singleton: Credentials | None = None
 
     @classmethod
     def get_current(cls):
@@ -156,7 +147,7 @@ class Credentials(object):
         import toml
 
         try:
-            with open(self._conf_file, "r") as f:
+            with open(self._conf_file) as f:
                 data = toml.load(f).get("general")
             if data is None:
                 raise Exception

streamlit/runtime/forward_msg_cache.py

@@ -111,9 +111,9 @@ class ForwardMsgCache(CacheStatsProvider):
 
         def __init__(self, msg: ForwardMsg | None):
             self.msg = msg
-            self._session_script_run_counts: MutableMapping[
-                AppSession, int
-            ] = WeakKeyDictionary()
+            self._session_script_run_counts: MutableMapping[AppSession, int] = (
+                WeakKeyDictionary()
+            )
 
         def __repr__(self) -> str:
             return util.repr_(self)
@@ -285,13 +285,12 @@ class ForwardMsgCache(CacheStatsProvider):
         self._entries.clear()
 
     def get_stats(self) -> list[CacheStat]:
-        stats: list[CacheStat] = []
-        for entry_hash, entry in self._entries.items():
-            stats.append(
-                CacheStat(
-                    category_name="ForwardMessageCache",
-                    cache_name="",
-                    byte_length=entry.msg.ByteSize() if entry.msg is not None else 0,
-                )
+        stats: list[CacheStat] = [
+            CacheStat(
+                category_name="ForwardMessageCache",
+                cache_name="",
+                byte_length=entry.msg.ByteSize() if entry.msg is not None else 0,
             )
+            for _, entry in self._entries.items()
+        ]
         return group_stats(stats)

streamlit/runtime/forward_msg_queue.py

@@ -14,11 +14,13 @@
 
 from __future__ import annotations
 
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
-from streamlit.proto.Delta_pb2 import Delta
 from streamlit.proto.ForwardMsg_pb2 import ForwardMsg
 
+if TYPE_CHECKING:
+    from streamlit.proto.Delta_pb2 import Delta
+
 
 class ForwardMsgQueue:
     """Accumulates a session's outgoing ForwardMsgs.
@@ -38,7 +40,7 @@ class ForwardMsgQueue:
         # redundant outgoing Deltas (where a newer Delta supersedes
         # an older Delta, with the same delta_path, that's still in the
         # queue).
-        self._delta_index_map: dict[tuple[int, ...], int] = dict()
+        self._delta_index_map: dict[tuple[int, ...], int] = {}
 
     def get_debug(self) -> dict[str, Any]:
         from google.protobuf.json_format import MessageToDict
@@ -102,7 +104,7 @@ class ForwardMsgQueue:
                 }
             ]
 
-        self._delta_index_map = dict()
+        self._delta_index_map = {}
 
     def flush(self) -> list[ForwardMsg]:
         """Clear the queue and return a list of the messages it contained

streamlit/runtime/fragment.py

@@ -19,15 +19,17 @@ import hashlib
 import inspect
 from abc import abstractmethod
 from copy import deepcopy
-from datetime import timedelta
 from functools import wraps
-from typing import Any, Callable, Protocol, TypeVar, overload
+from typing import TYPE_CHECKING, Any, Callable, Protocol, TypeVar, overload
 
 from streamlit.proto.ForwardMsg_pb2 import ForwardMsg
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner import get_script_run_ctx
 from streamlit.time_util import time_to_seconds
 
+if TYPE_CHECKING:
+    from datetime import timedelta
+
 F = TypeVar("F", bound=Callable[..., Any])
 Fragment = Callable[[], Any]
 
@@ -127,9 +129,7 @@ def _fragment(
         active_dg = dg_stack_snapshot[-1]
         h = hashlib.new("md5")
         h.update(
-            f"{non_optional_func.__module__}.{non_optional_func.__qualname__}{active_dg._get_delta_path_str()}".encode(
-                "utf-8"
-            )
+            f"{non_optional_func.__module__}.{non_optional_func.__qualname__}{active_dg._get_delta_path_str()}".encode()
         )
         fragment_id = h.hexdigest()
 
@@ -207,8 +207,7 @@ def fragment(
     func: F,
     *,
     run_every: int | float | timedelta | str | None = None,
-) -> F:
-    ...
+) -> F: ...
 
 
 # Support being able to pass parameters to this decorator (that is, being able to write
@@ -218,8 +217,7 @@ def fragment(
     func: None = None,
     *,
     run_every: int | float | timedelta | str | None = None,
-) -> Callable[[F], F]:
-    ...
+) -> Callable[[F], F]: ...
 
 
 @gather_metrics("experimental_fragment")

streamlit/runtime/media_file_manager.py

@@ -82,12 +82,12 @@ class MediaFileManager:
         self._storage = storage
 
         # Dict of [file_id -> MediaFileMetadata]
-        self._file_metadata: dict[str, MediaFileMetadata] = dict()
+        self._file_metadata: dict[str, MediaFileMetadata] = {}
 
         # Dict[session ID][coordinates] -> file_id.
-        self._files_by_session_and_coord: dict[
-            str, dict[str, str]
-        ] = collections.defaultdict(dict)
+        self._files_by_session_and_coord: dict[str, dict[str, str]] = (
+            collections.defaultdict(dict)
+        )
 
         # MediaFileManager is used from multiple threads, so all operations
         # need to be protected with a Lock. (This is not an RLock, which

streamlit/runtime/memory_media_file_storage.py

@@ -171,13 +171,12 @@ class MemoryMediaFileStorage(MediaFileStorage, CacheStatsProvider):
         # with other threads that may be manipulating the cache.
         files_by_id = self._files_by_id.copy()
 
-        stats: list[CacheStat] = []
-        for file_id, file in files_by_id.items():
-            stats.append(
-                CacheStat(
-                    category_name="st_memory_media_file_storage",
-                    cache_name="",
-                    byte_length=len(file.content),
-                )
+        stats: list[CacheStat] = [
+            CacheStat(
+                category_name="st_memory_media_file_storage",
+                cache_name="",
+                byte_length=len(file.content),
             )
+            for _, file in files_by_id.items()
+        ]
         return group_stats(stats)

streamlit/runtime/metrics_util.py

@@ -306,16 +306,14 @@ F = TypeVar("F", bound=Callable[..., Any])
 def gather_metrics(
     name: str,
     func: F,
-) -> F:
-    ...
+) -> F: ...
 
 
 @overload
 def gather_metrics(
     name: str,
     func: None = None,
-) -> Callable[[F], F]:
-    ...
+) -> Callable[[F], F]: ...
 
 
 def gather_metrics(name: str, func: F | None = None) -> Callable[[F], F] | F: