Fix adding set object to itemloader

scrapy · May 14, 2024 · 7ccd975 · 7ccd975
1 parent 5fe228a
commit 7ccd975
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/itemloaders/utils.py b/itemloaders/utils.py
@@ -20,7 +20,7 @@ def arg_to_iter(arg: Any) -> Iterable[Any]:
     """
     if arg is None:
         return []
-    if isinstance(arg, (list, tuple, Generator)):
+    if isinstance(arg, (list, tuple, set, Generator)):
         return arg
     return [arg]
 

diff --git a/tests/test_nested_items.py b/tests/test_nested_items.py
@@ -39,6 +39,13 @@ class TestItem:
     def test_dict(self):
         self._test_item({"foo": "bar"})
 
+    def test_set(self):
+        item = {"foo", "bar"}
+        il = ItemLoader()
+        il.add_value("item_list", item)
+
+        self.assertEqual(il.load_item(), {"item_list": ["foo", "bar"]})
+
     def test_scrapy_item(self):
         try:
             from scrapy import Field, Item