Add CollectorRule service

src/spaceone/inventory_v2/error/__init__.py

@@ -1,3 +1,4 @@
 from spaceone.inventory_v2.error.region import *
 from spaceone.inventory_v2.error.collector import *
+from spaceone.inventory_v2.error.collector_rule import *
 from spaceone.inventory_v2.error.metric import *

src/spaceone/inventory_v2/error/collector_rule.py

@@ -0,0 +1,13 @@
+from spaceone.core.error import *
+
+
+class ERROR_NOT_ALLOWED_TO_UPDATE_RULE(ERROR_INVALID_ARGUMENT):
+    _message = "If rule_type is MANAGED, it cannot be updated."
+
+
+class ERROR_NOT_ALLOWED_TO_CHANGE_ORDER(ERROR_INVALID_ARGUMENT):
+    _message = "If rule_type is MANAGED, it cannot be changed."
+
+
+class ERROR_NOT_ALLOWED_TO_DELETE_RULE(ERROR_INVALID_ARGUMENT):
+    _message = "If rule_type is MANAGED, it cannot be deleted."

src/spaceone/inventory_v2/interface/grpc/__init__.py

@@ -3,6 +3,7 @@
 from .asset import Asset
 from .region import Region
 from .collector import Collector
+from .collector_rule import CollectorRule
 from .job import Job
 from .job_task import JobTask
 from .metric import Metric
@@ -17,6 +18,7 @@
 app = GRPCServer()
 app.add_service(Region)
 app.add_service(Collector)
+app.add_service(CollectorRule)
 app.add_service(Job)
 app.add_service(JobTask)
 app.add_service(NamespaceGroup)

src/spaceone/inventory_v2/interface/grpc/collector_rule.py

@@ -0,0 +1,51 @@
+from spaceone.api.inventory_v2.v1 import collector_rule_pb2, collector_rule_pb2_grpc
+from spaceone.core.pygrpc import BaseAPI
+
+from spaceone.inventory_v2.service.collector_rule_service import CollectorRuleService
+
+
+class CollectorRule(BaseAPI, collector_rule_pb2_grpc.CollectorRuleServicer):
+    pb2 = collector_rule_pb2
+    pb2_grpc = collector_rule_pb2_grpc
+
+    def create(self, request, context):
+        params, metadata = self.parse_request(request, context)
+        collector_rule_svc = CollectorRuleService(metadata)
+        response: dict = collector_rule_svc.create(params)
+        return self.dict_to_message(response)
+
+    def update(self, request, context):
+        params, metadata = self.parse_request(request, context)
+        collector_rule_svc = CollectorRuleService(metadata)
+        response: dict = collector_rule_svc.update(params)
+        return self.dict_to_message(response)
+
+    def delete(self, request, context):
+        params, metadata = self.parse_request(request, context)
+        collector_rule_svc = CollectorRuleService(metadata)
+        collector_rule_svc.delete(params)
+        return self.empty()
+
+    def get(self, request, context):
+        params, metadata = self.parse_request(request, context)
+        collector_rule_svc = CollectorRuleService(metadata)
+        response: dict = collector_rule_svc.get(params)
+        return self.dict_to_message(response)
+
+    def change_order(self, request, context):
+        params, metadata = self.parse_request(request, context)
+        collector_rule_svc = CollectorRuleService(metadata)
+        response: dict = collector_rule_svc.change_order(params)
+        return self.dict_to_message(response)
+
+    def list(self, request, context):
+        params, metadata = self.parse_request(request, context)
+        collector_rule_svc = CollectorRuleService(metadata)
+        response: dict = collector_rule_svc.list(params)
+        return self.dict_to_message(response)
+
+    def stat(self, request, context):
+        params, metadata = self.parse_request(request, context)
+        collector_rule_svc = CollectorRuleService(metadata)
+        response: dict = collector_rule_svc.stat(params)
+        return self.dict_to_message(response)

src/spaceone/inventory_v2/manager/__init__.py

@@ -1,5 +1,6 @@
 from spaceone.inventory_v2.manager.asset_manager import AssetManager
 from spaceone.inventory_v2.manager.asset_type_manager import AssetTypeManager
+from spaceone.inventory_v2.manager.collector_rule_manager import CollectorRuleManager
 from spaceone.inventory_v2.manager.region_manager import RegionManager
 from spaceone.inventory_v2.manager.collecting_manager import CollectingManager
 from spaceone.inventory_v2.manager.namespace_group_manager import NamespaceGroupManager

src/spaceone/inventory_v2/model/collector_rule/database.py

@@ -33,6 +33,7 @@ class CollectorRule(MongoModel):
     workspace_id = StringField(max_length=40)
     domain_id = StringField(max_length=40)
     created_at = DateTimeField(auto_now_add=True)
+    updated_at = DateTimeField(auto_now=True)
 
     meta = {
         "updatable_fields": [

src/spaceone/inventory_v2/model/collector_rule/request.py

@@ -0,0 +1,73 @@
+from typing import Union, Literal
+from pydantic import BaseModel
+
+__all__ = [
+    "CollectorRuleCreateRequest",
+    "CollectorRuleUpdateRequest",
+    "CollectorRuleDeleteRequest",
+    "CollectorRuleGetRequest",
+    "CollectorRuleChangeOrderRequest",
+    "CollectorRuleSearchQueryRequest",
+    "CollectorRuleStatQueryRequest",
+]
+
+ResourceGroup = Literal["DOMAIN", "WORKSPACE"]
+
+
+class CollectorRuleCreateRequest(BaseModel):
+    collector_id: str
+    name: Union[str, None] = None
+    conditions: Union[list, None] = None
+    conditions_policy: str
+    actions: dict
+    options: Union[dict, None] = None
+    tags: Union[dict, None] = None
+    workspace_id: Union[str, None] = None
+    domain_id: str
+
+
+class CollectorRuleUpdateRequest(BaseModel):
+    collector_rule_id: str
+    name: Union[str, None] = None
+    conditions: Union[list, None] = None
+    conditions_policy: Union[str, None] = None
+    actions: Union[dict, None] = None
+    options: Union[dict, None] = None
+    tags: Union[dict, None] = None
+    workspace_id: Union[str, None] = None
+    domain_id: str
+
+
+class CollectorRuleDeleteRequest(BaseModel):
+    collector_rule_id: str
+    workspace_id: Union[str, None] = None
+    domain_id: str
+
+
+class CollectorRuleGetRequest(BaseModel):
+    collector_rule_id: str
+    workspace_id: Union[list, str, None] = None
+    domain_id: str
+
+
+class CollectorRuleChangeOrderRequest(BaseModel):
+    collector_rule_id: str
+    order: int
+    workspace_id: Union[str, None] = None
+    domain_id: str
+
+
+class CollectorRuleSearchQueryRequest(BaseModel):
+    query: Union[dict, None] = None
+    collector_rule_id: Union[str, None] = None
+    name: Union[str, None] = None
+    rule_type: Union[str, None] = None
+    collector_id: Union[str, None] = None
+    workspace_id: Union[list, str, None] = None
+    domain_id: str
+
+
+class CollectorRuleStatQueryRequest(BaseModel):
+    query: dict
+    workspace_id: Union[list, str, None] = None
+    domain_id: str

src/spaceone/inventory_v2/model/collector_rule/response.py

@@ -0,0 +1,36 @@
+from datetime import datetime
+from typing import Union, List
+from pydantic import BaseModel
+
+from spaceone.core import utils
+
+__all__ = ["CollectorRuleResponse", "CollectorRulesResponse"]
+
+
+class CollectorRuleResponse(BaseModel):
+    collector_rule_id: Union[str, None] = None
+    name: Union[str, None] = None
+    rule_type: Union[str, None] = None
+    order: Union[int, None] = None
+    conditions: Union[list, None] = None
+    conditions_policy: Union[str, None] = None
+    actions: Union[dict, None] = None
+    options: Union[dict, None] = None
+    tags: Union[dict, None] = None
+    collector_id: Union[str, None] = None
+    resource_group: Union[str, None] = None
+    workspace_id: Union[str, None] = None
+    domain_id: Union[str, None] = None
+    created_at: Union[datetime, None] = None
+    updated_at: Union[datetime, None] = None
+
+    def dict(self, *args, **kwargs):
+        data = super().dict(*args, **kwargs)
+        data["created_at"] = utils.datetime_to_iso8601(data["created_at"])
+        data["updated_at"] = utils.datetime_to_iso8601(data["updated_at"])
+        return data
+
+
+class CollectorRulesResponse(BaseModel):
+    results: List[CollectorRuleResponse]
+    total_count: int

src/spaceone/inventory_v2/model/namespace/__init__.py

@@ -1 +1 @@
-from spaceone.inventory_v2.model.namespace import *
+from spaceone.inventory_v2.model.namespace import *

src/spaceone/inventory_v2/model/namespace_group/__init__.py

@@ -1 +1 @@
-from spaceone.inventory_v2.model.namespace_group import *
+from spaceone.inventory_v2.model.namespace_group import *

src/spaceone/inventory_v2/service/__init__.py

@@ -1,7 +1,6 @@
 from spaceone.inventory_v2.service.asset_service import AssetService
 from spaceone.inventory_v2.service.asset_type_service import AssetTypeService
+from spaceone.inventory_v2.service.collector_rule_service import CollectorRuleService
 from spaceone.inventory_v2.service.region_service import RegionService
 from spaceone.inventory_v2.service.namespace_group_service import NamespaceGroupService
-from spaceone.inventory_v2.service.namespace_service import NamespaceService   
-
-
+from spaceone.inventory_v2.service.namespace_service import NamespaceService