import pandas as pd
import numpy as np
import os

def get_files_from_dir(path: str) -> list[str]:
    return [f for f in os.listdir(path) if os.path.isfile(os.path.join(path,f)) and not f.startswith('.')]

def get_first_valid_return_index(series: pd.Series) -> int:
    double_nested_results = np.where(np.logical_and(series != 0, np.logical_not(np.isnan(series))))
    if len(double_nested_results) == 0:
        return 0
    nested_result = double_nested_results[0]
    if len(nested_result) == 0:
        return 0
    return nested_result[0]

def flatten(list_of_lists: list) -> list:
    return [item for sublist in list_of_lists for item in sublist]

def weighted_average(df: pd.DataFrame, weights_source: str) -> pd.DataFrame:
    if df.shape[0] == 0:
        return df
    mean_df = df.iloc[:,0]
    weights = df.loc[weights_source]

    for i, row in df.iterrows():
        if i == weights_source: continue
        mean_df.loc[i] = (row * weights).sum() / df.loc[weights_source].sum()

    return mean_df

def deduplicate_indexes(df: pd.DataFrame) -> pd.DataFrame: return df[~df.index.duplicated(keep='last')]

def drop_columns_if_exist(df: pd.DataFrame, columns: list) -> pd.DataFrame:
    for column in columns:
        if column in df.columns:
            df = df.drop(column, axis=1)
    return df