Merge pull request #372 from code-kern-ai/pii-cognition-mappings

Added congition mapping as well as groupings to all PII bricks

Author: LeonardPuettmannKern

extractors/dates_and_times/date_extraction/config.py

@@ -15,8 +15,12 @@ def get_config():
         available_for=["refinery", "common"],
         part_of_group=[
             "dates_and_times",
+            "personal_identifiers"
         ],  # first entry should be parent directory
         # bricks integrator information
+        cognition_init_mapping={
+            "@@LABEL@@": "Date"
+        },
         integrator_inputs={
             "name": "date_extraction",
             "refineryDataType": RefineryDataType.TEXT.value,

extractors/dates_and_times/time_extraction/config.py

@@ -15,8 +15,12 @@ def get_config():
         available_for=["refinery", "common"],
         part_of_group=[
             "dates_and_times",
+            "personal_identifiers"
         ],  # first entry should be parent directory
         # bricks integrator information
+        cognition_init_mapping={
+            "@@LABEL@@": "Time"
+        },
         integrator_inputs={
             "name": "time_extraction",
             "refineryDataType": RefineryDataType.TEXT.value,

extractors/numbers/ip_extraction/code_snippet_refinery.md

@@ -2,7 +2,7 @@
 import re
 
 ATTRIBUTE: str = "text" # only text attributes
-LABEL: str = "IP-address"
+LABEL: str = "IP Address"
 
 def ip_extraction(record):
     regex = re.compile(r"\b\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}\b")

extractors/numbers/ip_extraction/config.py

@@ -15,8 +15,12 @@ def get_config():
         available_for=["refinery", "common"],
         part_of_group=[
             "numbers",
+            "personal_identifiers"
         ],  # first entry should be parent directory
         # bricks integrator information   
+        cognition_init_mapping={
+            "@@LABEL@@": "IP Address"
+        },
         integrator_inputs={
             "name": "ip_extraction",
             "refineryDataType": RefineryDataType.TEXT.value,
@@ -31,7 +35,7 @@ def get_config():
                 },
                 "LABEL": {
                     "selectionType": SelectionType.CHOICE.value,
-                    "defaultValue": "IP-address",
+                    "defaultValue": "IP Address",
                     "optional": "false",
                     "addInfo": [
                         BricksVariableType.LABEL.value,

extractors/numbers/phone_number_extraction/config.py

@@ -15,8 +15,12 @@ def get_config():
         available_for=["refinery", "common"],
         part_of_group=[
             "numbers",
+            "personal_identifiers"
         ],  # first entry should be parent directory
         # bricks integrator information
+        cognition_init_mapping={
+            "@@LABEL@@": "Phone number"
+        },
         integrator_inputs={
             "name": "phone_number_extraction",
             "refineryDataType": RefineryDataType.TEXT.value,

extractors/paths/url_extraction/config.py

@@ -15,8 +15,12 @@ def get_config():
         available_for=["refinery", "common"],
         part_of_group=[
             "paths",
+            "personal_identifiers"
         ],  # first entry should be parent directory
         # bricks integrator information
+        cognition_init_mapping={
+            "@@LABEL@@": "URL"
+        },
         integrator_inputs={
             "name": "url_extraction",
             "refineryDataType": RefineryDataType.TEXT.value,

extractors/personal_identifiers/address_extraction/config.py

@@ -18,6 +18,9 @@ def get_config():
             "personal_identifiers",
         ],  # first entry should be parent directory
         # bricks integrator information
+        cognition_init_mapping={
+            "@@LABEL@@": "Address"
+        },
         integrator_inputs={
             "name": "address_extraction",
             "refineryDataType": RefineryDataType.TEXT.value,

extractors/personal_identifiers/email_extraction/config.py

@@ -17,6 +17,9 @@ def get_config():
             "personal_identifiers",
         ],  # first entry should be parent directory
         # bricks integrator information
+        cognition_init_mapping={
+            "@@LABEL@@": "E-Mail"
+        },
         integrator_inputs={
             "name": "email_extraction",
             "refineryDataType": RefineryDataType.TEXT.value,

extractors/personal_identifiers/person_extraction/config.py

@@ -16,7 +16,10 @@ def get_config():
         part_of_group=[
             "personal_identifiers",
         ],  # first entry should be parent directory
-        # bricks integrator information        
+        # bricks integrator information  
+        cognition_init_mapping={
+            "@@LABEL@@": "Person"
+        },
         integrator_inputs={
             "name": "person_extraction",
             "refineryDataType": RefineryDataType.TEXT.value,

extractors/personal_identifiers/zipcode_extraction/config.py

@@ -17,6 +17,9 @@ def get_config():
             "personal_identifiers",
         ],  # first entry should be parent directory
         # bricks integrator information
+        cognition_init_mapping={
+            "@@LABEL@@": "Zip code"
+        },
         integrator_inputs={
             "name": "zipcode_extraction",
             "refineryDataType": RefineryDataType.TEXT.value,