include static ClassVar for discriminator property in both python cli…

…ent and python server codegens
wkarwacki · Oct 8, 2024 · 99ac9c9 · 99ac9c9
1 parent 0e81f90
commit 99ac9c9
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/src/lib/gen/python/client/templates/dto.hbs b/src/lib/gen/python/client/templates/dto.hbs
@@ -11,7 +11,8 @@
 {{#each val.adt.map}}
 {{>dto key=(fmtClass @key) val=this adtParent=../key adtVar=../val.adt.var}}
 
-    {{fmtName ../val.adt.var}}: Literal[{{json @key}}] = {{json @key}}
+    {{fmtName ../val.adt.var}}_val: ClassVar[str] = {{json @key}}
+    {{fmtName ../val.adt.var}}: Literal[{{json @key}}] = {{fmtName ../val.adt.var}}_val
 
 
 {{/each}}

diff --git a/src/lib/gen/python/client/templates/dtoFile.hbs b/src/lib/gen/python/client/templates/dtoFile.hbs
@@ -2,7 +2,7 @@ import httpx
 
 from enum import Enum
 from pydantic import Field, model_validator
-from typing import Annotated, Any, Literal
+from typing import Annotated, Any, ClassVar, Literal
 
 from trust import Dto
 

diff --git a/src/lib/gen/python/server/templates/dto.hbs b/src/lib/gen/python/server/templates/dto.hbs
@@ -11,7 +11,8 @@
 {{#each val.adt.map}}
 {{>dto key=(fmtClass @key) val=this adtParent=../key adtVar=../val.adt.var}}
 
-    {{fmtName ../val.adt.var}}: Literal[{{json @key}}] = {{json @key}}
+    {{fmtName ../val.adt.var}}_val: ClassVar[str] = {{json @key}}
+    {{fmtName ../val.adt.var}}: Literal[{{json @key}}] = {{fmtName ../val.adt.var}}_val
 
 
 {{/each}}

diff --git a/src/lib/gen/python/server/templates/dtoFile.hbs b/src/lib/gen/python/server/templates/dtoFile.hbs
@@ -3,7 +3,7 @@ import json
 from enum import Enum
 from fastapi import File, Form, UploadFile
 from pydantic import Field
-from typing import Annotated, Any, Literal
+from typing import Annotated, Any, ClassVar, Literal
 
 from trust import Dto