Merge pull request #351 from DataRecce/cherrypick/feature/drc-511-add…

…-snapshot-node-in-lineage Add snapshot in the lineage
DataRecce · Jun 21, 2024 · f3f438f · f3f438f
2 parents e3388fe + 005aa3d
commit f3f438f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/js/src/components/lineage/NodeView.tsx b/js/src/components/lineage/NodeView.tsx
@@ -49,7 +49,8 @@ export function NodeView({ node, onCloseNode }: NodeViewProps) {
   const withColumns =
     node.resourceType === "model" ||
     node.resourceType === "seed" ||
-    node.resourceType === "source";
+    node.resourceType === "source" ||
+    node.resourceType === "snapshot";
   const {
     isOpen: isCodeDiffOpen,
     onOpen: onCodeDiffOpen,

diff --git a/js/src/components/lineage/styles.tsx b/js/src/components/lineage/styles.tsx
@@ -1,5 +1,5 @@
 import { VscDiffAdded, VscDiffModified, VscDiffRemoved } from "react-icons/vsc";
-import { FaCube, FaDatabase, FaSeedling } from "react-icons/fa";
+import { FaCamera, FaCube, FaDatabase, FaSeedling } from "react-icons/fa";
 import { FaChartSimple, FaCircleNodes, FaGauge } from "react-icons/fa6";
 
 export const IconAdded = VscDiffAdded;
@@ -89,6 +89,8 @@ export function getIconForResourceType(resourceType?: string): {
     return { color: "#fb8caf", icon: FaCircleNodes };
   } else if (resourceType === "seed") {
     return { color: "#a6dda6", icon: FaSeedling };
+  } else if (resourceType === "snapshot") {
+    return { color: "#a6dda6", icon: FaCamera };
   } else {
     return { color: "inherit", icon: undefined };
   }

diff --git a/recce/adapter/dbt_adapter/__init__.py b/recce/adapter/dbt_adapter/__init__.py
@@ -398,7 +398,7 @@ def get_lineage(self, base: Optional[bool] = False):
             unique_id = node['unique_id']
             resource_type = node['resource_type']
 
-            if resource_type not in ['model', 'seed', 'exposure']:
+            if resource_type not in ['model', 'seed', 'exposure', 'snapshot']:
                 continue
 
             nodes[unique_id] = {