chalkpy 2.96.1__py3-none-any.whl → 2.96.3__py3-none-any.whl

chalk/_version.py

@@ -1 +1 @@
-__version__ = "2.96.1"
+__version__ = "2.96.3"

chalk/functions/__init__.py

@@ -4211,6 +4211,53 @@ def contains(arr: Underscore | list[Any] | set[Any], value: Any):
     return UnderscoreFunction("contains", arr, value)
 
 
+def is_in(value: Underscore | Any, arr: Underscore | list[Any] | set[Any]):
+    """
+    Returns whether the value is in the array.
+
+    Parameters
+    ----------
+    value
+        The value to check for in the array.
+    arr
+        The array to search for the value.
+
+    Returns
+    -------
+    Boolean indicating whether the value is present in the array.
+
+    Examples
+    --------
+    >>> import chalk.functions as F
+    >>> from chalk.features import _, features
+    >>> @features
+    ... class User:
+    ...    id: str
+    ...    role: str
+    ...    is_admin: bool = F.is_in(_.role, ["admin", "superuser"])
+
+    >>> # With dynamic array
+    >>> @features
+    ... class APIRequest:
+    ...    id: str
+    ...    user_id: str
+    ...    allowed_users: list[str]
+    ...    is_allowed: bool = F.is_in(_.user_id, _.allowed_users)
+    """
+    if isinstance(arr, Underscore):
+        return UnderscoreFunction("contains", arr, value)
+    if len(arr) < 1:
+        raise ValueError("Expected input array to have at least one element")
+    try:
+        map_const = pa.scalar(
+            {elem: None for elem in arr},
+            type=pa.map_(rich_to_pyarrow(type(next(iter(arr))), name="arr_type"), pa.null()),
+        )
+    except pa.ArrowInvalid as ai:
+        raise ValueError("Expected constant input array to have entries all of the same type") from ai
+    return UnderscoreFunction("map_key_exists", map_const, value)
+
+
 def cardinality(arr: Underscore):
     """
     Returns the number of elements in an array.

chalk/sql/_internal/integrations/snowflake.py

@@ -523,6 +523,9 @@ class SnowflakeSourceImpl(BaseSQLSource):
                             unload_job_identifier=job_id,
                             snowflake_unload_stage=query_execution_parameters.snowflake.snowflake_unload_stage,
                         )
+                    if env_var_bool("CHALK_SNOWFLAKE_TIMEZONE_UTC"):
+                        chalk_logger.info(f"Setting Snowflake timezone to UTC")
+                        cursor.execute(f"ALTER SESSION SET TIMEZONE = 'UTC';")
 
                     # add the query to the QueryRegistry so we can clean up on shutdown
                     chalk_logger.info(f"Compiled query: {repr(sql)}")
@@ -692,6 +695,10 @@ class SnowflakeSourceImpl(BaseSQLSource):
                         )
                     )
                 with con.cursor() as cursor:
+                    if env_var_bool("CHALK_SNOWFLAKE_TIMEZONE_UTC"):
+                        chalk_logger.info(f"Setting Snowflake timezone to UTC")
+                        cursor.execute(f"ALTER SESSION SET TIMEZONE = 'UTC';")
+
                     job_id = str(uuid.uuid4())
                     if query_execution_parameters.snowflake.snowflake_unload_stage is not None:
                         sql = _rewrite_query_for_unload(

{chalkpy-2.96.1.dist-info → chalkpy-2.96.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalkpy
-Version: 2.96.1
+Version: 2.96.3
 Summary: Python SDK for Chalk
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai

{chalkpy-2.96.1.dist-info → chalkpy-2.96.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
-chalk/_version.py,sha256=53oisDKLL-9FrciuE997yxhzB_eVulphF1oc5zt1rD8,23
+chalk/_version.py,sha256=XJHkYZYgBkcwjlprd1G1-3H4jEGG2wU1InKh3ZqOgZo,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -685,7 +685,7 @@ chalk/features/dataframe/__init__.py,sha256=h88XglttpuUBt0TDoltaxp-90jNtIVxZhKAp
 chalk/features/dataframe/_filters.py,sha256=F9c5wExtmO21zhn4HEWcM9t4qSpfIO9Z7ysvoeZRV0I,20034
 chalk/features/dataframe/_impl.py,sha256=6zrwip7jL4DjNDEzmXw3PdKzZ-RJZTFwafqBUdK_Toc,88131
 chalk/features/dataframe/_validation.py,sha256=PsseXqoTgmgqaYtr-sTJP0CzlTo2v2PezkF0A8Za-oo,11037
-chalk/functions/__init__.py,sha256=k3-saPn3SFVND55UabjGt2sYtJ32ZY9LQx5lfNIkTcM,167789
+chalk/functions/__init__.py,sha256=5Y2pFjCZjkPyfP8ygdknY5_HI9Yk-aob-4n2AgtSlyA,169215
 chalk/functions/holidays.py,sha256=7yhLjH4E1nDNGA0cXnvRQl5_h_W6rFZfIGYEBa-WiM8,3444
 chalk/functions/http.py,sha256=f3UpokWnjLyo0A_zZoSDgfGgRWmLu639FkREi6p82b4,6653
 chalk/functions/proto.py,sha256=RpM4GIj0-hVTA8RcK_k_fQOz9hvidgFtAbtcmD5Y4K0,5749
@@ -766,7 +766,7 @@ chalk/sql/_internal/integrations/mysql.py,sha256=RjIc0TaQceZrZ-q5AIGExbH5VHirbsc
 chalk/sql/_internal/integrations/postgres.py,sha256=bwxwEeJYH5-A7S22YumukwX6aN6c_B_MOOnrmJuTZyI,29169
 chalk/sql/_internal/integrations/redshift.py,sha256=7HDF6FaiusiPgk00kFXttIkowGNbuSsjO0sXxPwWw68,34119
 chalk/sql/_internal/integrations/redshift_compiler_overrides.py,sha256=eKFeaCamTVfoHhdiBv1_3A6CxvFrv86Ovsa-vBBqjEo,5343
-chalk/sql/_internal/integrations/snowflake.py,sha256=Y8kKSA3W02yxi144KSOeKtlud4ArsjLKNPvTG6XkkXI,35241
+chalk/sql/_internal/integrations/snowflake.py,sha256=hiR1nNTr1u_9gTqt_VFO25Ec5-lfcVUcibAL4OjraLE,35698
 chalk/sql/_internal/integrations/snowflake_compiler_overrides.py,sha256=GbD3rdFWMpbht8dE-h9kcSsxideYHvVTGOYIfrczJJ8,6712
 chalk/sql/_internal/integrations/spanner.py,sha256=OiIbffNBj8rZ8VHTy8LgXn6f1b1j86jxIkV6xATjF14,10069
 chalk/sql/_internal/integrations/sqlite.py,sha256=aaxvnkKBh96fxsm-8x-2fqSZUYXvj3aP0NyNIrs0lik,6002
@@ -823,8 +823,8 @@ chalk/utils/tracing.py,sha256=NiiM-9dbuJhSCv6R1npR1uYNKWlkqTR6Ygm0Voi2NrY,13078
 chalk/utils/weak_set_by_identity.py,sha256=VmikA_laYwFeOphCwXJIuyOIkrdlQe0bSzaXq7onoQw,953
 chalk/utils/pydanticutil/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chalk/utils/pydanticutil/pydantic_compat.py,sha256=O575lLYJ5GvZC4HMzR9yATxf9XwjC6NrDUXbNwZidlE,3031
-chalkpy-2.96.1.dist-info/METADATA,sha256=Yju64p2JSKrDw5doqpVwJk2-ePFHdTKDN6oKULBj3As,27754
-chalkpy-2.96.1.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-chalkpy-2.96.1.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
-chalkpy-2.96.1.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
-chalkpy-2.96.1.dist-info/RECORD,,
+chalkpy-2.96.3.dist-info/METADATA,sha256=f67CfiifytFHapQXl81lOur88VRBpbWj35VhMQDOnUc,27754
+chalkpy-2.96.3.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+chalkpy-2.96.3.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.96.3.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.96.3.dist-info/RECORD,,