craftbeerpi4-pione/cbpi/controller/crud_controller.py

import pprint
from abc import ABCMeta

from cbpi.api import *


class CRUDController(metaclass=ABCMeta):


    cache = {}
    caching = True
    name = None

    def __init__(self, cbpi):
        self.cbpi = cbpi
        self.cache = {}

    async def init(self):
        '''
        
        :return: 
        '''

        if self.caching is True:
            self.cache = await self.model.get_all()

    async def get_all(self, force_db_update=False):
        '''
        
        :param force_db_update: 
        :return: 
        '''
        if self.caching is False or force_db_update:
            self.cache = await self.model.get_all()

        return self.cache

    async def get_one(self, id):
        '''
        
        :param id: 
        :return: 
        '''
        if id not in self.cache:
            raise CBPiException("%s with id %s not found" % (self.name,id))
        return self.cache.get(id)

    async def _pre_add_callback(self, data):
        '''
        
        :param data: 
        :return: 
        '''
        pass

    async def _post_add_callback(self, m):
        '''
        
        :param m: 
        :return: 
        '''
        pass

    async def add(self, **data):
        '''
        
        :param data: 
        :return: 
        '''


        await self._pre_add_callback(data)
        print("INSSERT ADD", data)

        m = await self.model.insert(**data)

        self.cache[m.id] = m
        await self._post_add_callback(m)

        await self.cbpi.bus.fire(topic="actor/%s/added" % m.id, actor=m)

        return m

    async def _pre_update_callback(self, m):
        pass

    async def _post_update_callback(self, m):
        pass

    async def update(self, id, data):
        '''
        
        :param id: 
        :param data: 
        :return: 
        '''

        self.logger.debug("Update Sensor %s - %s " % (id, data))
        id = int(id)

        if id not in self.cache:
            self.logger.debug("Sensor %s Not in Cache" % (id,))
            raise CBPiException("%s with id %s not found" % (self.name,id))

        data["id"] = id

        try:
            ### DELETE INSTANCE BEFORE UPDATE
            del data["instance"]
        except Exception as e:
            pass

        if self.caching is True:
            await self._pre_update_callback(self.cache[id])
            self.cache[id].__dict__.update(**data)
            m = self.cache[id] = await self.model.update(**self.cache[id].__dict__)
            await self._post_update_callback(self.cache[id])

        else:

            m = await self.model.update(**data)
        return m



    async def _pre_delete_callback(self, m):
        '''
        
        :param m: 
        :return: 
        '''
        pass

    async def _post_delete_callback(self, id):
        '''
        
        :param id: 
        :return: 
        '''
        pass

    async def delete(self, id):
        '''
        
        :param id: 
        :return: 
        '''

        if id not in self.cache:
            raise CBPiException("%s with id %s not found" % (self.name,id))

        await self._pre_delete_callback(id)
        m = await self.model.delete(id)
        await self._post_delete_callback(id)
        try:
            if self.caching is True:
                del self.cache[int(id)]
        except Exception as e:
            pass

        await self.cbpi.bus.fire(topic="actor/%s/deleted" % id, id=id)

    async def delete_all(self):
        '''
        
        :return: 
        '''
        self.model.delete_all()
        if self.caching is True:
            self.cache = {}
        #self.cbpi.push_ws("DELETE_ALL_%s" % self.key, None)