sqlmesh 0.205.1.dev20__py3-none-any.whl → 0.205.1.dev21__py3-none-any.whl

sqlmesh/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.205.1.dev20'
-__version_tuple__ = version_tuple = (0, 205, 1, 'dev20')
+__version__ = version = '0.205.1.dev21'
+__version_tuple__ = version_tuple = (0, 205, 1, 'dev21')

sqlmesh/core/linter/rules/builtin.py

@@ -7,13 +7,14 @@ import typing as t
 from sqlglot.expressions import Star
 from sqlglot.helper import subclasses
 
+from sqlmesh.core.constants import EXTERNAL_MODELS_YAML
 from sqlmesh.core.dialect import normalize_model_name
 from sqlmesh.core.linter.helpers import (
     TokenPositionDetails,
     get_range_of_model_block,
     read_range_from_string,
 )
-from sqlmesh.core.linter.rule import Rule, RuleViolation, Range, Fix, TextEdit
+from sqlmesh.core.linter.rule import Rule, RuleViolation, Range, Fix, TextEdit, Position
 from sqlmesh.core.linter.definition import RuleSet
 from sqlmesh.core.model import Model, SqlModel, ExternalModel
 from sqlmesh.utils.lineage import extract_references_from_query, ExternalModelReference
@@ -185,12 +186,14 @@ class NoMissingExternalModels(Rule):
         violations = []
         for ref_name, ref in external_references.items():
             if ref_name in not_registered_external_models:
+                fix = self.create_fix(ref_name)
                 violations.append(
                     RuleViolation(
                         rule=self,
                         violation_msg=f"Model '{model.fqn}' depends on unregistered external model '{ref_name}'. "
                         "Please register it in the external models file. This can be done by running 'sqlmesh create_external_models'.",
                         violation_range=ref.range,
+                        fixes=[fix] if fix else [],
                     )
                 )
 
@@ -212,5 +215,46 @@ class NoMissingExternalModels(Rule):
             "Please register them in the external models file. This can be done by running 'sqlmesh create_external_models'.",
         )
 
+    def create_fix(self, model_name: str) -> t.Optional[Fix]:
+        """
+        Add an external model to the external models file.
+        - If no external models file exists, it will create one with the model.
+        - If the model already exists, it will not add it again.
+        """
+        root = self.context.path
+        if not root:
+            return None
+
+        external_models_path = root / EXTERNAL_MODELS_YAML
+        if not external_models_path.exists():
+            return None
+
+        # Figure out the position to insert the new external model at the end of the file, whether
+        # needs new line or not.
+        with open(external_models_path, "r", encoding="utf-8") as file:
+            lines = file.read()
+
+        # If a file ends in newline, we can add the new model directly.
+        split_lines = lines.splitlines()
+        if lines.endswith("\n"):
+            new_text = f"- name: '{model_name}'\n"
+            position = Position(line=len(split_lines), character=0)
+        else:
+            new_text = f"\n- name: '{model_name}'\n"
+            position = Position(
+                line=len(split_lines) - 1, character=len(split_lines[-1]) if split_lines else 0
+            )
+
+        return Fix(
+            title="Add external model",
+            edits=[
+                TextEdit(
+                    path=external_models_path,
+                    range=Range(start=position, end=position),
+                    new_text=new_text,
+                )
+            ],
+        )
+
 
 BUILTIN_RULES = RuleSet(subclasses(__name__, Rule, (Rule,)))

{sqlmesh-0.205.1.dev20.dist-info → sqlmesh-0.205.1.dev21.dist-info}/METADATA

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

{sqlmesh-0.205.1.dev20.dist-info → sqlmesh-0.205.1.dev21.dist-info}/RECORD

@@ -1,5 +1,5 @@
 sqlmesh/__init__.py,sha256=QjwaamJIdW7voXoS880yAx7tHm-Im4_mqGA-XJgafos,7631
-sqlmesh/_version.py,sha256=GkqA44FmLbBN3arn4mtDTndnuKRKMu7IjVw8OfFWe8k,530
+sqlmesh/_version.py,sha256=uvArl7UUkT0NZv1Ks5i0VNvGFS9GXml10bM7Ox2RGTs,530
 sqlmesh/magics.py,sha256=bEzYqXHcM0P3CjyUZ1O9arlmXAdLAhoglKbQ29fD9AE,41842
 sqlmesh/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlmesh/cicd/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -80,7 +80,7 @@ sqlmesh/core/linter/definition.py,sha256=JDJTkhByUw4Tr3nF-BOHKFAx_eASQqEqzX6AUVQ
 sqlmesh/core/linter/helpers.py,sha256=HNNSiuKB2o4Z2AGbwgrGr9hE0lQnIR7VHQfj8D8M_Jw,5202
 sqlmesh/core/linter/rule.py,sha256=MgFwfvlGEvh4nijsxc7Bbk_S9ctJ_m-nKCom9LIN4W4,3734
 sqlmesh/core/linter/rules/__init__.py,sha256=gevzfb67vFqckTCoVAe_TBGf6hQ-YtE1_YuGuXyh1L0,77
-sqlmesh/core/linter/rules/builtin.py,sha256=RmKZrp3EWLSmP-X_gT2gl_SVmIImPoZjwJWki1sYBIo,8180
+sqlmesh/core/linter/rules/builtin.py,sha256=2H4yV4eVUCY25uFgO2M62czpZt5dXdK5mW6XGGOc12Q,9857
 sqlmesh/core/metric/__init__.py,sha256=H1HmoD5IwN4YWe9iJXyueLYNmTQFZwok5nSWNJcZIBQ,237
 sqlmesh/core/metric/definition.py,sha256=Yd5aVgsZCDPJ43aGP7WqtzZOuuSUtB8uJGVA6Jw9x9M,7201
 sqlmesh/core/metric/rewriter.py,sha256=GiSTHfn2kinqCfNPYgZPRk93JFLzVaaejHtHDQ0yXZI,7326
@@ -287,7 +287,7 @@ sqlmesh/utils/pydantic.py,sha256=o_NsXbIpDqNpUA1Uc5xF0ZzoXQYYB0DfHwdRxBwXPNk,120
 sqlmesh/utils/rich.py,sha256=cwQ5nJ6sgz64xHtoh6_ec7ReV5YpsOGhMtUJnwoRfEI,3549
 sqlmesh/utils/windows.py,sha256=r7Yin4WEwbvH7sKA3hOJtRsclmscNuBeCklCfVPvtQs,611
 sqlmesh/utils/yaml.py,sha256=hvo5lmybYLcIx4p6CQwufS8X4MbwS60lEBnCmYJCXmU,2336
-sqlmesh-0.205.1.dev20.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
+sqlmesh-0.205.1.dev21.dist-info/licenses/LICENSE,sha256=OlMefUjgWJdULtf84BLW0AZZcY8DwdgQqb_1j2862j8,11346
 web/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 web/client/dist/index.html,sha256=SXykdSRNgrT1k6I2dPZPEfpxPjIsi8-yVFijt-naSA4,5472
 web/client/dist/assets/Audits-Jm353fhk.js,sha256=ry6b5IKT3hLAfD5C3T2BqxulfauEV0n0YZ2vsP4eQ_U,1041
@@ -405,8 +405,8 @@ web/server/api/endpoints/models.py,sha256=kwj0s7uve3iZSMfmjkoPVMFMeY1sD0peTeyrWf
 web/server/api/endpoints/modules.py,sha256=8hqqgonGay_mJmpCw0IdbjsPhWlQH2VLdKAqha-myac,468
 web/server/api/endpoints/plan.py,sha256=bbbY50W_2MsZSTxOHWMKz0tbIm75nsRSlPy8GI2fg9Q,9306
 web/server/api/endpoints/table_diff.py,sha256=SCy3-pr5x6bEEiOaZGCDbiHQhpgxwmnz3Acqa2Ie6jI,2090
-sqlmesh-0.205.1.dev20.dist-info/METADATA,sha256=LvQ9ZUIDmgH6RTQH9OI6m4stcnRpdLCCiRWU28i70KM,26686
-sqlmesh-0.205.1.dev20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sqlmesh-0.205.1.dev20.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
-sqlmesh-0.205.1.dev20.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
-sqlmesh-0.205.1.dev20.dist-info/RECORD,,
+sqlmesh-0.205.1.dev21.dist-info/METADATA,sha256=RRkaqgMA5WuwjWQjkA2zY_0vkVgX7NXRMk03OHjWxps,26686
+sqlmesh-0.205.1.dev21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sqlmesh-0.205.1.dev21.dist-info/entry_points.txt,sha256=pIYKtg-u-uO7BK_Y0qrdmfyj0bYrcMc3Xiqr5WE3VFk,121
+sqlmesh-0.205.1.dev21.dist-info/top_level.txt,sha256=bspjo5vjELQkuN0qKuTFIbsrUt3Zeo8D2mlKeHG1LdE,12
+sqlmesh-0.205.1.dev21.dist-info/RECORD,,