Don't forget the serialization alias!

Author: Henry Walshaw

pydantic_csv/basemodel_csv_writer.py

@@ -53,7 +53,9 @@ def __init__(
         self._use_alias = use_alias
 
         if self._use_alias:
-            self._fieldnames = [field.alias or name for name, field in fields.items()]
+            self._fieldnames = [
+                field.alias or getattr(field, "serialization_alias", None) or name for name, field in fields.items()
+            ]
         else:
             self._fieldnames = fields.keys()
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pydantic-csv"
-version = "0.1.1"
+version = "0.1.2"
 description = "convert CSV to pydantic.BaseModel and vice versa"
 authors = ["Nathan Richard <contact@nathanrichard.dev>"]
 license = "LICENSE"

tests/models.py

@@ -56,6 +56,7 @@ def parse_end_date(cls, value):
 class ExcludedPassword(pydantic.BaseModel):
     username: str = "Wagstaff"
     password: str = Field(default="swordfish", exclude=True)
+    email: str = Field(default="wagstaff@marx.bros", serialization_alias="contact")
 
 
 class ComputedPropertyField(pydantic.BaseModel):

tests/test_basemodel_csv_writer.py

@@ -66,7 +66,7 @@ def test_excluded_field():
     w = BasemodelCSVWriter(output, [user], ExcludedPassword)
     w.write()
 
-    assert output.getvalue() == "username\r\nWagstaff\r\n"
+    assert output.getvalue() == "username,contact\r\nWagstaff,wagstaff@marx.bros\r\n"
 
 
 @pytest.mark.parametrize(