PyYAML에 대한 대입 연산자는 어떻게 작성합니까?

Question

json.dump 함수에 'default'라는 옵션 arg가있는 것처럼 임의의 파이썬 개체를 직렬화하는 사용자 지정 함수가 있어야합니다.이 개체는 json serialum이 아닌 개체 인 경우 json dumper가 호출하는 함수 여야합니다.

json 패키지에서 이와 동일한 작업을 수행하기 만하면됩니다. 나는 PyYAML 워드 프로세서에서, yaml.add_representer 작성해야처럼

json.dump(tests_dump, file('somefile', 'w+'), default = lambda x: x.__dict__) 

그것은 보이지만, 정말이 작업을 수행하는 방법을 명확하지 않다.

Answer 1

다음은 add_representer의 샘플입니다. 이것이 정확히 원하는 것인지 확실하지 않습니다. 그럼에도 불구하고 ...

import yaml 

#Arbitrary Class 
class MyClass: 
    def __init__(self, someNumber, someString): 
    self.var1 = someNumber 
    self.var2 = someString 

#define the representer, responsible for serialization 
def MyClass_representer(dumper, data): 
    serializedData = str(data.var1) + "|" + data.var2 
    return dumper.represent_scalar('!MyClass', serializedData) 

#'register' it  
yaml.add_representer(MyClass, MyClass_representer) 

obj = MyClass(100,'test') 

print ('original Object\nvar1:{0}, var2:{1}\n'.format(obj.var1, obj.var2)) 

#serialize 
yamlData = yaml.dump(obj) 

print('serialized as:\n{0}'.format(yamlData)) 

#Now to deserialize you need a constructor 
def MyClass_constructor(loader,node): 
    value = loader.construct_scalar(node) 
    someNumber,sep,someString = value.partition("|") 
    return MyClass(someNumber,someString) 

#'register' it  
yaml.add_constructor('!MyClass', MyClass_constructor) 

#deserialize 
obj2 = yaml.load(yamlData) 

print ('after deserialization\nvar1:{0}, var2:{1}\n'.format(obj2.var1, obj2.var2)) 

물론 코드 중복이 있으며 코드가 최적화되지 않았습니다. 이 두 함수를 클래스의 일부로 만들고 __repr__을 구현하여 serializedData을 MyClass_representer