Module pii_codex.services.adapters.detection_adapters.azure_detection_adapter

Classes

class AzurePIIDetectionAdapter

Expand source code

class AzurePIIDetectionAdapter(BasePIIDetectionAdapter):
    def convert_analyzed_item(self, pii_detection: dict):
        """
        Converts a detection result into a collection of DetectionResultItem

        @param pii_detection: dict
        @return: List[DetectionResultItem]
        """
        return [
            DetectionResultItem(
                entity_type=PII_MAPPER.convert_azure_pii_to_common_pii_type(
                    entity["category"]
                ).name,
                score=entity["confidence_score"],
                start=entity["offset"],
                end=entity["offset"] + entity["length"],
            )
            for entity in pii_detection["entities"]
        ]

    def convert_analyzed_collection(
        self, pii_detections: List[dict]
    ) -> List[DetectionResult]:
        """
        Converts a collection of detection results to a collection of DetectionResult.

        @param pii_detections: List[dict]
        @return: List[DetectionResultItem]
        """
        detection_results: List[DetectionResult] = []
        for i, detection in enumerate(pii_detections):
            # Return results in formatted Analysis Result List object
            detection_results.append(
                DetectionResult(
                    index=i,
                    detections=self.convert_analyzed_item(pii_detection=detection),
                )
            )

        return detection_results

Ancestors

• BasePIIDetectionAdapter

Methods

def convert_analyzed_collection(self, pii_detections: List[dict]) ‑> List[DetectionResult]

Converts a collection of detection results to a collection of DetectionResult.

@param pii_detections: List[dict] @return: List[DetectionResultItem]

Expand source code

def convert_analyzed_collection(
    self, pii_detections: List[dict]
) -> List[DetectionResult]:
    """
    Converts a collection of detection results to a collection of DetectionResult.

    @param pii_detections: List[dict]
    @return: List[DetectionResultItem]
    """
    detection_results: List[DetectionResult] = []
    for i, detection in enumerate(pii_detections):
        # Return results in formatted Analysis Result List object
        detection_results.append(
            DetectionResult(
                index=i,
                detections=self.convert_analyzed_item(pii_detection=detection),
            )
        )

    return detection_results

Inherited members

• BasePIIDetectionAdapter:
  • convert_analyzed_item