+# Copyright (C) 2021 Wind River Systems, Inc.\r
+#\r
+# Licensed under the Apache License, Version 2.0 (the "License");\r
+# you may not use this file except in compliance with the License.\r
+# You may obtain a copy of the License at\r
+#\r
+# http://www.apache.org/licenses/LICENSE-2.0\r
+#\r
+# Unless required by applicable law or agreed to in writing, software\r
+# distributed under the License is distributed on an "AS IS" BASIS,\r
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
+# See the License for the specific language governing permissions and\r
+# limitations under the License.\r
+\r
+import abc\r
+from typing import List, Set\r
+from o2ims.domain.stx_object import StxGenericModel\r
+from o2ims.domain.resource_type import ResourceTypeEnum\r
+\r
+\r
+class StxObjectRepository(abc.ABC):\r
+ def __init__(self):\r
+ self.seen = set() # type: Set[StxGenericModel]\r
+\r
+ def add(self, stx_obj: StxGenericModel):\r
+ self._add(stx_obj)\r
+ self.seen.add(stx_obj)\r
+\r
+ def get(self, stx_obj_id) -> StxGenericModel:\r
+ stx_obj = self._get(stx_obj_id)\r
+ if stx_obj:\r
+ self.seen.add(stx_obj)\r
+ return stx_obj\r
+\r
+ def list(self, type: ResourceTypeEnum) -> List[StxGenericModel]:\r
+ return self._list(type)\r
+\r
+ def update(self, stx_obj: StxGenericModel):\r
+ self._update(stx_obj)\r
+\r
+ # def update_fields(self, stx_obj_id: str, updatefields: dict):\r
+ # self._update(stx_obj_id, updatefields)\r
+\r
+ @abc.abstractmethod\r
+ def _add(self, stx_obj: StxGenericModel):\r
+ raise NotImplementedError\r
+\r
+ @abc.abstractmethod\r
+ def _get(self, stx_obj_id) -> StxGenericModel:\r
+ raise NotImplementedError\r
+\r
+ @abc.abstractmethod\r
+ def _update(self, stx_obj: StxGenericModel):\r
+ raise NotImplementedError\r
+\r
+ @abc.abstractmethod\r
+ def _list(self, type: ResourceTypeEnum):\r
+ raise NotImplementedError\r