mikewooster / jsonmarshal Goto Github PK

@dataclass 
   ...: class XXX: 
   ...:     a: int 
   ...:                                                                         

In [4]: marshal(XXX(a=1))                                                       
Out[4]: {'a': 1}

In [5]: x = marshal(XXX(a=1))                                                   

In [6]: unmarshal(x, XXX)                                                       
Out[6]: XXX(a=1)

In [7]: @dataclass 
   ...: class YYY(XXX): 
   ...:     b: int 
   ...:                                                                         

In [8]: x = marshal(YYY(a=1, b=2))                                              

In [9]: x                                                                       
Out[9]: {'a': 1, 'b': 2}

In [10]: unmarshal(x, YYY)                                                      
---------------------------------------------------------------------------
KeyError                                  Traceback (most recent call last)
<ipython-input-10-dc946d818557> in <module>
----> 1 unmarshal(x, YYY)

~/.local/lib/python3.8/site-packages/jsonmarshal/unmarshal.py in unmarshal(response, schema, datetime_fmt, date_fmt)
     40     """
     41     unmarshaller = _Unmarshaller(response, schema, datetime_fmt, date_fmt)
---> 42     return unmarshaller.unmarshal()
     43 
     44 

~/.local/lib/python3.8/site-packages/jsonmarshal/unmarshal.py in unmarshal(self)
    126 
    127             processor = self.processors[item.schema_type]
--> 128             processor(item)
    129 
    130             self.promote()

~/.local/lib/python3.8/site-packages/jsonmarshal/unmarshal.py in process_dict(self, item)
    170         if not item.cleaned:
    171             # Go through each known field and fix the keys in the original dictionary
--> 172             item = self.clean_item(item)
    173 
    174         if not self.dump and item.unmarshalled is False:

~/.local/lib/python3.8/site-packages/jsonmarshal/unmarshal.py in clean_item(self, item)
    205 
    206         for data_key, v in item.data.items():
--> 207             schema_type = item.schema.__annotations__[data_key]
    208             child = _ResultContainer(
    209                 data=v,

KeyError: 'a'

from jsonmarshal import unmarshal, marshal, json_field

from dataclasses import dataclass, field
from typing import List, Set, Optional


from dataclasses import dataclass

@dataclass
class Child:
	name : str

@dataclass
class Parent:
	child : Optional[ Child ] = json_field( omitempty = True, default = None )


parent = Parent()

child = Child( 'name' )


print('jsonmarshal can marshal/unmarshal both object types individually...\n')
print( marshal( parent ) )
print( marshal( child ) )
print( unmarshal( marshal( parent ), Parent ) )
print( unmarshal( marshal( child ), Child ) )
print('\n\n\n')


parent.child = child
parent_json = marshal( parent )
print( 'jsonmarshal can marshal the object\n' )
print( parent_json )
print( '\n' )

try:
	print('unmarshalling with an optional parameter that is a nested class fails')
	unmarshal( parent_json, Parent ) # throws an error
except AttributeError as e:
	print( str( e ) )