import os
from typing import Dict, Union
[docs]
def get_max_parquet_loc(
root: str, claim_type: str, state: str, year: Union[int, str]
) -> str:
return os.path.join(
root,
"medicaid",
str(year),
state.upper(),
"max",
claim_type.lower(),
"parquet",
)
[docs]
def get_taf_parquet_loc(
root: str, claim_type: str, state: str, year: Union[int, str]
) -> Dict[str, str]:
dct_fileloc: Dict[str, str] = {}
data_folder = os.path.join(
root, "medicaid", str(year), state.upper(), "taf"
)
if claim_type in ["ip", "ot", "lt"]:
dct_fileloc["base"] = os.path.join(
data_folder, claim_type, f"{claim_type.lower()}h", "parquet"
)
dct_fileloc["line"] = os.path.join(
data_folder, claim_type, f"{claim_type.lower()}l", "parquet"
)
dct_fileloc["occurrence_code"] = os.path.join(
data_folder, claim_type, f"{claim_type.lower()}occr", "parquet"
)
dct_fileloc["occurrence_code"] = os.path.join(
data_folder, claim_type, f"{claim_type.lower()}occr", "parquet"
)
dct_fileloc["base_diag_codes"] = os.path.join(
data_folder, claim_type, f"{claim_type.lower()}dx", "parquet"
)
if claim_type == "ps":
dct_fileloc["dates"] = os.path.join(
data_folder, "de", "dedts", "parquet"
)
dct_fileloc["base"] = os.path.join(
data_folder, "de", "debse", "parquet"
)
dct_fileloc["managed_care"] = os.path.join(
data_folder, "de", "demc", "parquet"
)
dct_fileloc["disability"] = os.path.join(
data_folder, "de", "dedsb", "parquet"
)
dct_fileloc["mfp"] = os.path.join(
data_folder, "de", "demfp", "parquet"
)
dct_fileloc["waiver"] = os.path.join(
data_folder, "de", "dewvr", "parquet"
)
dct_fileloc["home_health"] = os.path.join(
data_folder, "de", "dehsp", "parquet"
)
dct_fileloc["diag_and_ndc_codes"] = os.path.join(
data_folder, "de", "dedxndc", "parquet"
)
if claim_type in ["rx"]:
dct_fileloc["base"] = os.path.join(
data_folder, claim_type, f"{claim_type.lower()}h", "parquet"
)
dct_fileloc["line"] = os.path.join(
data_folder, claim_type, f"{claim_type.lower()}l", "parquet"
)
if claim_type in ["ip", "ot", "lt", "rx"]:
dct_fileloc["line_ndc_codes"] = os.path.join(
data_folder, claim_type, f"{claim_type.lower()}ndc", "parquet"
)
return dct_fileloc