empresa-libre/source/app/models/db.py

#!/usr/bin/env python

from . import main


class StorageEngine(object):

    def __init__(self):
        pass

    def authenticate(self, args):
        return main.authenticate(args)

    def get_values(self, table, values=None):
        return getattr(self, '_get_{}'.format(table))(values)

    def _get_cp(self, values):
        return main.get_cp(values['cp'])

    def _get_formapago(self, values):
        return main.SATFormaPago.get_activos()

    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())