model_app.py

from typing import Any
from pandas import DataFrame
from typing import Dict
from typing import List
from typing import Tuple

import os
import yaml

import src.model_classes as mc 

package_dir = os.path.dirname(os.path.abspath(__file__))
config_fp = os.path.join(package_dir, "config_app.yaml")

with open(config_fp) as f:
    config_app = yaml.safe_load(f)
    
class Model:
    
    """
    
    """
    
    def __init__(
        self,
        df: DataFrame,
        config: Dict[str, Any]
    ):
        
        """
        
        """
        self.df = df
        self.config = config

    def prep(
        self
    ) -> Tuple[DataFrame, List[float]]:
               
        """
        
        """
        
        df = self.df
        config = self.config
        
        # ingest
        prep = mc.Ingest(config)
        df_prep: DataFrame = prep.run_harmonize(df)

        # transform
        trans = mc.Transform(df=df_prep)
        tbl: List[float] = trans.run_build_cont_table()
        
        return df_prep, tbl
    
    def analysis(
        self,
        df_prep: DataFrame,
        tbl: List[float]
    ) -> DataFrame:
                     
        """

        """
        
        config = self.config

        combined_config = {**config, **config_app}

        stats = mc.StatsTesting2x2Cont(
            config=combined_config,
            tbl=tbl,
            df=df_prep
        )
        df_result: DataFrame = stats.run_testing()
        
        return df_result
    
    
if __name__ == "__main__":
    
    model = Model()
    
    df_prep, tbl = model.prep()
    
    df_result = model.analysis(df_prep.copy(), tbl)