From d39ac10c035071abf3f8d2d604f5aa475485efab Mon Sep 17 00:00:00 2001
From: phaer <hello@phaer.org>
Date: Sun, 19 Sep 2021 09:47:16 +0200
Subject: [PATCH] Throw ExpandError for unknown in .extract_expand()

---
 sqlite_utils/db.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/sqlite_utils/db.py b/sqlite_utils/db.py
index 649ec3a..5995e4a 100644
--- a/sqlite_utils/db.py
+++ b/sqlite_utils/db.py
@@ -144,6 +144,10 @@ class InvalidColumns(Exception):
     pass
 
 
+class ExpandError(Exception):
+    pass
+
+
 _COUNTS_TABLE_CREATE_SQL = """
 CREATE TABLE IF NOT EXISTS [{}](
    [table] TEXT PRIMARY KEY,
@@ -1168,6 +1172,9 @@ class Table(Queryable):
                 for new_row in expanded:
                     new_pk = self.db[table].insert(new_row, pk="id", replace=True).last_pk
                     self.update(row_pk, {fk_column: new_pk})
+            else:
+                raise ExpandError("expanded value needs to be list or dict")
+
         # Can drop the original column now
         self.transform(drop=[column])
         # And add that foreign key