From ce4f905aaaddd2a2376eb3f920d4338d27126b24 Mon Sep 17 00:00:00 2001 From: Daniel Townsend Date: Thu, 24 Mar 2022 10:47:25 +0000 Subject: [PATCH] prototype Pydantic choice support in `create_pydantic_model` --- piccolo/utils/pydantic.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/piccolo/utils/pydantic.py b/piccolo/utils/pydantic.py index 216807539..3fd981caf 100644 --- a/piccolo/utils/pydantic.py +++ b/piccolo/utils/pydantic.py @@ -207,8 +207,10 @@ def create_pydantic_model( ####################################################################### # Work out the column type - if isinstance(column, (Decimal, Numeric)): - value_type: t.Type = pydantic.condecimal( + if column._meta.choices: + value_type: t.Type = column._meta.choices + elif isinstance(column, (Decimal, Numeric)): + value_type = pydantic.condecimal( max_digits=column.precision, decimal_places=column.scale ) elif isinstance(column, Varchar):