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_categorias(self, values):
        return main.Categorias.get_all()

    def _get_newkey(self, values):
        return main.Productos.next_key()

    def _get_unidades(self, values):
        return main.SATUnidades.get_activos()

    def _get_taxes(self, values):
        return main.SATImpuestos.get_activos()

    def _get_satkey(self, values):
        return main.get_sat_key(values['key'])

    def delete(self, table, id):
        if table == 'partner':
            return main.Socios.remove(id)
        if table == 'product':
            return main.Productos.remove(id)
        return False

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

    def partner(self, values):
        id = int(values['id'])
        del values['id']
        if id:
            return main.Socios.actualizar(values, id)
        return main.Socios.add(values)

    def get_products(self, values):
        return main.Productos.get_(values)

    def product(self, values):
        id = int(values['id'])
        del values['id']
        if id:
            return main.Productos.actualizar(values, id)
        return main.Productos.add(values)