streamlit-nightly 1.31.2.dev20240213__py2.py3-none-any.whl → 1.31.2.dev20240214__py2.py3-none-any.whl

streamlit/case_converters.py

@@ -12,10 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import re
+from typing import Any, Callable
 
 
-def to_upper_camel_case(snake_case_str):
+def to_upper_camel_case(snake_case_str: str) -> str:
     """Converts snake_case to UpperCamelCase.
 
     Example
@@ -26,7 +29,7 @@ def to_upper_camel_case(snake_case_str):
     return "".join(map(str.title, snake_case_str.split("_")))
 
 
-def to_lower_camel_case(snake_case_str):
+def to_lower_camel_case(snake_case_str: str) -> str:
     """Converts snake_case to lowerCamelCase.
 
     Example
@@ -44,7 +47,7 @@ def to_lower_camel_case(snake_case_str):
         return snake_case_str
 
 
-def to_snake_case(camel_case_str):
+def to_snake_case(camel_case_str: str) -> str:
     """Converts UpperCamelCase and lowerCamelCase to snake_case.
 
     Examples
@@ -57,7 +60,9 @@ def to_snake_case(camel_case_str):
     return re.sub("([a-z0-9])([A-Z])", r"\1_\2", s1).lower()
 
 
-def convert_dict_keys(func, in_dict):
+def convert_dict_keys(
+    func: Callable[[str], str], in_dict: dict[Any, Any]
+) -> dict[Any, Any]:
     """Apply a conversion function to all keys in a dict.
 
     Parameters

streamlit/cli_util.py

@@ -14,6 +14,8 @@
 
 """ Utilities related to the CLI."""
 
+from __future__ import annotations
+
 
 def print_to_cli(message: str, **kwargs) -> None:
     """Print a message to the terminal using click if available, else print

streamlit/code_util.py

@@ -14,10 +14,13 @@
 
 """A bunch of useful code utilities."""
 
+from __future__ import annotations
+
 import re
+from typing import Any
 
 
-def extract_args(line):
+def extract_args(line: str) -> list[str]:
     """Parse argument strings from all outer parentheses in a line of code.
 
     Parameters
@@ -53,7 +56,7 @@ def extract_args(line):
     return results
 
 
-def get_method_args_from_code(args, line):
+def get_method_args_from_code(args: list[Any], line: str) -> list[str]:
     """Parse arguments from a stringified arguments list inside parentheses
 
     Parameters

streamlit/color_util.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 from typing import Any, Callable, Collection, Tuple, Union, cast
 
 from typing_extensions import TypeAlias

streamlit/column_config.py

@@ -14,6 +14,8 @@
 
 """Column types that can be configured via the ``column_config`` parameter of ``st.dataframe`` and ``st.data_editor``."""
 
+from __future__ import annotations
+
 __all__ = [
     "Column",
     "TextColumn",

streamlit/commands/execution_control.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import os
-from typing import NoReturn
+from typing import Final, NoReturn
 
 import streamlit as st
 from streamlit import source_util
@@ -24,7 +26,7 @@ from streamlit.logger import get_logger
 from streamlit.runtime.metrics_util import gather_metrics
 from streamlit.runtime.scriptrunner import RerunData, get_script_run_ctx
 
-_LOGGER = get_logger(__name__)
+_LOGGER: Final = get_logger(__name__)
 
 
 @gather_metrics("stop")

streamlit/commands/experimental_query_params.py

@@ -12,8 +12,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import urllib.parse as parse
-from typing import Any, Dict, List, Union
+from typing import Any
 
 from streamlit import util
 from streamlit.constants import (
@@ -28,7 +30,7 @@ from streamlit.runtime.scriptrunner import get_script_run_ctx
 
 
 @gather_metrics("experimental_get_query_params")
-def get_query_params() -> Dict[str, List[str]]:
+def get_query_params() -> dict[str, list[str]]:
     """Return the query parameters that is currently showing in the browser's URL bar.
 
     Returns
@@ -106,10 +108,11 @@ def set_query_params(**query_params: Any) -> None:
 
 
 def _ensure_no_embed_params(
-    query_params: Dict[str, Union[List[str], str]], query_string: str
+    query_params: dict[str, list[str] | str], query_string: str
 ) -> str:
     """Ensures there are no embed params set (raises StreamlitAPIException) if there is a try,
-    also makes sure old param values in query_string are preserved. Returns query_string : str."""
+    also makes sure old param values in query_string are preserved. Returns query_string : str.
+    """
     # Get query params dict without embed, embed_options params
     query_params_without_embed = util.exclude_keys_in_dict(
         query_params, keys_to_exclude=EMBED_QUERY_PARAMS_KEYS

streamlit/commands/page_config.py

@@ -12,11 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import random
 from textwrap import dedent
-from typing import TYPE_CHECKING, Mapping, Optional, Union, cast
+from typing import TYPE_CHECKING, Final, Literal, Mapping, Union, cast
 
-from typing_extensions import Final, Literal, TypeAlias
+from typing_extensions import TypeAlias
 
 from streamlit.elements import image
 from streamlit.errors import StreamlitAPIException
@@ -42,7 +44,7 @@ _GetHelp: TypeAlias = Literal["Get help", "Get Help", "get help"]
 _ReportABug: TypeAlias = Literal["Report a bug", "report a bug"]
 _About: TypeAlias = Literal["About", "about"]
 MenuKey: TypeAlias = Literal[_GetHelp, _ReportABug, _About]
-MenuItems: TypeAlias = Mapping[MenuKey, Optional[str]]
+MenuItems: TypeAlias = Mapping[MenuKey, Union[str, None]]
 
 # Emojis recommended by https://share.streamlit.io/rensdimmendaal/emoji-recommender/main/app/streamlit.py
 # for the term "streamlit". Watch out for zero-width joiners,
@@ -113,11 +115,11 @@ def _get_favicon_string(page_icon: PageIcon) -> str:
 
 @gather_metrics("set_page_config")
 def set_page_config(
-    page_title: Optional[str] = None,
-    page_icon: Optional[PageIcon] = None,
+    page_title: str | None = None,
+    page_icon: PageIcon | None = None,
     layout: Layout = "centered",
     initial_sidebar_state: InitialSideBarState = "auto",
-    menu_items: Optional[MenuItems] = None,
+    menu_items: MenuItems | None = None,
 ) -> None:
     """
     Configures the default settings of the page.
@@ -188,7 +190,7 @@ def set_page_config(
     if page_icon is not None:
         msg.page_config_changed.favicon = _get_favicon_string(page_icon)
 
-    pb_layout: "PageConfigProto.Layout.ValueType"
+    pb_layout: PageConfigProto.Layout.ValueType
     if layout == "centered":
         pb_layout = PageConfigProto.CENTERED
     elif layout == "wide":
@@ -199,7 +201,7 @@ def set_page_config(
         )
     msg.page_config_changed.layout = pb_layout
 
-    pb_sidebar_state: "PageConfigProto.SidebarState.ValueType"
+    pb_sidebar_state: PageConfigProto.SidebarState.ValueType
     if initial_sidebar_state == "auto":
         pb_sidebar_state = PageConfigProto.AUTO
     elif initial_sidebar_state == "expanded":
@@ -265,5 +267,5 @@ def validate_menu_items(menu_items: MenuItems) -> None:
             raise StreamlitAPIException(f'"{v}" is a not a valid URL!')
 
 
-def valid_menu_item_key(key: str) -> "TypeGuard[MenuKey]":
+def valid_menu_item_key(key: str) -> TypeGuard[MenuKey]:
     return key in {GET_HELP_KEY, REPORT_A_BUG_KEY, ABOUT_KEY}

streamlit/components/v1/components.py

@@ -12,11 +12,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import inspect
 import json
 import os
 import threading
-from typing import Any, Dict, Optional, Type, Union
+from typing import TYPE_CHECKING, Any, Final
 
 import streamlit
 from streamlit import type_util, util
@@ -32,7 +34,10 @@ from streamlit.runtime.state import NoValue, register_widget
 from streamlit.runtime.state.common import compute_widget_id
 from streamlit.type_util import to_bytes
 
-LOGGER = get_logger(__name__)
+if TYPE_CHECKING:
+    from streamlit.delta_generator import DeltaGenerator
+
+_LOGGER: Final = get_logger(__name__)
 
 
 class MarshallComponentException(StreamlitAPIException):
@@ -47,8 +52,8 @@ class CustomComponent:
     def __init__(
         self,
         name: str,
-        path: Optional[str] = None,
-        url: Optional[str] = None,
+        path: str | None = None,
+        url: str | None = None,
     ):
         if (path is None and url is None) or (path is not None and url is not None):
             raise StreamlitAPIException(
@@ -63,7 +68,7 @@ class CustomComponent:
         return util.repr_(self)
 
     @property
-    def abspath(self) -> Optional[str]:
+    def abspath(self) -> str | None:
         """The absolute path that the component is served from."""
         if self.path is None:
             return None
@@ -73,7 +78,7 @@ class CustomComponent:
         self,
         *args,
         default: Any = None,
-        key: Optional[str] = None,
+        key: str | None = None,
         **kwargs,
     ) -> Any:
         """An alias for create_instance."""
@@ -84,7 +89,7 @@ class CustomComponent:
         self,
         *args,
         default: Any = None,
-        key: Optional[str] = None,
+        key: str | None = None,
         **kwargs,
     ) -> Any:
         """Create a new instance of the component.
@@ -155,7 +160,9 @@ And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
                 "Could not convert component args to JSON", ex
             )
 
-        def marshall_component(dg, element: Element) -> Union[Any, Type[NoValue]]:
+        def marshall_component(
+            dg: DeltaGenerator, element: Element
+        ) -> Any | type[NoValue]:
             element.component_instance.component_name = self.name
             element.component_instance.form_id = current_form_id(dg)
             if self.url is not None:
@@ -264,8 +271,8 @@ And if you're using Streamlit Cloud, add "pyarrow" to your requirements.txt."""
 
 def declare_component(
     name: str,
-    path: Optional[str] = None,
-    url: Optional[str] = None,
+    path: str | None = None,
+    url: str | None = None,
 ) -> CustomComponent:
     """Create and register a custom component.
 
@@ -324,10 +331,10 @@ def declare_component(
 
 class ComponentRegistry:
     _instance_lock: threading.Lock = threading.Lock()
-    _instance: Optional["ComponentRegistry"] = None
+    _instance: ComponentRegistry | None = None
 
     @classmethod
-    def instance(cls) -> "ComponentRegistry":
+    def instance(cls) -> ComponentRegistry:
         """Returns the singleton ComponentRegistry"""
         # We use a double-checked locking optimization to avoid the overhead
         # of acquiring the lock in the common case:
@@ -339,7 +346,7 @@ class ComponentRegistry:
         return cls._instance
 
     def __init__(self):
-        self._components: Dict[str, CustomComponent] = {}
+        self._components: dict[str, CustomComponent] = {}
         self._lock = threading.Lock()
 
     def __repr__(self) -> str:
@@ -364,15 +371,15 @@ class ComponentRegistry:
             self._components[component.name] = component
 
         if existing is not None and component != existing:
-            LOGGER.warning(
+            _LOGGER.warning(
                 "%s overriding previously-registered %s",
                 component,
                 existing,
             )
 
-        LOGGER.debug("Registered component %s", component)
+        _LOGGER.debug("Registered component %s", component)
 
-    def get_component_path(self, name: str) -> Optional[str]:
+    def get_component_path(self, name: str) -> str | None:
         """Return the filesystem path for the component with the given name.
 
         If no such component is registered, or if the component exists but is

streamlit/config.py

@@ -115,9 +115,7 @@ def set_user_option(key: str, value: Any) -> None:
     try:
         opt = _config_options_template[key]
     except KeyError as ke:
-        raise StreamlitAPIException(
-            "Unrecognized config option: {key}".format(key=key)
-        ) from ke
+        raise StreamlitAPIException(f"Unrecognized config option: {key}") from ke
     if opt.scriptable:
         set_option(key, value)
         return
@@ -245,7 +243,7 @@ def _create_option(
     )
     assert (
         option.section in _section_descriptions
-    ), 'Section "%s" must be one of %s.' % (
+    ), 'Section "{}" must be one of {}.'.format(
         option.section,
         ", ".join(_section_descriptions.keys()),
     )
@@ -1310,7 +1308,7 @@ def get_config_options(
             if not os.path.exists(filename):
                 continue
 
-            with open(filename, "r", encoding="utf-8") as input:
+            with open(filename, encoding="utf-8") as input:
                 file_contents = input.read()
 
             _update_config_with_toml(file_contents, filename)

streamlit/config_option.py

@@ -14,14 +14,15 @@
 
 """Class to store a key-value pair for the config system."""
 
+from __future__ import annotations
+
 import datetime
 import re
 import textwrap
-from typing import Any, Callable, Optional
+from typing import Any, Callable
 
 from streamlit import util
 from streamlit.case_converters import to_snake_case
-from streamlit.errors import DeprecationError
 
 
 class ConfigOption:
@@ -97,14 +98,14 @@ class ConfigOption:
     def __init__(
         self,
         key: str,
-        description: Optional[str] = None,
-        default_val: Optional[Any] = None,
+        description: str | None = None,
+        default_val: Any | None = None,
         visibility: str = "visible",
         scriptable: bool = False,
         deprecated: bool = False,
-        deprecation_text: Optional[str] = None,
-        expiration_date: Optional[str] = None,
-        replaced_by: Optional[str] = None,
+        deprecation_text: str | None = None,
+        expiration_date: str | None = None,
+        replaced_by: str | None = None,
         type_: type = str,
         sensitive: bool = False,
     ):
@@ -174,7 +175,7 @@ class ConfigOption:
         self.deprecated = deprecated
         self.replaced_by = replaced_by
         self.is_default = True
-        self._get_val_func: Optional[Callable[[], Any]] = None
+        self._get_val_func: Callable[[], Any] | None = None
         self.where_defined = ConfigOption.DEFAULT_DEFINITION
         self.type = type_
         self.sensitive = sensitive
@@ -195,7 +196,7 @@ class ConfigOption:
     def __repr__(self) -> str:
         return util.repr_(self)
 
-    def __call__(self, get_val_func: Callable[[], Any]) -> "ConfigOption":
+    def __call__(self, get_val_func: Callable[[], Any]) -> ConfigOption:
         """Assign a function to compute the value for this option.
 
         This method is called when ConfigOption is used as a decorator.
@@ -226,7 +227,7 @@ class ConfigOption:
             return None
         return self._get_val_func()
 
-    def set_value(self, value: Any, where_defined: Optional[str] = None) -> None:
+    def set_value(self, value: Any, where_defined: str | None = None) -> None:
         """Set the value of this option.
 
         Parameters
@@ -313,5 +314,5 @@ class ConfigOption:
 
 
 def _parse_yyyymmdd_str(date_str: str) -> datetime.datetime:
-    year, month, day = [int(token) for token in date_str.split("-", 2)]
+    year, month, day = (int(token) for token in date_str.split("-", 2))
     return datetime.datetime(year, month, day)

streamlit/connections/base_connection.py

@@ -12,9 +12,11 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 import json
 from abc import ABC, abstractmethod
-from typing import Any, Generic, Optional, TypeVar
+from typing import Any, Generic, TypeVar
 
 from streamlit.runtime.secrets import AttrDict, secrets_singleton
 from streamlit.util import calc_md5
@@ -68,7 +70,7 @@ class BaseConnection(ABC, Generic[RawConnectionT]):
         self._config_section_hash = calc_md5(json.dumps(self._secrets.to_dict()))
         secrets_singleton.file_change_listener.connect(self._on_secrets_changed)
 
-        self._raw_instance: Optional[RawConnectionT] = self._connect(**kwargs)
+        self._raw_instance: RawConnectionT | None = self._connect(**kwargs)
 
     def __del__(self) -> None:
         secrets_singleton.file_change_listener.disconnect(self._on_secrets_changed)

streamlit/connections/snowflake_connection.py

@@ -49,7 +49,7 @@ class SnowflakeConnection(BaseConnection["InternalSnowflakeConnection"]):
     initialized directly.
     """
 
-    def _connect(self, **kwargs) -> "InternalSnowflakeConnection":
+    def _connect(self, **kwargs) -> InternalSnowflakeConnection:
         import snowflake.connector  # type:ignore[import]
         from snowflake.connector import Error as SnowflakeError  # type:ignore[import]
 
@@ -260,7 +260,7 @@ class SnowflakeConnection(BaseConnection["InternalSnowflakeConnection"]):
 
         return (success, nchunks, nrows)
 
-    def cursor(self) -> "SnowflakeCursor":
+    def cursor(self) -> SnowflakeCursor:
         """Return a PEP 249-compliant cursor object.
 
         For more information, see the `Snowflake Python Connector documentation
@@ -269,7 +269,7 @@ class SnowflakeConnection(BaseConnection["InternalSnowflakeConnection"]):
         return self._instance.cursor()
 
     @property
-    def raw_connection(self) -> "InternalSnowflakeConnection":
+    def raw_connection(self) -> InternalSnowflakeConnection:
         """Access the underlying Snowflake Python connector object.
 
         Information on how to use the Snowflake Python Connector can be found in the
@@ -277,7 +277,7 @@ class SnowflakeConnection(BaseConnection["InternalSnowflakeConnection"]):
         """
         return self._instance
 
-    def session(self) -> "Session":
+    def session(self) -> Session:
         """Create a new Snowpark Session from this connection.
 
         Information on how to use Snowpark sessions can be found in the `Snowpark documentation

streamlit/connections/snowpark_connection.py

@@ -63,7 +63,7 @@ class SnowparkConnection(BaseConnection["Session"]):
         self._lock = threading.RLock()
         super().__init__(connection_name, **kwargs)
 
-    def _connect(self, **kwargs) -> "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,
@@ -165,7 +165,7 @@ class SnowparkConnection(BaseConnection["Session"]):
         return _query(sql)
 
     @property
-    def session(self) -> "Session":
+    def session(self) -> Session:
         """Access the underlying Snowpark session.
 
         .. note::
@@ -188,7 +188,7 @@ class SnowparkConnection(BaseConnection["Session"]):
         return self._instance
 
     @contextmanager
-    def safe_session(self) -> Iterator["Session"]:
+    def safe_session(self) -> Iterator[Session]:
         """Grab the underlying Snowpark session in a thread-safe manner.
 
         As operations on a Snowpark session are not thread safe, we need to take care

streamlit/connections/sql_connection.py

@@ -17,7 +17,7 @@ from __future__ import annotations
 from collections import ChainMap
 from copy import deepcopy
 from datetime import timedelta
-from typing import TYPE_CHECKING, List, cast
+from typing import TYPE_CHECKING, cast
 
 from streamlit.connections import BaseConnection
 from streamlit.connections.util import extract_from_dict
@@ -76,7 +76,7 @@ class SQLConnection(BaseConnection["Engine"]):
     >>> st.dataframe(df)
     """
 
-    def _connect(self, autocommit: bool = False, **kwargs) -> "Engine":
+    def _connect(self, autocommit: bool = False, **kwargs) -> Engine:
         import sqlalchemy
 
         kwargs = deepcopy(kwargs)
@@ -125,7 +125,7 @@ class SQLConnection(BaseConnection["Engine"]):
         *,  # keyword-only arguments:
         show_spinner: bool | str = "Running `sql.query(...)`.",
         ttl: float | int | timedelta | None = None,
-        index_col: str | List[str] | None = None,
+        index_col: str | list[str] | None = None,
         chunksize: int | None = None,
         params=None,
         **kwargs,
@@ -245,7 +245,7 @@ class SQLConnection(BaseConnection["Engine"]):
             **kwargs,
         )
 
-    def connect(self) -> "SQLAlchemyConnection":
+    def connect(self) -> SQLAlchemyConnection:
         """Call ``.connect()`` on the underlying SQLAlchemy Engine, returning a new\
         ``sqlalchemy.engine.Connection`` object.
 
@@ -257,7 +257,7 @@ class SQLConnection(BaseConnection["Engine"]):
         return self._instance.connect()
 
     @property
-    def engine(self) -> "Engine":
+    def engine(self) -> Engine:
         """The underlying SQLAlchemy Engine.
 
         This is equivalent to accessing ``self._instance``.
@@ -273,7 +273,7 @@ class SQLConnection(BaseConnection["Engine"]):
         return self._instance.driver
 
     @property
-    def session(self) -> "Session":
+    def session(self) -> Session:
         """Return a SQLAlchemy Session.
 
         Users of this connection should use the contextmanager pattern for writes,

streamlit/connections/util.py

@@ -18,17 +18,17 @@
 # way to configure this at a per-line level :(
 # mypy: no-warn-unused-ignores
 
+from __future__ import annotations
 
-import configparser
 import os
-from typing import Any, Collection, Dict, cast
+from typing import Any, Collection, cast
 
 SNOWSQL_CONNECTION_FILE = "~/.snowsql/config"
 
 
 def extract_from_dict(
-    keys: Collection[str], source_dict: Dict[str, Any]
-) -> Dict[str, Any]:
+    keys: Collection[str], source_dict: dict[str, Any]
+) -> dict[str, Any]:
     """Extract the specified keys from source_dict and return them in a new dict.
 
     Parameters
@@ -52,12 +52,15 @@ def extract_from_dict(
     return d
 
 
-def load_from_snowsql_config_file(connection_name: str) -> Dict[str, Any]:
+def load_from_snowsql_config_file(connection_name: str) -> dict[str, Any]:
     """Loads the dictionary from snowsql config file."""
     snowsql_config_file = os.path.expanduser(SNOWSQL_CONNECTION_FILE)
     if not os.path.exists(snowsql_config_file):
         return {}
 
+    # Lazy-load config parser for better import / startup performance
+    import configparser
+
     config = configparser.ConfigParser(inline_comment_prefixes="#")
     config.read(snowsql_config_file)
 

streamlit/constants.py

@@ -12,6 +12,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from __future__ import annotations
+
 EMBED_QUERY_PARAM = "embed"
 EMBED_OPTIONS_QUERY_PARAM = "embed_options"
 EMBED_QUERY_PARAMS_KEYS = [EMBED_QUERY_PARAM, EMBED_OPTIONS_QUERY_PARAM]