from products.models import Product


class ProductDBRouter(object):

    def db_for_read(self, model, **hints):
        if model == Product:
            return 'products'
        return None

    def db_for_write(self, model, **hints):
        if model == Product:
            return 'products'
        return None