marshmallow-models

marshmodel.py

import six
import inflection
import marshmallow as ma


class Model(object):
    
    def __init__(self, **kwargs):
        self._schema = self.Schema()
        self.load(**kwargs)
        
    def load(self, **kwargs):
        data = {key: None for key in self._schema.fields}
        data.update(self._schema.load(kwargs).data)
        self.__dict__.update(data)
    
    def dump(self):
        return self._schema.dump(self).data
    

class InflectSchema(ma.Schema):

    @ma.pre_load
    def pre_load(self, data):
        return {
            inflection.underscore(key): value
            for key, value in six.iteritems(data)
        }
        
    @ma.post_dump
    def post_dump(self, data):
        return {
            inflection.camelize(key, uppercase_first_letter=False): value
            for key, value in six.iteritems(data)
        }


class Cat(Model):
    
    class Schema(InflectSchema):
        name = ma.fields.String()
        fave_food = ma.fields.String()


cat = Cat(name='calici', faveFood='pellets')