sqlmesh 0.181.1.dev3__py3-none-any.whl → 0.181.1.dev5__py3-none-any.whl

sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.181.1.dev3'
-__version_tuple__ = version_tuple = (0, 181, 1, 'dev3')
+__version__ = version = '0.181.1.dev5'
+__version_tuple__ = version_tuple = (0, 181, 1, 'dev5')

sqlmesh/core/engine_adapter/spark.py

@@ -105,6 +105,7 @@ class SparkEngineAdapter(
             exp.DataType.Type.BINARY: spark_types.BinaryType,
             exp.DataType.Type.BOOLEAN: spark_types.BooleanType,
             exp.DataType.Type.DATE: spark_types.DateType,
+            exp.DataType.Type.TIMESTAMPNTZ: spark_types.TimestampNTZType,
             exp.DataType.Type.DATETIME: spark_types.TimestampNTZType,
             exp.DataType.Type.TIMESTAMPLTZ: spark_types.TimestampType,
             exp.DataType.Type.TIMESTAMP: spark_types.TimestampType,

sqlmesh/lsp/main.py

@@ -15,8 +15,7 @@ from sqlmesh.lsp.completions import get_sql_completions
 from sqlmesh.lsp.context import LSPContext, ModelTarget
 from sqlmesh.lsp.custom import ALL_MODELS_FEATURE, AllModelsRequest, AllModelsResponse
 from sqlmesh.lsp.reference import (
-    get_model_definitions_for_a_path,
-    filter_references_by_position,
+    get_references,
 )
 
 
@@ -178,6 +177,34 @@ class SQLMeshLanguageServer:
                 ls.show_message(f"Error formatting SQL: {e}", types.MessageType.Error)
                 return []
 
+        @self.server.feature(types.TEXT_DOCUMENT_HOVER)
+        def hover(ls: LanguageServer, params: types.HoverParams) -> t.Optional[types.Hover]:
+            """Provide hover information for an object."""
+            try:
+                self._ensure_context_for_document(params.text_document.uri)
+                document = ls.workspace.get_document(params.text_document.uri)
+                if self.lsp_context is None:
+                    raise RuntimeError(f"No context found for document: {document.path}")
+
+                references = get_references(
+                    self.lsp_context, params.text_document.uri, params.position
+                )
+                if not references:
+                    return None
+                reference = references[0]
+                if not reference.description:
+                    return None
+                return types.Hover(
+                    contents=types.MarkupContent(
+                        kind=types.MarkupKind.Markdown, value=reference.description
+                    ),
+                    range=reference.range,
+                )
+
+            except Exception as e:
+                ls.show_message(f"Error getting hover information: {e}", types.MessageType.Error)
+                return None
+
         @self.server.feature(types.TEXT_DOCUMENT_DEFINITION)
         def goto_definition(
             ls: LanguageServer, params: types.DefinitionParams
@@ -189,10 +216,9 @@ class SQLMeshLanguageServer:
                 if self.lsp_context is None:
                     raise RuntimeError(f"No context found for document: {document.path}")
 
-                references = get_model_definitions_for_a_path(
-                    self.lsp_context, params.text_document.uri
+                references = get_references(
+                    self.lsp_context, params.text_document.uri, params.position
                 )
-                filtered_references = filter_references_by_position(references, params.position)
                 return [
                     types.LocationLink(
                         target_uri=reference.uri,
@@ -206,9 +232,8 @@ class SQLMeshLanguageServer:
                         ),
                         origin_selection_range=reference.range,
                     )
-                    for reference in filtered_references
+                    for reference in references
                 ]
-
             except Exception as e:
                 ls.show_message(f"Error getting references: {e}", types.MessageType.Error)
                 return []

sqlmesh/lsp/reference.py

@@ -10,40 +10,64 @@ from sqlmesh.utils.pydantic import PydanticModel
 
 
 class Reference(PydanticModel):
+    """
+    A reference to a model.
+
+    Attributes:
+        range: The range of the reference in the source file
+        uri: The uri of the referenced model
+        description: The description of the referenced model
+    """
+
     range: Range
     uri: str
+    description: t.Optional[str] = None
 
 
-def filter_references_by_position(
-    references: t.List[Reference], position: Position
-) -> t.List[Reference]:
+def by_position(position: Position) -> t.Callable[[Reference], bool]:
     """
-    Filter references to only include those that contain the given position.
+    Filter reference to only filter references that contain the given position.
 
     Args:
-        references: List of Reference objects
         position: The cursor position to check
 
     Returns:
-        List of Reference objects that contain the position
+        A function that returns True if the reference contains the position, False otherwise
     """
-    filtered_references = []
-
-    for reference in references:
-        # Check if position is within the reference range
-        range_start = reference.range.start
-        range_end = reference.range.end
 
-        # Position is within range if it's after or at start and before or at end
-        if (
-            range_start.line < position.line
-            or (range_start.line == position.line and range_start.character <= position.character)
+    def contains_position(r: Reference) -> bool:
+        return (
+            r.range.start.line < position.line
+            or (
+                r.range.start.line == position.line
+                and r.range.start.character <= position.character
+            )
         ) and (
-            range_end.line > position.line
-            or (range_end.line == position.line and range_end.character >= position.character)
-        ):
-            filtered_references.append(reference)
+            r.range.end.line > position.line
+            or (r.range.end.line == position.line and r.range.end.character >= position.character)
+        )
+
+    return contains_position
 
+
+def get_references(
+    lint_context: LSPContext, document_uri: str, position: Position
+) -> t.List[Reference]:
+    """
+    Get references at a specific position in a document.
+
+    Used for hover information.
+
+    Args:
+        lint_context: The LSP context
+        document_uri: The URI of the document
+        position: The position to check for references
+
+    Returns:
+        A list of references at the given position
+    """
+    references = get_model_definitions_for_a_path(lint_context, document_uri)
+    filtered_references = list(filter(by_position(position), references))
     return filtered_references
 
 
@@ -154,7 +178,11 @@ def get_model_definitions_for_a_path(
             start_pos = catalog_or_db_range.start
 
         references.append(
-            Reference(uri=referenced_model_uri, range=Range(start=start_pos, end=end_pos))
+            Reference(
+                uri=referenced_model_uri,
+                range=Range(start=start_pos, end=end_pos),
+                description=referenced_model.description,
+            )
         )
 
     return references

{sqlmesh-0.181.1.dev3.dist-info → sqlmesh-0.181.1.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlmesh
-Version: 0.181.1.dev3
+Version: 0.181.1.dev5
 Summary: Next-generation data transformation framework
 Author-email: "TobikoData Inc." <engineering@tobikodata.com>
 License:                                  Apache License

{sqlmesh-0.181.1.dev3.dist-info → sqlmesh-0.181.1.dev5.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlmesh/__init__.py,sha256=uPXld_xTZ2NTruSfVJ_vR_w-uAxR3T-gMXAb-EphTe8,7167
-sqlmesh/_version.py,sha256=UmyRudwujStwkg8fCiTZPPqrQz-uxV47SXbuZcuhJOI,528
+sqlmesh/_version.py,sha256=V8bpUP0oeVbvVW84PK7nOlJ-2zYX3cciqF_2AM_17dM,528
 sqlmesh/magics.py,sha256=7BU9ec9zE4tjvZ3hdEqwFX2VVV-CI0h06TGZo_Ssi8M,39070
 sqlmesh/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/cicd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -73,7 +73,7 @@ sqlmesh/core/engine_adapter/redshift.py,sha256=_rZWv4rchDAzB6m6idoV1NLHFBKK_gM4n
 sqlmesh/core/engine_adapter/risingwave.py,sha256=NXXuiPfbB5ZSp7igMqZpaX2c-5TYreOlIouPU7iCVJ8,936
 sqlmesh/core/engine_adapter/shared.py,sha256=NyF2osSJDWPp7KlJUBR4GZcS_cJFr4daUPvhGyp4XcQ,12909
 sqlmesh/core/engine_adapter/snowflake.py,sha256=LKiQq43JNukLcDq5OnKALpiZ0xjFK7fVJYJvlXHAWxo,23361
-sqlmesh/core/engine_adapter/spark.py,sha256=0wTMRs3emclwSThp05qvWPobu-Zor9Oc4pDCGMwQlZg,22310
+sqlmesh/core/engine_adapter/spark.py,sha256=VOXStkBBZIzebPo_ivdQFvZtRMiWJHHo7h118Lh5Yt8,22384
 sqlmesh/core/engine_adapter/trino.py,sha256=pZ2E1owyC6DPmkU3RkibR3CIE1gE-8fNd016YHh52II,14915
 sqlmesh/core/linter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/core/linter/definition.py,sha256=UL05OjFBfoLSYGWXNf4zjZYzIKr-OyFIMIcIh7RhR8s,5237
@@ -158,8 +158,8 @@ sqlmesh/integrations/github/cicd/controller.py,sha256=AiEMsRW4WSVxXP_7oQSIdzBzmY
 sqlmesh/lsp/completions.py,sha256=ScW0H9couhfW6QckEX2djVYTClRPXpT2Hdcg1qLfUUY,3479
 sqlmesh/lsp/context.py,sha256=vUpbWYjKYuwDvU3zkzhAgw9An-TeVYe_5KTk-ec_UwM,1569
 sqlmesh/lsp/custom.py,sha256=GvNAiW7imPI8xmIx3qLWN5FljGPHFJKh0N5b6aZnwRo,489
-sqlmesh/lsp/main.py,sha256=Xt7cwoocmmefsFO4UvBPlIsiXWs7YEGRp6jgT2tKn60,13672
-sqlmesh/lsp/reference.py,sha256=veet4JZpVmUOIMiRPOvMtzu4DQEPT9Vk9c2bWAsFuQA,7366
+sqlmesh/lsp/main.py,sha256=OftpChYMN1_sNeaLM5wxK5umiHCiOwPNnFFpaG6sAu8,14774
+sqlmesh/lsp/reference.py,sha256=2rANWlO4Z4Pt9KuUhZSWoaTdPfmeNVDgi7iiXFTb7zY,8076
 sqlmesh/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/migrations/v0001_init.py,sha256=h8FwMYJj9_O8yGpk12guaSaGwX5A3we2GiKeHrXi15o,2048
 sqlmesh/migrations/v0002_remove_identify.py,sha256=ZeHuodZcQzRRZX0B0Y7SSkK3qqlDuz316kmvzojeyoU,113
@@ -265,7 +265,7 @@ sqlmesh/utils/pandas.py,sha256=j_tYhuPcWpzuPceZZ6IhdTVbmxYyGJW7l4FHr4bQqjM,1824
 sqlmesh/utils/pydantic.py,sha256=ioaAzYF42PutEJRyAS11-AHguiCG3yWI6VRcTVa2G-g,10470
 sqlmesh/utils/rich.py,sha256=xNZyPCqEc4TAlJlsEthg-5eYUWpaizz0QjBzO7a1Yv4,1509
 sqlmesh/utils/yaml.py,sha256=np-WVcwmwsKghCgAj1OBBMsK1SXSRn4wR-H37cGiFr8,2282
-sqlmesh-0.181.1.dev3.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
+sqlmesh-0.181.1.dev5.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
 web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 web/client/dist/index.html,sha256=cQRIlDgkl6OgeWuHI9hM8bJhKUE9-Ji2NpQa2hlgNM8,5472
 web/client/dist/assets/Audits-4U9LnWyN.js,sha256=reXu-sKJ76kiaAZIGkCPvgO68Km9tj1PN_9mx6X6WAY,997
@@ -382,8 +382,8 @@ web/server/api/endpoints/models.py,sha256=SzS5bshzdXyM6hb948sZ_3k-xhpGCHkxkb7pHz
 web/server/api/endpoints/modules.py,sha256=8hqqgonGay_mJmpCw0IdbjsPhWlQH2VLdKAqha-myac,468
 web/server/api/endpoints/plan.py,sha256=OYuQ_4dgbhG9ibTcX1a8mvYgH3ANxw9Fhbb4wg6QWqc,9222
 web/server/api/endpoints/table_diff.py,sha256=Wd5YxeKbZB7AkrDvqFgvV9ZQlnKYOF1n5W_w2EHE_Zw,2085
-sqlmesh-0.181.1.dev3.dist-info/METADATA,sha256=l36AoWMTkn-01VKUdqHoDQDnrgOqRuaB8_RQ6WXFDhE,25307
-sqlmesh-0.181.1.dev3.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
-sqlmesh-0.181.1.dev3.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
-sqlmesh-0.181.1.dev3.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
-sqlmesh-0.181.1.dev3.dist-info/RECORD,,
+sqlmesh-0.181.1.dev5.dist-info/METADATA,sha256=4zzJW4BjZ10b_SHIa919RjR9Auj8Wfi-V2P402pQ-0c,25307
+sqlmesh-0.181.1.dev5.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+sqlmesh-0.181.1.dev5.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
+sqlmesh-0.181.1.dev5.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
+sqlmesh-0.181.1.dev5.dist-info/RECORD,,