From abe711685030d22641b203011f645baea6613aca Mon Sep 17 00:00:00 2001 From: marinellirubens Date: Sun, 11 Jun 2023 22:24:19 -0300 Subject: [PATCH] small type hints change --- magic_formula/main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/magic_formula/main.py b/magic_formula/main.py index 484a9c8..1e63cc7 100644 --- a/magic_formula/main.py +++ b/magic_formula/main.py @@ -249,8 +249,8 @@ def export_file(format, tickers_df: pandas.DataFrame, indexes, logger, number_of def export_dataframe_formating(tickers_df: pandas.DataFrame, logger: logging.Logger, - number_of_lines: int = None, - indexes: list = None) -> pandas.DataFrame: + number_of_lines: int = 0, + indexes: list = []) -> pandas.DataFrame: """Exports the ticker dataframe into an excel file :param tickers_df: Dataframe with the stocks information @@ -276,7 +276,7 @@ def export_dataframe_formating(tickers_df: pandas.DataFrame, def export_dataframe_to_sql(tickers_df: pandas.DataFrame, logger: logging.Logger, connection_string: str, - number_of_lines: int = None) -> None: + number_of_lines: int = 0) -> None: """Exportts the ticker dataframe into an postgresql :param tickers_df: Dataframe with the stocks information @@ -296,7 +296,7 @@ def export_dataframe_to_sql(tickers_df: pandas.DataFrame, logger: logging.Logger engine = sqlalchemy.create_engine(connection_string) tickers_df.to_sql('magicformula', engine, if_exists='append', index=False) - except sqlalchemy.exc.OperationalError as error: + except Exception as error: logger.error(f'Error on conection with database {error}')