From 7bc41121c06fb905c0dbd2a464a14b8f3f3d743b Mon Sep 17 00:00:00 2001 From: Chavithra PARANA Date: Sat, 22 May 2021 16:53:05 +0200 Subject: [PATCH] degiro integration : apply black --- .../portfolio/degiro_controller.py | 2 +- gamestonk_terminal/portfolio/degiro_model.py | 36 +++++++++++-------- gamestonk_terminal/portfolio/degiro_view.py | 3 +- 3 files changed, 24 insertions(+), 17 deletions(-) diff --git a/gamestonk_terminal/portfolio/degiro_controller.py b/gamestonk_terminal/portfolio/degiro_controller.py index 018a028b696..13730f70a30 100644 --- a/gamestonk_terminal/portfolio/degiro_controller.py +++ b/gamestonk_terminal/portfolio/degiro_controller.py @@ -1,4 +1,4 @@ - # IMPORTATION STANDARD +# IMPORTATION STANDARD import argparse # IMPORTATION THIRDPARTY diff --git a/gamestonk_terminal/portfolio/degiro_model.py b/gamestonk_terminal/portfolio/degiro_model.py index 654637cffef..cf64a42a5e2 100644 --- a/gamestonk_terminal/portfolio/degiro_model.py +++ b/gamestonk_terminal/portfolio/degiro_model.py @@ -22,6 +22,9 @@ from trading.pb.trading_pb2 import ( # IMPORTATION INTERNAL import gamestonk_terminal.config_terminal as config +# pylint: disable=no-member +# pylint: disable=no-else-return + class DegiroModel: def __init__(self): @@ -35,10 +38,10 @@ class DegiroModel: self.__trading_api = TradingAPI( credentials=self.__default_credentials, ) - + def __hold_fetch_additional_information( self, - positions:pd.DataFrame, + positions: pd.DataFrame, ) -> pd.DataFrame: """Fetch extra information about the positions like : - name @@ -103,7 +106,7 @@ class DegiroModel: request_list=request_list, raw=False, ) - + # CHECK EMPTINESS if len(update_pb.portfolio.values) == 0: return pd.DataFrame() @@ -200,8 +203,7 @@ class DegiroModel: ) product = products_lookup_dict["products"][0] - if len(products_lookup.products) <= 0 \ - or product["symbol"] != symbol: + if len(products_lookup.products) <= 0 or product["symbol"] != symbol: return None else: return int(product["id"]) @@ -231,10 +233,10 @@ class DegiroModel: confirmation_id=confirmation_id, order=order, ) - + def hold_positions(self) -> pd.DataFrame: return self.__hold_fetch_current_positions() - + def lastnews(self, limit: int) -> LatestNews: trading_api = self.__trading_api request = LatestNews.Request( @@ -279,25 +281,29 @@ class DegiroModel: def pending(self) -> Update.Orders: trading_api = self.__trading_api request_list = Update.RequestList() - request_list.values.extend([ - Update.Request(option=Update.Option.ORDERS, last_updated=0), - ]) + request_list.values.extend( + [ + Update.Request(option=Update.Option.ORDERS, last_updated=0), + ] + ) update = trading_api.get_update(request_list=request_list) return update.orders def topnews(self) -> TopNewsPreview: return self.__trading_api.get_top_news_preview(raw=False) - + def update(self, order: Order) -> Update.Orders: return self.__trading_api.update_order(order=order) def update_pending_order(self, order_id: str) -> Union[None, Order]: trading_api = self.__trading_api request_list = Update.RequestList() - request_list.values.extend([ - Update.Request(option=Update.Option.ORDERS, last_updated=0), - ]) + request_list.values.extend( + [ + Update.Request(option=Update.Option.ORDERS, last_updated=0), + ] + ) update = trading_api.get_update(request_list=request_list) if len(update.orders.values) > 0: @@ -306,4 +312,4 @@ class DegiroModel: return order return None else: - return None \ No newline at end of file + return None diff --git a/gamestonk_terminal/portfolio/degiro_view.py b/gamestonk_terminal/portfolio/degiro_view.py index 3b805f1d44c..402b179c422 100644 --- a/gamestonk_terminal/portfolio/degiro_view.py +++ b/gamestonk_terminal/portfolio/degiro_view.py @@ -11,6 +11,7 @@ from trading.pb.trading_pb2 import ( Update, ) + class DegiroView: @staticmethod def help_display(): @@ -222,7 +223,7 @@ class DegiroView: @staticmethod def update_display_not_found(order_id: str): print("The following `order` was not found:", order_id) - + @staticmethod def update_display_success(): print("`Order` updated .")