Skip to content

Commit 8d393d7

Browse files
committed
Updating deserializers
1 parent fcc3905 commit 8d393d7

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

pylasu/lionweb/ast_generation.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -130,8 +130,12 @@ def _generate_constructor(concept: Concept) -> ast.FunctionDef:
130130
slice=ast.Name(id="Position", ctx=ast.Load()),
131131
ctx=ast.Load()
132132
)),
133+
ast.arg(arg="concept", annotation=ast.Name(id="Concept", ctx=ast.Load()))
133134
],
134-
kwonlyargs=[], kw_defaults=[], defaults=[]
135+
kwonlyargs=[], kw_defaults=[],
136+
defaults = [
137+
ast.Constant(value=to_snake_case(concept.get_name()).upper())
138+
]
135139
),
136140
body=[
137141
# super().__init__(id=id, position=position, concept=concept)

pylasu/lionweb/deserializer_generation.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,7 @@ def deserializer_generation(click, language: Language, output):
223223
for e in language.get_elements():
224224
if isinstance(e, Enumeration):
225225
arg_serialized = ast.arg(arg="serialized", annotation=ast.Name(id="str", ctx=ast.Load()))
226+
arg_required = ast.arg(arg="required", annotation=ast.Name(id="bool", ctx=ast.Load()))
226227
# The raise: raise ValueError(f"...")
227228
raise_stmt = ast.Raise(
228229
exc=ast.Call(
@@ -266,7 +267,7 @@ def deserializer_generation(click, language: Language, output):
266267
name=f"_deserialize_{to_snake_case(e.get_name())}",
267268
args=ast.arguments(
268269
posonlyargs=[],
269-
args=[arg_serialized],
270+
args=[arg_serialized, arg_required],
270271
kwonlyargs=[],
271272
kw_defaults=[],
272273
defaults=[]

0 commit comments

Comments
 (0)