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

chalk/_version.py

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

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.2.dist-info}/METADATA

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

{chalkpy-2.96.1.dist-info → chalkpy-2.96.2.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=HpL6vv36-wwmrNNpEYb3KnuPyL47UtSutBVx1OH2Fu8,23
 chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
 chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
 chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -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.2.dist-info/METADATA,sha256=sjxhJOD06KjA88uVl_3obnZJti5PY9qP8joRem129p0,27754
+chalkpy-2.96.2.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+chalkpy-2.96.2.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
+chalkpy-2.96.2.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
+chalkpy-2.96.2.dist-info/RECORD,,