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

streamlit/__init__.py

@@ -44,7 +44,6 @@ For more detailed info, see https://docs.streamlit.io.
 
 # IMPORTANT: Prefix with an underscore anything that the user shouldn't see.
 
-
 import os as _os
 
 # Set Matplotlib backend to avoid a crash.
@@ -266,3 +265,8 @@ experimental_data_editor = _main.experimental_data_editor
 experimental_connection = _deprecate_func_name(
     connection, "experimental_connection", "2024-04-01", name_override="connection"
 )
+
+# make it possible to call streamlit.components.v1.html etc. by importing it here
+# import in the very end to avoid partially-initialized module import errors, because
+# streamlit.components.v1 also uses some streamlit imports
+import streamlit.components.v1  # noqa: F401

streamlit/case_converters.py

@@ -80,11 +80,11 @@ def convert_dict_keys(
         converted by `func`.
 
     """
-    out_dict = dict()
+    out_dict = {}
     for k, v in in_dict.items():
         converted_key = func(k)
 
-        if type(v) is dict:
+        if isinstance(v, dict):
             out_dict[converted_key] = convert_dict_keys(func, v)
         else:
             out_dict[converted_key] = v

streamlit/cli_util.py

@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-""" Utilities related to the CLI."""
+"""Utilities related to the CLI."""
 
 from __future__ import annotations
 
@@ -28,7 +28,7 @@ def print_to_cli(message: str, **kwargs) -> None:
 
         click.secho(message, **kwargs)
     except ImportError:
-        print(message, flush=True)
+        print(message, flush=True)  # noqa: T201
 
 
 def style_for_cli(message: str, **kwargs) -> str:

streamlit/color_util.py

@@ -196,8 +196,8 @@ def _to_color_tuple(
 
         try:
             color = int(r, 16), int(g, 16), int(b, 16), int(a, 16)
-        except:
-            raise InvalidColorException(color)
+        except Exception as ex:
+            raise InvalidColorException(color) from ex
 
     if is_color_tuple_like(color):
         color_tuple = cast(ColorTuple, color)

streamlit/commands/experimental_query_params.py

@@ -125,18 +125,16 @@ def _ensure_no_embed_params(
     all_current_params = parse.parse_qs(query_string, keep_blank_values=True)
     current_embed_params = parse.urlencode(
         {
-            EMBED_QUERY_PARAM: [
-                param
-                for param in util.extract_key_query_params(
+            EMBED_QUERY_PARAM: list(
+                util.extract_key_query_params(
                     all_current_params, param_key=EMBED_QUERY_PARAM
                 )
-            ],
-            EMBED_OPTIONS_QUERY_PARAM: [
-                param
-                for param in util.extract_key_query_params(
+            ),
+            EMBED_OPTIONS_QUERY_PARAM: list(
+                util.extract_key_query_params(
                     all_current_params, param_key=EMBED_OPTIONS_QUERY_PARAM
                 )
-            ],
+            ),
         },
         doseq=True,
     )

streamlit/commands/logo.py

@@ -16,8 +16,6 @@
 
 from __future__ import annotations
 
-from typing import TYPE_CHECKING
-
 from streamlit import url_util
 from streamlit.elements.image import AtomicImage, WidthBehaviour, image_to_url
 from streamlit.errors import StreamlitAPIException
@@ -25,9 +23,6 @@ from streamlit.proto.ForwardMsg_pb2 import ForwardMsg
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner import get_script_run_ctx
 
-if TYPE_CHECKING:
-    from PIL import Image
-
 
 def _invalid_logo_text(field_name: str):
     return f"The {field_name} passed to st.logo is invalid - See [documentation](https://docs.streamlit.io/develop/api-reference/media/st.logo) for more information on valid types"
@@ -126,8 +121,8 @@ def logo(
             image_id="logo",
         )
         fwd_msg.logo.image = image_url
-    except Exception:
-        raise StreamlitAPIException(_invalid_logo_text("image"))
+    except Exception as ex:
+        raise StreamlitAPIException(_invalid_logo_text("image")) from ex
 
     if link:
         # Handle external links:
@@ -149,7 +144,7 @@ def logo(
                 image_id="icon-image",
             )
             fwd_msg.logo.icon_image = icon_image_url
-        except Exception:
-            raise StreamlitAPIException(_invalid_logo_text("icon_image"))
+        except Exception as ex:
+            raise StreamlitAPIException(_invalid_logo_text("icon_image")) from ex
 
     ctx.enqueue(fwd_msg)

streamlit/commands/navigation.py

@@ -15,12 +15,11 @@
 from __future__ import annotations
 
 from pathlib import Path
-from typing import Literal
+from typing import TYPE_CHECKING, Literal
 
 from typing_extensions import TypeAlias
 
 from streamlit.errors import StreamlitAPIException
-from streamlit.navigation.page import StreamlitPage
 from streamlit.proto.ForwardMsg_pb2 import ForwardMsg
 from streamlit.proto.Navigation_pb2 import Navigation as NavigationProto
 from streamlit.runtime.metrics_util import gather_metrics
@@ -28,13 +27,16 @@ from streamlit.runtime.scriptrunner.script_run_context import (
     ScriptRunContext,
     get_script_run_ctx,
 )
-from streamlit.source_util import PageHash, PageInfo
+
+if TYPE_CHECKING:
+    from streamlit.navigation.page import StreamlitPage
+    from streamlit.source_util import PageHash, PageInfo
 
 SectionHeader: TypeAlias = str
 
 
 def pages_from_nav_sections(
-    nav_sections: dict[SectionHeader, list[StreamlitPage]]
+    nav_sections: dict[SectionHeader, list[StreamlitPage]],
 ) -> list[StreamlitPage]:
     page_list = []
     for pages in nav_sections.values():

streamlit/components/lib/local_component_registry.py

@@ -16,14 +16,16 @@ from __future__ import annotations
 
 import os
 import threading
-from typing import Final
+from typing import TYPE_CHECKING, Final
 
 from streamlit import util
 from streamlit.components.types.base_component_registry import BaseComponentRegistry
-from streamlit.components.types.base_custom_component import BaseCustomComponent
 from streamlit.errors import StreamlitAPIException
 from streamlit.logger import get_logger
 
+if TYPE_CHECKING:
+    from streamlit.components.types.base_custom_component import BaseCustomComponent
+
 _LOGGER: Final = get_logger(__name__)
 
 

streamlit/components/types/base_component_registry.py

@@ -15,9 +15,10 @@
 from __future__ import annotations
 
 from abc import abstractmethod
-from typing import Protocol
+from typing import TYPE_CHECKING, Protocol
 
-from streamlit.components.types.base_custom_component import BaseCustomComponent
+if TYPE_CHECKING:
+    from streamlit.components.types.base_custom_component import BaseCustomComponent
 
 
 class BaseComponentRegistry(Protocol):

streamlit/components/v1/component_arrow.py

@@ -22,11 +22,12 @@ from typing import TYPE_CHECKING, Any
 
 from streamlit import type_util
 from streamlit.elements.lib import pandas_styler_utils
-from streamlit.proto.Components_pb2 import ArrowTable as ArrowTableProto
 
 if TYPE_CHECKING:
     from pandas import DataFrame, Index, Series
 
+    from streamlit.proto.Components_pb2 import ArrowTable as ArrowTableProto
+
 
 def marshall(
     proto: ArrowTableProto, data: Any, default_uuid: str | None = None

streamlit/components/v1/component_registry.py

@@ -16,13 +16,17 @@ from __future__ import annotations
 
 import inspect
 import os
-from types import FrameType
+from typing import TYPE_CHECKING
 
-from streamlit.components.types.base_component_registry import BaseComponentRegistry
 from streamlit.components.v1.custom_component import CustomComponent
 from streamlit.runtime import get_instance
 from streamlit.runtime.scriptrunner import get_script_run_ctx
 
+if TYPE_CHECKING:
+    from types import FrameType
+
+    from streamlit.components.types.base_component_registry import BaseComponentRegistry
+
 
 def _get_module_name(caller_frame: FrameType) -> str:
     # Get the caller's module name. `__name__` gives us the module's

streamlit/components/v1/components.py

@@ -19,5 +19,7 @@
 # Since we do not know what other internals are used out in the wild, let's try to
 # model the old behavior and not to break things.
 
-from streamlit.components.v1.component_registry import declare_component
-from streamlit.components.v1.custom_component import *
+from streamlit.components.v1.component_registry import (
+    declare_component as declare_component,
+)
+from streamlit.components.v1.custom_component import *  # noqa: F403

streamlit/components/v1/custom_component.py

@@ -17,8 +17,8 @@ from __future__ import annotations
 import json
 from typing import TYPE_CHECKING, Any
 
-from streamlit import _main, type_util
 from streamlit.components.types.base_custom_component import BaseCustomComponent
+from streamlit.delta_generator import main_dg
 from streamlit.elements.form import current_form_id
 from streamlit.elements.lib.policies import check_cache_replay_rules
 from streamlit.errors import StreamlitAPIException
@@ -29,7 +29,7 @@ from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner import get_script_run_ctx
 from streamlit.runtime.state import NoValue, register_widget
 from streamlit.runtime.state.common import compute_widget_id
-from streamlit.type_util import to_bytes
+from streamlit.type_util import is_bytes_like, is_dataframe_like, to_bytes
 
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
@@ -100,7 +100,7 @@ class CustomComponent(BaseCustomComponent):
             raise MarshallComponentException(f"Argument '{args[0]}' needs a label")
 
         try:
-            import pyarrow
+            import pyarrow  # noqa: F401
 
             from streamlit.components.v1 import component_arrow
         except ImportError:
@@ -122,12 +122,12 @@ And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
         json_args = {}
         special_args = []
         for arg_name, arg_val in all_args.items():
-            if type_util.is_bytes_like(arg_val):
+            if is_bytes_like(arg_val):
                 bytes_arg = SpecialArg()
                 bytes_arg.key = arg_name
                 bytes_arg.bytes = to_bytes(arg_val)
                 special_args.append(bytes_arg)
-            elif type_util.is_dataframe_like(arg_val):
+            elif is_dataframe_like(arg_val):
                 dataframe_arg = SpecialArg()
                 dataframe_arg.key = arg_name
                 component_arrow.marshall(dataframe_arg.arrow_dataframe.data, arg_val)
@@ -221,7 +221,7 @@ And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
 
         # We currently only support writing to st._main, but this will change
         # when we settle on an improved API in a post-layout world.
-        dg = _main
+        dg = main_dg
 
         element = Element()
         return_value = marshall_component(dg, element)

streamlit/config.py

@@ -121,9 +121,7 @@ def set_user_option(key: str, value: Any) -> None:
         return
 
     raise StreamlitAPIException(
-        "{key} cannot be set on the fly. Set as command line option, e.g. streamlit run script.py --{key}, or in config.toml instead.".format(
-            key=key
-        )
+        f"{key} cannot be set on the fly. Set as command line option, e.g. streamlit run script.py --{key}, or in config.toml instead."
     )
 
 
@@ -142,7 +140,7 @@ def get_option(key: str) -> Any:
         config_options = get_config_options()
 
         if key not in config_options:
-            raise RuntimeError('Config key "%s" not defined.' % key)
+            raise RuntimeError(f'Config key "{key}" not defined.')
         return config_options[key].value
 
 
@@ -174,9 +172,9 @@ def get_options_for_section(section: str) -> dict[str, Any]:
 
 def _create_section(section: str, description: str) -> None:
     """Create a config section and store it globally in this module."""
-    assert section not in _section_descriptions, (
-        'Cannot define section "%s" twice.' % section
-    )
+    assert (
+        section not in _section_descriptions
+    ), f'Cannot define section "{section}" twice.'
     _section_descriptions[section] = description
 
 
@@ -247,7 +245,7 @@ def _create_option(
         option.section,
         ", ".join(_section_descriptions.keys()),
     )
-    assert key not in _config_options_template, 'Cannot define option "%s" twice.' % key
+    assert key not in _config_options_template, f'Cannot define option "{key}" twice.'
     _config_options_template[key] = option
     return option
 
@@ -467,6 +465,9 @@ _create_option(
     visibility="hidden",
     type_=bool,
     scriptable=True,
+    deprecated=True,
+    deprecation_text="logger.enableRich has been deprecated and will be removed in a future version. Exception formatting via rich will be automatically used if rich is enable.",
+    expiration_date="2024-09-10",
 )
 
 # Config Section: Client #
@@ -1354,9 +1355,9 @@ def _check_conflicts() -> None:
             "server.port"
         ), "server.port does not work when global.developmentMode is true."
 
-        assert _is_unset("browser.serverPort"), (
-            "browser.serverPort does not work when global.developmentMode is " "true."
-        )
+        assert _is_unset(
+            "browser.serverPort"
+        ), "browser.serverPort does not work when global.developmentMode is true."
 
     # XSRF conflicts
     if get_option("server.enableXsrfProtection"):
@@ -1405,12 +1406,14 @@ def on_config_parsed(
     Callable[[], bool]
         A function that the caller can use to deregister func.
     """
+
     # We need to use the same receiver when we connect or disconnect on the
     # Signal. If we don't do this, then the registered receiver won't be released
     # leading to a memory leak because the Signal will keep a reference of the
     # callable argument. When the callable argument is an object method, then
     # the reference to that object won't be released.
-    receiver = lambda _: func_with_lock()
+    def receiver(_):
+        return func_with_lock()
 
     def disconnect():
         return _on_config_parsed.disconnect(receiver)

streamlit/connections/snowflake_connection.py

@@ -20,7 +20,6 @@
 
 from __future__ import annotations
 
-from datetime import timedelta
 from typing import TYPE_CHECKING, cast
 
 from streamlit.connections import BaseConnection
@@ -29,6 +28,8 @@ from streamlit.errors import StreamlitAPIException
 from streamlit.runtime.caching import cache_data
 
 if TYPE_CHECKING:
+    from datetime import timedelta
+
     from pandas import DataFrame
     from snowflake.connector.cursor import SnowflakeCursor  # type:ignore[import]
     from snowflake.snowpark.session import Session  # type:ignore[import]

streamlit/connections/snowpark_connection.py

@@ -23,7 +23,6 @@ from __future__ import annotations
 import threading
 from collections import ChainMap
 from contextlib import contextmanager
-from datetime import timedelta
 from typing import TYPE_CHECKING, Iterator, cast
 
 from streamlit.connections import BaseConnection
@@ -36,6 +35,8 @@ from streamlit.errors import StreamlitAPIException
 from streamlit.runtime.caching import cache_data
 
 if TYPE_CHECKING:
+    from datetime import timedelta
+
     from pandas import DataFrame
     from snowflake.snowpark.session import Session  # type:ignore[import]
 
@@ -65,9 +66,6 @@ class SnowparkConnection(BaseConnection["Session"]):
 
     def _connect(self, **kwargs) -> Session:
         from snowflake.snowpark.context import get_active_session  # type:ignore[import]
-        from snowflake.snowpark.exceptions import (  # type:ignore[import]
-            SnowparkSessionException,
-        )
         from snowflake.snowpark.session import Session
 
         # If we're running in SiS, just call get_active_session(). Otherwise, attempt to

streamlit/connections/sql_connection.py

@@ -16,7 +16,6 @@ from __future__ import annotations
 
 from collections import ChainMap
 from copy import deepcopy
-from datetime import timedelta
 from typing import TYPE_CHECKING, cast
 
 from streamlit.connections import BaseConnection
@@ -25,6 +24,8 @@ from streamlit.errors import StreamlitAPIException
 from streamlit.runtime.caching import cache_data
 
 if TYPE_CHECKING:
+    from datetime import timedelta
+
     from pandas import DataFrame
     from sqlalchemy.engine import Connection as SQLAlchemyConnection
     from sqlalchemy.engine.base import Engine

streamlit/delta_generator.py

@@ -26,7 +26,6 @@ from typing import (
     Final,
     Hashable,
     Iterable,
-    List,
     Literal,
     NoReturn,
     TypeVar,
@@ -45,7 +44,6 @@ from streamlit import (
     type_util,
     util,
 )
-from streamlit.cursor import Cursor
 from streamlit.elements.alert import AlertMixin
 from streamlit.elements.arrow import ArrowMixin
 from streamlit.elements.balloons import BalloonsMixin
@@ -101,6 +99,7 @@ if TYPE_CHECKING:
     from numpy import typing as npt
     from pandas import DataFrame
 
+    from streamlit.cursor import Cursor
     from streamlit.elements.arrow import Data
     from streamlit.elements.lib.built_in_chart_utils import AddRowsMetadata
 
@@ -358,7 +357,7 @@ class DeltaGenerator(
         return dg
 
     @property
-    def _ancestors(self) -> Iterable["DeltaGenerator"]:
+    def _ancestors(self) -> Iterable[DeltaGenerator]:
         current_dg: DeltaGenerator | None = self
         while current_dg is not None:
             yield current_dg
@@ -735,7 +734,7 @@ def _writes_directly_to_sidebar(dg: DG) -> bool:
 
 
 def _check_nested_element_violation(
-    dg: DeltaGenerator, block_type: str | None, ancestor_block_types: List[BlockType]
+    dg: DeltaGenerator, block_type: str | None, ancestor_block_types: list[BlockType]
 ) -> None:
     """Check if elements are nested in a forbidden way.
 

streamlit/elements/arrow.py

@@ -24,7 +24,6 @@ from typing import (
     Iterable,
     List,
     Literal,
-    Set,
     TypedDict,
     Union,
     cast,
@@ -185,7 +184,7 @@ class DataframeSelectionSerde:
 
 def parse_selection_mode(
     selection_mode: SelectionMode | Iterable[SelectionMode],
-) -> Set[ArrowProto.SelectionMode.ValueType]:
+) -> set[ArrowProto.SelectionMode.ValueType]:
     """Parse and check the user provided selection modes."""
     if isinstance(selection_mode, str):
         # Only a single selection mode was passed
@@ -238,8 +237,7 @@ class ArrowMixin:
         key: Key | None = None,
         on_select: Literal["ignore"],  # No default value here to make it work with mypy
         selection_mode: SelectionMode | Iterable[SelectionMode] = "multi-row",
-    ) -> DeltaGenerator:
-        ...
+    ) -> DeltaGenerator: ...
 
     @overload
     def dataframe(
@@ -255,8 +253,7 @@ class ArrowMixin:
         key: Key | None = None,
         on_select: Literal["rerun"] | WidgetCallback = "rerun",
         selection_mode: SelectionMode | Iterable[SelectionMode] = "multi-row",
-    ) -> DataframeState:
-        ...
+    ) -> DataframeState: ...
 
     @gather_metrics("dataframe")
     def dataframe(

streamlit/elements/code.py

@@ -19,10 +19,10 @@ from typing import TYPE_CHECKING, cast
 from streamlit.proto.Code_pb2 import Code as CodeProto
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.string_util import clean_text
-from streamlit.type_util import SupportsStr
 
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
+    from streamlit.type_util import SupportsStr
 
 
 class CodeMixin:

streamlit/elements/dialog_decorator.py

@@ -15,14 +15,16 @@
 from __future__ import annotations
 
 from functools import wraps
-from typing import Callable, TypeVar, cast, overload
+from typing import TYPE_CHECKING, Callable, TypeVar, cast, overload
 
 from streamlit.delta_generator import event_dg, get_last_dg_added_to_context_stack
-from streamlit.elements.lib.dialog import DialogWidth
 from streamlit.errors import StreamlitAPIException
 from streamlit.runtime.fragment import _fragment
 from streamlit.runtime.metrics_util import gather_metrics
 
+if TYPE_CHECKING:
+    from streamlit.elements.lib.dialog import DialogWidth
+
 
 def _assert_no_nested_dialogs() -> None:
     """Check the current stack for existing DeltaGenerator's of type 'dialog'.
@@ -79,8 +81,9 @@ def _dialog_decorator(
 
 
 @overload
-def dialog_decorator(title: str, *, width: DialogWidth = "small") -> Callable[[F], F]:
-    ...
+def dialog_decorator(
+    title: str, *, width: DialogWidth = "small"
+) -> Callable[[F], F]: ...
 
 
 # 'title' can be a function since `dialog_decorator` is a decorator. We just call it 'title' here though
@@ -88,8 +91,7 @@ def dialog_decorator(title: str, *, width: DialogWidth = "small") -> Callable[[F
 # The user is supposed to call it like @st.dialog("my_title") , which makes 'title' a positional arg, hence
 # this 'trick'. The overload is required to have a good type hint for the decorated function args.
 @overload
-def dialog_decorator(title: F, *, width: DialogWidth = "small") -> F:
-    ...
+def dialog_decorator(title: F, *, width: DialogWidth = "small") -> F: ...
 
 
 @gather_metrics("experimental_dialog")
@@ -175,7 +177,7 @@ def dialog_decorator(
     """
 
     func_or_title = title
-    if type(func_or_title) is str:
+    if isinstance(func_or_title, str):
         # Support passing the params via function decorator
         def wrapper(f: F) -> F:
             title: str = func_or_title
@@ -183,5 +185,5 @@ def dialog_decorator(
 
         return wrapper
 
-    func: F = cast(F, func_or_title)
+    func: F = func_or_title
     return _dialog_decorator(func, "", width=width)

streamlit/elements/doc_string.py

@@ -375,7 +375,7 @@ def _is_stcommand(tree, command_name):
     """Checks whether the AST in tree is a call for command_name."""
     root_node = tree.body[0].value
 
-    if not type(root_node) is ast.Call:
+    if not isinstance(root_node, ast.Call):
         return False
 
     return (

streamlit/elements/exception.py

@@ -103,7 +103,7 @@ def marshall(exception_proto: ExceptionProto, exception: BaseException) -> None:
     # Some exceptions (like UserHashError) have an alternate_name attribute so
     # we can pretend to the user that the exception is called something else.
     if getattr(exception, "alternate_name", None) is not None:
-        exception_proto.type = getattr(exception, "alternate_name")
+        exception_proto.type = exception.alternate_name  # type: ignore[attr-defined]
     else:
         exception_proto.type = type(exception).__name__
 

streamlit/elements/form.py

@@ -21,10 +21,10 @@ from streamlit.errors import StreamlitAPIException
 from streamlit.proto import Block_pb2
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner import ScriptRunContext, get_script_run_ctx
-from streamlit.runtime.state import WidgetArgs, WidgetCallback, WidgetKwargs
 
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
+    from streamlit.runtime.state import WidgetArgs, WidgetCallback, WidgetKwargs
 
 
 class FormData(NamedTuple):

streamlit/elements/heading.py

@@ -23,10 +23,10 @@ from streamlit.errors import StreamlitAPIException
 from streamlit.proto.Heading_pb2 import Heading as HeadingProto
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.string_util import clean_text
-from streamlit.type_util import SupportsStr
 
 if TYPE_CHECKING:
     from streamlit.delta_generator import DeltaGenerator
+    from streamlit.type_util import SupportsStr
 
 
 class HeadingProtoTag(Enum):

streamlit/elements/html.py

@@ -75,7 +75,7 @@ class HtmlMixin:
         html_proto = HtmlProto()
         # Check if the body is a file path
         if os.path.isfile(body):
-            with open(body, "r", encoding="utf-8") as f:
+            with open(body, encoding="utf-8") as f:
                 html_proto.body = f.read()
         else:
             html_proto.body = clean_text(body)