Commit 655d6a0f authored by Lluis Gifre Renom's avatar Lluis Gifre Renom
Browse files

Common > tests > InMem Obj DB: 

- Reduce log messages
parent 536786b5

src/common/tests/InMemoryObjectDatabase.py

+8 −8
Original line number Diff line number Diff line
@@ -12,10 +12,10 @@ 
# See the License for the specific language governing permissions and

# limitations under the License.



import grpc, logging

import grpc #, logging

from typing import Any, Dict, List, Set



LOGGER = logging.getLogger(__name__)

#LOGGER = logging.getLogger(__name__)



class InMemoryObjectDatabase:

    def __init__(self) -> None:
@@ -29,12 +29,12 @@ class InMemoryObjectDatabase: 
        return [container[entry_uuid] for entry_uuid in sorted(container.keys())]



    def has_entry(self, container_name : str, entry_uuid : str) -> Any:

        LOGGER.debug('[has_entry] BEFORE database={:s}'.format(str(self._database)))

        #LOGGER.debug('[has_entry] BEFORE database={:s}'.format(str(self._database)))

        container = self._get_container(container_name)

        return entry_uuid in container



    def get_entry(self, container_name : str, entry_uuid : str, context : grpc.ServicerContext) -> Any:

        LOGGER.debug('[get_entry] BEFORE database={:s}'.format(str(self._database)))

        #LOGGER.debug('[get_entry] BEFORE database={:s}'.format(str(self._database)))

        container = self._get_container(container_name)

        if entry_uuid not in container:

            MSG = '{:s}({:s}) not found; available({:s})'
@@ -44,18 +44,18 @@ class InMemoryObjectDatabase: 


    def set_entry(self, container_name : str, entry_uuid : str, entry : Any) -> Any:

        container = self._get_container(container_name)

        LOGGER.debug('[set_entry] BEFORE database={:s}'.format(str(self._database)))

        #LOGGER.debug('[set_entry] BEFORE database={:s}'.format(str(self._database)))

        container[entry_uuid] = entry

        LOGGER.debug('[set_entry] AFTER database={:s}'.format(str(self._database)))

        #LOGGER.debug('[set_entry] AFTER database={:s}'.format(str(self._database)))

        return entry



    def del_entry(self, container_name : str, entry_uuid : str, context : grpc.ServicerContext) -> None:

        container = self._get_container(container_name)

        LOGGER.debug('[del_entry] BEFORE database={:s}'.format(str(self._database)))

        #LOGGER.debug('[del_entry] BEFORE database={:s}'.format(str(self._database)))

        if entry_uuid not in container:

            context.abort(grpc.StatusCode.NOT_FOUND, str('{:s}({:s}) not found'.format(container_name, entry_uuid)))

        del container[entry_uuid]

        LOGGER.debug('[del_entry] AFTER database={:s}'.format(str(self._database)))

        #LOGGER.debug('[del_entry] AFTER database={:s}'.format(str(self._database)))



    def select_entries(self, container_name : str, entry_uuids : Set[str]) -> List[Any]:

        if len(entry_uuids) == 0: return self.get_entries(container_name)