def __init__(self, datasource, dataset):
self.csv_df = pd.DataFrame()
# create the Extract object
extract_obj = Extract()
if datasource == 'api':
self.data = extract_obj.get_api_data(dataset)
func_name = datasource + "_" + dataset
getattr(self, func_name)()
elif datasource == 'csv':
self.data = extract_obj.get_csv_data(dataset)
func_name = datasource + "_" + dataset
getattr(self, func_name)()
else:
print('Unknown data source!!! Please try again...')
def __init__(self, data_source, data_set):
extract_obj = Extract()
func_name = f"{data_source}_{data_set}"
if data_source == 'api':
self.data = extract_obj.get_api_data(data_set)
elif data_source == 'csv':
self.data = extract_obj.get_csv_data(data_set)
else:
print('Unkown Data Source')
return
# getattr function takes in function name of class and calls it.
getattr(self, func_name)()
