#!/usr/bin/env python

from . import main


class StorageEngine(object):

    def __init__(self):
        main.conectar()

    def authenticate(self, args):
        return main.authenticate(args['usuario'], args['contra'])

    def get_partners(self, values):
        return main.get_partners(values)

    def new_partner(self, values):
        return main.new_partner(values)

    def update_partner(self, values):
        id = int(values['id'])
        del values['id']
        fields = util.clean(values)
        fields['rfc'] = fields['rfc'].upper()
        q = Clients.update(**values).where(Clients.id==id)
        q.execute()
        row = {
            'id': id,
            'cia': fields['cia'],
            'rfc': fields['rfc'],
            'name': fields['name'],
        }
        data = {'ok': True, 'row': row, 'new': False}
        return data

    def delete_partner(self, id):
        q = Clients.delete().where(Clients.id==id)
        return bool(q.execute())