idvpackage 3.0.10__py3-none-any.whl → 3.0.12__py3-none-any.whl

idvpackage/genai_utils.py

@@ -1,309 +0,0 @@
-import re
-from datetime import timedelta
-from dateutil.parser import parse
-from pydantic import BaseModel
-from typing import Type
-from datetime import datetime
-
-
-def find_gender_from_back(text):
-    gender = ""
-    gender_pattern = r"(\d)([A-Za-z])(\d)"
-    gender_match = re.search(gender_pattern, text)
-    if gender_match:
-        gender = gender_match.group(2)
-
-    if not gender:
-        gender_pattern = r"(\d)([MFmf])(\d)"
-        gender_match = re.search(gender_pattern, text)
-        if gender_match:
-            gender = gender_match.group(2)
-
-    return gender
-
-
-def is_valid_date(date_str):
-    """Returns True if the string can be parsed as a valid date, regardless of format."""
-    try:
-        parse(date_str, fuzzy=False)
-        return True
-    except (ValueError, TypeError):
-        return False
-
-
-def is_expiry_issue_diff_valid(issue_date_str, expiry_date_str, time_period):
-    """Check if expiry date = issue date + 5 years - 1 day"""
-    if is_valid_date(issue_date_str) and is_valid_date(expiry_date_str):
-        issue_date = datetime.strptime(issue_date_str, "%Y/%m/%d")
-        expiry_date = datetime.strptime(expiry_date_str, "%Y/%m/%d")
-        expected_expiry = issue_date.replace(
-            year=issue_date.year + time_period
-        ) - timedelta(days=1)
-        return expiry_date == expected_expiry
-    return False
-
-
-def is_mrz_dob_mrz_field_match(dob_str, mrz_line2):
-    """Check if DOB in MRZ matches the printed DOB"""
-    dob = datetime.strptime(dob_str, "%Y/%m/%d")
-    mrz_dob_raw = mrz_line2[:6]  # First 6 characters (YYMMDD)
-    current_year_last2 = int(str(datetime.today().year)[-2:])
-    year_prefix = "19" if int(mrz_dob_raw[:2]) > current_year_last2 else "20"
-    mrz_dob = datetime.strptime(year_prefix + mrz_dob_raw, "%Y%m%d")
-    return mrz_dob == dob
-
-
-def is_age_18_above(dob_str):
-    """
-    Check if the person is 18 or older as of today
-
-    Parameters:
-    dob_str (str): Date of birth in 'YYYY-MM-DD', 'DD.MM.YYYY', 'YYYY/MM/DD', or 'DD/MM/YYYY' format.
-
-    Returns:
-    bool: True if the person is 18 or older, False otherwise.
-    """
-    date_formats = ["%Y-%m-%d", "%d.%m.%Y", "%Y/%m/%d", "%d/%m/%Y"]
-
-    for fmt in date_formats:
-        try:
-            dob = datetime.strptime(dob_str, fmt)
-            today = datetime.today()
-            age = (
-                today.year
-                - dob.year
-                - ((today.month, today.day) < (dob.month, dob.day))
-            )
-            return age >= 18
-        except ValueError:
-            continue
-
-    return "invalid_format"
-
-
-def is_age_less_than_100(dob_str):
-    """
-    Check if the person is less than 100 years old as of today.
-
-    Parameters:
-    dob_str (str): Date of birth in 'YYYY-MM-DD', 'DD.MM.YYYY', 'YYYY/MM/DD', or 'DD/MM/YYYY' format.
-
-    Returns:
-    bool: True if the person is less than 100 years old, False otherwise.
-    """
-    date_formats = ["%Y-%m-%d", "%d.%m.%Y", "%Y/%m/%d", "%d/%m/%Y"]
-
-    for fmt in date_formats:
-        try:
-            dob = datetime.strptime(dob_str, fmt)
-            today = datetime.today()
-            age = (
-                today.year
-                - dob.year
-                - ((today.month, today.day) < (dob.month, dob.day))
-            )
-            return age <= 100
-        except ValueError:
-            continue
-
-    return "invalid_format"
-
-
-def is_expired_id(expiry_date):
-    """
-    Checks if an ID is expired.
-
-    Parameters:
-    expiry_date (str): Expiry date in 'YYYY-MM-DD', 'DD.MM.YYYY', or 'YYYY/MM/DD' format.
-
-    Returns:
-    bool: True if the passport is expired, False otherwise.
-    """
-    date_formats = ["%d-%m-%Y","%Y-%m-%d", "%d.%m.%Y", "%Y/%m/%d","%d/%m/%Y"]
-
-    for fmt in date_formats:
-        try:
-            expiry = datetime.strptime(expiry_date, fmt).date()
-            today = datetime.today().date()
-            return expiry < today
-        except ValueError:
-            continue
-
-    raise ValueError(
-        "Invalid date format. Expected 'YYYY-MM-DD', 'DD.MM.YYYY', or 'YYYY/MM/DD'."
-    )
-
-
-def parse_yymmdd(yymmdd_str):
-    """
-    Converts a 'YYMMDD' string to a 'YYYY-MM-DD' formatted string.
-    Assumes years < 50 are 2000s, otherwise 1900s.
-
-    Parameters:
-    yymmdd_str (str): A string in 'YYMMDD' format.
-
-    Returns:
-    str: A date string in 'YYYY-MM-DD' format.
-    """
-    if len(yymmdd_str) != 6 or not yymmdd_str.isdigit():
-        raise ValueError("Invalid YYMMDD format")
-
-    try:
-        parsed_date = datetime.strptime(yymmdd_str, "%y%m%d")
-        if parsed_date.year < 1950:
-            parsed_date = parsed_date.replace(year=parsed_date.year + 100)
-        return parsed_date.strftime("%Y-%m-%d")
-    except ValueError:
-        raise ValueError(f"Could not parse YYMMDD string: {yymmdd_str}")
-
-
-def convert_pydantic_to_openai_function2(model: Type[BaseModel]) -> dict:
-    """
-    Convert a Pydantic model into OpenAI function calling format,
-    inferring the function name and description from the model.
-
-    - Function name is derived from the class name in snake_case.
-    - Description is taken from the class docstring.
-
-    Args:
-        model (BaseModel): The Pydantic model class.
-
-    Returns:
-        dict: A dictionary formatted for OpenAI function calling.
-    """
-    import re
-
-    def camel_to_snake(name: str) -> str:
-        return re.sub(r"(?<!^)(?=[A-Z])", "_", name).lower()
-
-    return {
-        "name": camel_to_snake(model.__name__),
-        "description": model.__doc__ or "No description provided.",
-        "parameters": model.schema(),
-    }
-
-
-def fix_family_number(ocr_output):
-    ocr_output = ocr_output.replace(" ", "")
-
-    if len(ocr_output) == 18:
-        return ocr_output
-    # Step 1: Remove all non-digit characters just in case
-    digits_only = "".join(filter(str.isdigit, ocr_output))
-
-    # Step 3: Insert 'L' at position 4 (index 4)
-    before_L = digits_only[:4]
-    after_L = digits_only[4:]
-
-    if after_L[0] != "0":
-        fixed = before_L + "E" + after_L
-        return fixed
-
-    fixed = before_L + "L" + after_L
-
-    if len(fixed) == 18:
-        return fixed
-
-    # Step 4: After 'L', find where the zeros end and non-zero digits begin
-    after_L_part = fixed[5:]  # characters after 'L'
-    zero_count = 0
-
-    for ch in after_L_part:
-        if ch == "0":
-            zero_count += 1
-        else:
-            break  # stop at the first non-zero digit
-
-    # Step 5: Insert 'M' just after the zeros (before the first non-zero digit)
-    insertion_index = 5 + zero_count  # 5 = index right after L
-    fixed = fixed[:insertion_index] + "M" + fixed[insertion_index:]
-
-    return fixed
-
-
-def convert_dob_to_standard(date_str):
-    input_formats = ["%Y-%m-%d", "%d.%m.%Y", "%Y/%m/%d", "%d/%m/%Y"]
-
-    for fmt in input_formats:
-        try:
-            parsed_date = datetime.strptime(date_str, fmt)
-            return parsed_date.strftime("%d/%m/%Y")
-        except ValueError:
-            continue
-    raise ValueError("Date format not recognized.")
-
-
-def check_irq_back_mrz1_format(s):
-    placeholder = s.endswith("<<<") and all(c != "<" for c in s[:-3])
-    pattern = r"^IDIRQ([A-Z]{2}|[A-Z][0-9])[0-9]{20}<<<$"
-    if re.fullmatch(pattern, s) and len(s) == 30:
-        return s
-    else:
-        mrz1_stripped = s.strip("<")
-        if len(mrz1_stripped) == 27:
-            mrz1 = mrz1_stripped + "<<<"
-            if re.fullmatch(pattern, mrz1) and len(mrz1) == 30:
-                return mrz1
-            else:
-                raise False
-
-    return False
-
-
-def check_irq_back_mrz3_format(s):
-    parts = [part for part in s.strip("<").split("<") if part]
-
-    if len(parts) == 2:
-        if re.fullmatch(r"[A-Z]+<<[A-Z]+<*", s) and len(s) == 30:
-            return s
-        else:
-            mrz3 = parts[0] + "<<" + parts[1]
-            mrz3 = mrz3.ljust(30, "<")
-            if re.fullmatch(r"[A-Z]+<<[A-Z]+<*", mrz3) and len(mrz3) == 30:
-                return mrz3
-            else:
-                return False
-    elif len(parts) == 1:
-        if re.fullmatch(r"<<[A-Z]+<*", s) and len(s) == 30:
-            return s
-        else:
-            mrz3 = "<<" + parts[0]
-            mrz3 = mrz3.ljust(30, "<")
-            if re.fullmatch(r"<<[A-Z]+<*", mrz3) and len(mrz3) == 30:
-                return mrz3
-            else:
-                return False
-
-
-def check_irq_back_mrz2_format(s):
-    pattern = r"^\d{7}[MF]\d{7}[A-Z]{3}<{11}\d$"
-    if re.fullmatch(pattern, s) and len(s) == 30:
-        return s
-    if s[-1].isdigit():
-        # Manually add <<< and see if that fixes mrz line2.
-        head = s[:18]  # 6 DOB + 1 + 1 Gender + 6 Expiry + 1 + 3 'IRQ' = 18
-        tail_digit = s[-1]
-        mrz2 = head + ("<" * 11) + tail_digit
-        if re.fullmatch(pattern, mrz2) and len(mrz2) == 30:
-            return mrz2
-        else:
-            return False
-    else:
-        return False
-
-
-def normalise_dates_for_nfc(date_str):
-    """
-    Convert date from 'YYYY/MM/DD' to 'DD/MM/YYYY' format for NFC usage.
-
-    Parameters:
-    date_str (str): Date in 'YYYY/MM/DD' format.
-
-    Returns:
-    str: Date in 'DD/MM/YYYY' format.
-    """
-    try:
-        dt = datetime.strptime(date_str, "%Y/%m/%d")
-        return dt.strftime("%d/%m/%Y")
-    except Exception:
-        raise ValueError("Input date must be in 'YYYY/MM/DD' format.")