refactor: support also generators, tuples, etc.

Author: Nathan Richard

README.md

@@ -31,7 +31,7 @@ ___
 - Familiar syntax. The `BasemodelCSVReader` is used almost the same way as the `DictReader` in the standard library.
 - It uses `BaseModel` features that let you define Field properties or Config so the data can be parsed exactly the way you want.
 - Make the code cleaner. No more extra loops to convert data to the correct type, perform validation, set default values, the `BasemodelCSVReader` will do all this for you.
-- In addition to the `BasemodelCSVReader`, the library also provides a `BasemodelCSVWriter` which enables creating a CSV file using a list of instances of a BaseModel.
+- In addition to the `BasemodelCSVReader`, the library also provides a `BasemodelCSVWriter` which enables creating a CSV file using an Iterable with instances of a BaseModel.
 - Because [sqlmodel](https://github.com/tiangolo/sqlmodel) uses pydantic.BaseModels too, you can directly fill a database with data from a CSV
 
 
@@ -298,7 +298,7 @@ class User(BaseModel):
     age: int
 ```
 
-And in your program we have a list of users:
+And in your program we have a list (also supports Generator and Tuples. Just any Iterable that supports storing Objects) of users:
 
 ```python
 users = [

pydantic_csv/basemodel_csv_writer.py

@@ -3,6 +3,7 @@
 """
 
 import csv
+from collections.abc import Iterable
 from typing import Any
 
 import pydantic
@@ -19,7 +20,7 @@ class BasemodelCSVWriter:
     def __init__(
         self,
         file_obj: Any,
-        data: list[Any],
+        data: Iterable,
         model: type[BaseModel],
         *,
         use_alias: bool = True,
@@ -29,8 +30,11 @@ def __init__(
         if not file_obj:
             raise ValueError("The 'file_obj' argument is required")
 
-        if not isinstance(data, list):
-            raise ValueError("Invalid 'data' argument. It must be a list")
+        if not isinstance(data, Iterable) or isinstance(data, (str, BaseModel)):
+            raise ValueError(
+                "Invalid 'data' argument. It must be an Iterable which can hold multiple BaseModel "
+                "instances. eg: list, generator, tuple"
+            )
 
         if model is None or not issubclass(model, pydantic.BaseModel):
             raise ValueError("Invalid 'cls' argument. It must be a pydantic BaseModel")