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")