staff_info.py

# pylint: disable=import-error

"""
Provide information about an specific employee based on the user's query

Functions:
    employee_info(query: str) -> str

Author: @dilshan-h (https://github.com/dilshan-h)
"""

import os
from functools import lru_cache
from cryptography.fernet import Fernet
from dotenv import load_dotenv
from thefuzz import fuzz

load_dotenv()

BASE_DIR: str = os.path.dirname(os.path.abspath(__file__))
data_path: str = os.path.join(BASE_DIR, "DATA", "staff_info.txt.crypt")
fernet = Fernet(os.environ["SECRET_KEY"])


@lru_cache(maxsize=16)
def employee_info(query: str) -> str:
    """Construct info and return message body text"""
    found_info: list = []
    message_body: str = "<b>🔎 Here's what I have found:</b>\n\n"

    with open(file=data_path, mode="rb") as data_file:
        stream = data_file.read()
        decrypted_data = fernet.decrypt(stream).decode().strip()
        employees: list = decrypted_data.split("\n\n")
        for employee in employees:
            if fuzz.partial_ratio(query.lower(), employee.lower()) > 80:
                found_info.append(employee)

    if not found_info:
        return "😕 No matching data found!\nCan you try again with a different query?"
    if len(found_info) > 3:
        return (
            f"😕 <b>'{query}' found in {len(found_info)} places!"
            "</b>\nCan you narrow your search query?"
        )
    for info in found_info:
        message_body += f"{info}\n\n"

    return message_body