cartography 0.102.0rc2__py3-none-any.whl → 0.103.0__py3-none-any.whl

cartography/graph/cleanupbuilder.py

@@ -35,18 +35,28 @@ def build_cleanup_queries(node_schema: CartographyNodeSchema) -> List[str]:
     :param node_schema: The given CartographyNodeSchema
     :return: A list of Neo4j queries to clean up nodes and relationships.
     """
-    if not node_schema.sub_resource_relationship and not node_schema.other_relationships:
+    if (
+        not node_schema.sub_resource_relationship
+        and not node_schema.other_relationships
+    ):
         return []
 
     if not node_schema.sub_resource_relationship:
         queries = []
-        other_rels = node_schema.other_relationships.rels if node_schema.other_relationships else []
+        other_rels = (
+            node_schema.other_relationships.rels
+            if node_schema.other_relationships
+            else []
+        )
         for rel in other_rels:
             query = _build_cleanup_rel_query_no_sub_resource(node_schema, rel)
             queries.append(query)
         return queries
 
-    result = _build_cleanup_node_and_rel_queries(node_schema, node_schema.sub_resource_relationship)
+    result = _build_cleanup_node_and_rel_queries(
+        node_schema,
+        node_schema.sub_resource_relationship,
+    )
     if node_schema.other_relationships:
         for rel in node_schema.other_relationships.rels:
             # [0] is the delete node query, [1] is the delete relationship query. We only want the latter.
@@ -57,16 +67,16 @@ def build_cleanup_queries(node_schema: CartographyNodeSchema) -> List[str]:
 
 
 def _build_cleanup_rel_query_no_sub_resource(
-        node_schema: CartographyNodeSchema,
-        selected_relationship: CartographyRelSchema,
+    node_schema: CartographyNodeSchema,
+    selected_relationship: CartographyRelSchema,
 ) -> str:
     """
     Helper function to delete stale relationships for node_schemas that have no sub resource relationship defined.
     """
     if node_schema.sub_resource_relationship:
         raise ValueError(
-            f'Expected {node_schema.label} to not exist. '
-            'This function is intended for node_schemas without sub_resource_relationships.',
+            f"Expected {node_schema.label} to not exist. "
+            "This function is intended for node_schemas without sub_resource_relationships.",
         )
     # Ensure the node is attached to the sub resource and delete the node
     query_template = Template(
@@ -85,8 +95,8 @@ def _build_cleanup_rel_query_no_sub_resource(
 
 
 def _build_cleanup_node_and_rel_queries(
-        node_schema: CartographyNodeSchema,
-        selected_relationship: CartographyRelSchema,
+    node_schema: CartographyNodeSchema,
+    selected_relationship: CartographyRelSchema,
 ) -> List[str]:
     """
     Private function that performs the main string template logic for generating cleanup node and relationship queries.
@@ -129,7 +139,9 @@ def _build_cleanup_node_and_rel_queries(
     ]
     # Now clean up the relationships
     if selected_relationship == node_schema.sub_resource_relationship:
-        _validate_target_node_matcher_for_cleanup_job(node_schema.sub_resource_relationship.target_node_matcher)
+        _validate_target_node_matcher_for_cleanup_job(
+            node_schema.sub_resource_relationship.target_node_matcher,
+        )
         delete_action_clauses.append(
             """
             WHERE s.lastupdated <> $UPDATE_TAG
@@ -159,13 +171,17 @@ def _build_cleanup_node_and_rel_queries(
             node_label=node_schema.label,
             sub_resource_link=sub_resource_link,
             sub_resource_label=node_schema.sub_resource_relationship.target_node_label,
-            match_sub_res_clause=_build_match_clause(node_schema.sub_resource_relationship.target_node_matcher),
+            match_sub_res_clause=_build_match_clause(
+                node_schema.sub_resource_relationship.target_node_matcher,
+            ),
             selected_rel_clause=(
-                "" if selected_relationship == node_schema.sub_resource_relationship
+                ""
+                if selected_relationship == node_schema.sub_resource_relationship
                 else _build_selected_rel_clause(selected_relationship)
             ),
             delete_action_clause=delete_action_clause,
-        ) for delete_action_clause in delete_action_clauses
+        )
+        for delete_action_clause in delete_action_clauses
     ]
 
 
@@ -179,8 +195,12 @@ def _build_selected_rel_clause(selected_relationship: CartographyRelSchema) -> s
         selected_rel_template = Template("<-[r:$SelectedRelLabel]-")
     else:
         selected_rel_template = Template("-[r:$SelectedRelLabel]->")
-    selected_rel = selected_rel_template.safe_substitute(SelectedRelLabel=selected_relationship.rel_label)
-    selected_rel_clause_template = Template("""MATCH (n)$selected_rel(:$other_node_label)""")
+    selected_rel = selected_rel_template.safe_substitute(
+        SelectedRelLabel=selected_relationship.rel_label,
+    )
+    selected_rel_clause_template = Template(
+        """MATCH (n)$selected_rel(:$other_node_label)""",
+    )
     selected_rel_clause = selected_rel_clause_template.safe_substitute(
         selected_rel=selected_rel,
         other_node_label=selected_relationship.target_node_label,

cartography/graph/job.py

@@ -32,7 +32,7 @@ def _get_identifiers(template: string.Template) -> List[str]:
             filter(
                 lambda v: v is not None,
                 (
-                    mo.group('named') or mo.group('braced')
+                    mo.group("named") or mo.group("braced")
                     for mo in template.pattern.finditer(template.template)
                 ),
             ),
@@ -71,7 +71,12 @@ class GraphJob:
     A job that will run against the cartography graph. A job is a sequence of statements which execute sequentially.
     """
 
-    def __init__(self, name: str, statements: List[GraphStatement], short_name: Optional[str] = None):
+    def __init__(
+        self,
+        name: str,
+        statements: List[GraphStatement],
+        short_name: Optional[str] = None,
+    ):
         # E.g. "Okta intel module cleanup"
         self.name = name
         self.statements: List[GraphStatement] = statements
@@ -100,7 +105,11 @@ class GraphJob:
                     e,
                 )
                 raise
-        log_msg = f"Finished job {self.short_name}" if self.short_name else f"Finished job {self.name}"
+        log_msg = (
+            f"Finished job {self.short_name}"
+            if self.short_name
+            else f"Finished job {self.name}"
+        )
         logger.info(log_msg)
 
     def as_dict(self) -> Dict:
@@ -114,7 +123,7 @@ class GraphJob:
         }
 
     @classmethod
-    def from_json(cls, blob: str, short_name: Optional[str] = None) -> 'GraphJob':
+    def from_json(cls, blob: str, short_name: Optional[str] = None) -> "GraphJob":
         """
         Create a job from a JSON blob.
         """
@@ -125,10 +134,10 @@ class GraphJob:
 
     @classmethod
     def from_node_schema(
-            cls,
-            node_schema: CartographyNodeSchema,
-            parameters: Dict[str, Any],
-    ) -> 'GraphJob':
+        cls,
+        node_schema: CartographyNodeSchema,
+        parameters: Dict[str, Any],
+    ) -> "GraphJob":
         """
         Create a cleanup job from a CartographyNodeSchema object.
         For a given node, the fields used in the node_schema.sub_resource_relationship.target_node_node_matcher.keys()
@@ -139,14 +148,14 @@ class GraphJob:
         expected_param_keys: Set[str] = get_parameters(queries)
         actual_param_keys: Set[str] = set(parameters.keys())
         # Hacky, but LIMIT_SIZE is specified by default in cartography.graph.statement, so we exclude it from validation
-        actual_param_keys.add('LIMIT_SIZE')
+        actual_param_keys.add("LIMIT_SIZE")
 
         missing_params: Set[str] = expected_param_keys - actual_param_keys
 
         if missing_params:
             raise ValueError(
                 f'GraphJob is missing the following expected query parameters: "{missing_params}". Please check the '
-                f'value passed to `parameters`.',
+                f"value passed to `parameters`.",
             )
 
         statements: List[GraphStatement] = [
@@ -157,7 +166,8 @@ class GraphJob:
                 iterationsize=100,
                 parent_job_name=node_schema.label,
                 parent_job_sequence_num=idx,
-            ) for idx, query in enumerate(queries, start=1)
+            )
+            for idx, query in enumerate(queries, start=1)
         ]
 
         return cls(
@@ -167,7 +177,7 @@ class GraphJob:
         )
 
     @classmethod
-    def from_json_file(cls, file_path: Union[str, Path]) -> 'GraphJob':
+    def from_json_file(cls, file_path: Union[str, Path]) -> "GraphJob":
         """
         Create a job from a JSON file.
         """
@@ -175,13 +185,20 @@ class GraphJob:
             data: Dict = json.load(j_file)
 
         job_shortname: str = get_job_shortname(file_path)
-        statements: List[GraphStatement] = _get_statements_from_json(data, job_shortname)
+        statements: List[GraphStatement] = _get_statements_from_json(
+            data,
+            job_shortname,
+        )
         name: str = data["name"]
         return cls(name, statements, job_shortname)
 
     @classmethod
     def run_from_json(
-        cls, neo4j_session: neo4j.Session, blob: str, parameters: Dict, short_name: Optional[str] = None,
+        cls,
+        neo4j_session: neo4j.Session,
+        blob: str,
+        parameters: Dict,
+        short_name: Optional[str] = None,
     ) -> None:
         """
         Run a job from a JSON blob. This will deserialize the job and execute all statements sequentially.
@@ -194,7 +211,12 @@ class GraphJob:
         job.run(neo4j_session)
 
     @classmethod
-    def run_from_json_file(cls, file_path: Union[str, Path], neo4j_session: neo4j.Session, parameters: Dict) -> None:
+    def run_from_json_file(
+        cls,
+        file_path: Union[str, Path],
+        neo4j_session: neo4j.Session,
+        parameters: Dict,
+    ) -> None:
         """
         Run a job from a JSON file. This will deserialize the job and execute all statements sequentially.
         """
@@ -207,14 +229,21 @@ class GraphJob:
         job.run(neo4j_session)
 
 
-def _get_statements_from_json(blob: Dict, short_job_name: Optional[str] = None) -> List[GraphStatement]:
+def _get_statements_from_json(
+    blob: Dict,
+    short_job_name: Optional[str] = None,
+) -> List[GraphStatement]:
     """
     Deserialize all statements from the JSON blob.
     """
     statements: List[GraphStatement] = []
     for i, statement_data in enumerate(blob["statements"]):
         # i+1 to make it 1-based and not 0-based to help with log readability
-        statement: GraphStatement = GraphStatement.create_from_json(statement_data, short_job_name, i + 1)
+        statement: GraphStatement = GraphStatement.create_from_json(
+            statement_data,
+            short_job_name,
+            i + 1,
+        )
         statements.append(statement)
 
     return statements