import json
from collections import namedtuple

import flask
from sqlalchemy.orm.session import Session

ModelConfiguration = namedtuple('ModelConfiguration',
                                'collection_name view blueprint')


def abort(msg):
    resp = flask.jsonify(message=msg)
    resp.status_code = 500
    flask.abort(resp)


def cr():
    return flask.current_app.extensions['cereal']


def register_serializer(model, pk_name, serialize, deserialize, cr):
    def load_model(value):
        pkval = value.get(pk_name)
        if pkval:
            return model.query.filter_by(**{pk_name: pkval}).one_or_none()
        return deserialize(value)

    def serialize_model(value):
        return serialize(value)

    cr.register_class(model.__name__, model, serialize_model, load_model)


def run_object_method(instid, function_name, model, commit_on_return):
    instance = model.query.get(instid)
    if not instance:
        return {}
    params = cr().loads(flask.request.get_json()['payload'])
    try:
        result = getattr(instance, function_name)(*params['args'],
                                                  **params['kwargs'])
        result = json.dumps({'payload': cr().dumps(result)})
    except Exception as e:
        msg = '{}: {}'.format(e.__class__.__name__, str(e))
        abort(msg)

    if commit_on_return:
        try:
            session = Session.object_session(instance)
            session.commit()
        except Exception:
            pass

    return result


def object_property(instid, model, property_name):
    if flask.request.method == 'GET':
        return get_object_property(instid, model, property_name)
    else:
        return set_object_property(instid, model, property_name)


def get_object_property(instid, model, property_name):
    instance = model.query.get(instid)
    if not instance:
        return {}
    result = getattr(instance, property_name)
    return json.dumps({'payload': cr().dumps(result)})


def set_object_property(instid, model, property_name):
    instance = model.query.get(instid)
    if not instance:
        return {}

    value = cr().loads(flask.request.get_json())
    try:
        setattr(instance, property_name, value)
        session = Session.object_session(instance)
        session.commit()
    except Exception as e:
        abort("Could not set property: {}".format(e))
    return json.dumps({'message': 'success'})

import json
from collections import namedtuple

import flask
from sqlalchemy.orm.session import Session

ModelConfiguration = namedtuple('ModelConfiguration',
                                'collection_name view blueprint')


def abort(msg):
    resp = flask.jsonify(message=msg)
    resp.status_code = 500
    flask.abort(resp)


def cr():
    return flask.current_app.extensions['cereal']


def register_serializer(model, pk_name, serialize, deserialize, cr):
    def load_model(value):
        pkval = value.get(pk_name)
        if pkval:
            return model.query.filter_by(**{pk_name: pkval}).one_or_none()
        return deserialize(value)

    def serialize_model(value):
        return serialize(value)

    cr.register_class(model.__name__, model, serialize_model, load_model)


def run_object_method(instid, function_name, model, commit_on_return):
    instance = model.query.get(instid)
    if not instance:
        return {}
    params = cr().loads(flask.request.get_json()['payload'])
    try:
        result = getattr(instance, function_name)(*params['args'],
                                                  **params['kwargs'])
        result = json.dumps({'payload': cr().dumps(result)})
    except Exception as e:
        msg = '{}: {}'.format(e.__class__.__name__, str(e))
        abort(msg)

    if commit_on_return:
        try:
            session = Session.object_session(instance)
            session.commit()
        except Exception:
            pass

    return result


def object_property(instid, model, property_name):
    if flask.request.method == 'GET':
        return get_object_property(instid, model, property_name)
    else:
        return set_object_property(instid, model, property_name)


def get_object_property(instid, model, property_name):
    instance = model.query.get(instid)
    if not instance:
        return {}
    result = getattr(instance, property_name)
    return json.dumps({'payload': cr().dumps(result)})


def set_object_property(instid, model, property_name):
    instance = model.query.get(instid)
    if not instance:
        return {}

    value = cr().loads(flask.request.get_json())
    try:
        setattr(instance, property_name, value)
        session = Session.object_session(instance)
        session.commit()
    except Exception as e:
        abort("Could not set property: {}".format(e))
    return json.dumps({'message': 'success'})
