traning_zone/data_import/data_importation.py

import pandas as pd
from traning_zone.requetes_classe.requetes import *

from dotenv import load_dotenv
import os
load_dotenv()
project_id = os.getenv('PROJECT_ID')



def data_import(*args):
    """This function import data from bigquery

    Returns:
        data: dataframe with specific list of bem_class_key
    """
    classes ='","'.join(args)
    query = Query(classes)
    data = pd.read_gbq(query,project_id=project_id)
    print("data shape", data.shape)
    return data