ensure motherduck token isn't included in primitive ID

launchflow · Dec 21, 2023 · 5564b2e · 5564b2e
1 parent df84fce
commit 5564b2e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/buildflow/io/duckdb/duckdb.py b/buildflow/io/duckdb/duckdb.py
@@ -26,7 +26,10 @@ def __post_init__(self):
             self.database = f"{self.database}?{self.motherduck_token}"
 
     def primitive_id(self):
-        return f"{self.database}:{self.table}"
+        db = self.database
+        if "?" in self.database:
+            db = self.database.split("?")[0]
+        return f"{db}:{self.table}"
 
     @classmethod
     def from_local_options(