❇️ Add hooks to MelusineRegex

melusine/base.py

@@ -19,7 +19,7 @@
 import logging
 import re
 from abc import ABC, abstractmethod
-from typing import Any, Callable, Dict, Iterable, TypeVar, Union
+from typing import Any, Callable, Dict, Iterable, List, TypeVar, Union
 
 import pandas as pd
 from sklearn.base import BaseEstimator, TransformerMixin
@@ -308,7 +308,7 @@ class MissingFieldError(Exception):
     """
 
 
-MatchData = dict[str, list[dict[str, Any]]]
+MatchData = Dict[str, List[Dict[str, Any]]]
 
 
 class MelusineRegex(ABC):
@@ -530,7 +530,7 @@ def describe(self, text: str, position: bool = False) -> None:
             position: If True, print regex match start and stop positions.
         """
 
-        def _describe_match_field(match_field_data: dict[str, list[dict[str, Any]]]) -> None:
+        def _describe_match_field(match_field_data: Dict[str, List[Dict[str, Any]]]) -> None:
             """
             Format and print result description text.
 
@@ -609,7 +609,7 @@ def pre_match_hook(self, text: str) -> str:
         """
         return text
 
-    def post_match_hook(self, match_dict: dict[str, Any]) -> dict[str, Any]:
+    def post_match_hook(self, match_dict: Dict[str, Any]) -> Dict[str, Any]:
         """
         Hook to run after the Melusine regex match.
 

tests/base/test_melusine_regex.py

@@ -1,4 +1,4 @@
-from typing import Dict, List, Optional, Union, Any
+from typing import Any, Dict, List, Optional, Union
 
 import pytest
 
@@ -169,7 +169,7 @@ def test_pre_match_hook():
 
 class PostMatchHookVirusRegex(VirusRegex):
 
-    def post_match_hook(self, match_dict: dict[str, Any]) -> dict[str, Any]:
+    def post_match_hook(self, match_dict: Dict[str, Any]) -> Dict[str, Any]:
         """Test custom post processing of match data"""
         if match_dict[self.MATCH_RESULT] is True:
             if "NEUTRAL_MEDICAL_VIRUS" in match_dict[self.NEUTRAL_MATCH_FIELD] and "NEUTRAL_INSECT" in match_dict[self.NEUTRAL_MATCH_FIELD]: