from web_scraping.fundamentus.src.dao.hadoop_hdfs.ReadHDFS import ReadHDFS


class IndicadoresEmpresaHDFS(ReadHDFS):

    def __init__(self, url_complementar = "fundamentus/detalhes/indicadores_empresa"):
        self.__app_name = "write_read_fundamentus_hdfs"
        super().__init__(self.__app_name, url_complementar)

    def buscar_dados_empresa(self, papel, data=None):
        dados_spark_df = self._ler_dados_spark_dataframe_no_hdfs()
        filtro_papel = "Papel == '%s'" % papel
        filtro_data = "Data_ult_cot == '%s'" % data
        indicadores_empresa = dados_spark_df.filter(filtro_papel).filter(filtro_data).collect()
        return indicadores_empresa
