class Server:
"""Flask server for anonymizer."""
def __init__(self):
fileConfig(Path(Path(__file__).parent, LOGGING_CONF_FILE))
self.logger = logging.getLogger("presidio-anonymizer")
self.logger.setLevel(os.environ.get("LOG_LEVEL", self.logger.level))
self.app = Flask(__name__)
self.logger.info("Starting anonymizer engine")
self.engine = AnonymizerEngine()
self.decryptor = AnonymizerDecryptor()
self.logger.info(WELCOME_MESSAGE)
@self.app.route("/health")
def health() -> str:
"""Return basic health probe result."""
return "Presidio Anonymizer service is up"
@self.app.route("/anonymize", methods=["POST"])
def anonymize():
content = request.get_json()
if not content:
return ErrorResponse("Invalid request json").to_json(), 400
anonymizers_config = AnonymizerRequest.get_anonymizer_configs_from_json(
content)
analyzer_results = AnonymizerRequest.handle_analyzer_results_json(
content)
anoymizer_result = self.engine.anonymize(
text=content.get("text"),
analyzer_results=analyzer_results,
anonymizers_config=anonymizers_config,
)
return anoymizer_result.to_json()
@self.app.route("/decrypt", methods=["POST"])
def decrypt() -> Union[str, Tuple[str, int]]:
content = request.get_json()
if not content:
return ErrorResponse("Invalid request json").to_json(), 400
decrypted_text = self.decryptor.decrypt(key=content.get("key"),
text=content.get("text"))
return jsonify(result=decrypted_text)
@self.app.route("/anonymizers", methods=["GET"])
def anonymizers() -> Tuple[str, int]:
"""Return a list of supported anonymizers."""
return json.dumps(self.engine.get_anonymizers()), 200
@self.app.errorhandler(InvalidParamException)
def invalid_param(err):
self.logger.warning(
f"failed to anonymize text with validation error: {err.err_msg}"
)
return ErrorResponse(err.err_msg).to_json(), 422
@self.app.errorhandler(Exception)
def server_error(e):
self.logger.error(f"A fatal error occurred during execution: {e}")
return ErrorResponse("Internal server error").to_json(), 500
def test_given_request_anonymizers_return_list():
engine = AnonymizerEngine()
expected_list = ["hash", "mask", "redact", "replace", "encrypt"]
anon_list = engine.get_anonymizers()
assert anon_list == expected_list
class Server:
"""Flask server for anonymizer."""
def __init__(self):
fileConfig(Path(Path(__file__).parent, LOGGING_CONF_FILE))
self.logger = logging.getLogger("presidio-anonymizer")
self.logger.setLevel(os.environ.get("LOG_LEVEL", self.logger.level))
self.app = Flask(__name__)
self.logger.info("Starting anonymizer engine")
self.anonymizer = AnonymizerEngine()
self.deanonymize = DeanonymizeEngine()
self.logger.info(WELCOME_MESSAGE)
@self.app.route("/health")
def health() -> str:
"""Return basic health probe result."""
return "Presidio Anonymizer service is up"
@self.app.route("/anonymize", methods=["POST"])
def anonymize() -> Response:
content = request.get_json()
if not content:
raise BadRequest("Invalid request json")
anonymizers_config = AppEntitiesConvertor.operators_config_from_json(
content.get("anonymizers"))
if AppEntitiesConvertor.check_custom_operator(anonymizers_config):
raise BadRequest("Custom type anonymizer is not supported")
analyzer_results = AppEntitiesConvertor.analyzer_results_from_json(
content.get("analyzer_results"))
anoymizer_result = self.anonymizer.anonymize(
text=content.get("text"),
analyzer_results=analyzer_results,
operators=anonymizers_config,
)
return Response(anoymizer_result.to_json(),
mimetype="application/json")
@self.app.route("/deanonymize", methods=["POST"])
def deanonymize() -> Response:
content = request.get_json()
if not content:
raise BadRequest("Invalid request json")
text = content.get("text")
deanonymize_entities = AppEntitiesConvertor.deanonymize_entities_from_json(
content)
deanonymize_config = AppEntitiesConvertor.operators_config_from_json(
content.get("deanonymizers"))
deanonymized_response = self.deanonymize.deanonymize(
text=text,
entities=deanonymize_entities,
operators=deanonymize_config)
return Response(deanonymized_response.to_json(),
mimetype="application/json")
@self.app.route("/anonymizers", methods=["GET"])
def anonymizers():
"""Return a list of supported anonymizers."""
return jsonify(self.anonymizer.get_anonymizers())
@self.app.route("/deanonymizers", methods=["GET"])
def deanonymizers():
"""Return a list of supported deanonymizers."""
return jsonify(self.deanonymize.get_deanonymizers())
@self.app.errorhandler(InvalidParamException)
def invalid_param(err):
self.logger.warning(
f"Request failed with parameter validation error: {err.err_msg}"
)
return jsonify(error=err.err_msg), 422
@self.app.errorhandler(HTTPException)
def http_exception(e):
return jsonify(error=e.description), e.code
@self.app.errorhandler(Exception)
def server_error(e):
self.logger.error(f"A fatal error occurred during execution: {e}")
return jsonify(error="Internal server error"), 500
