🐛 init toolbar after login

easytrader/clienttrader.py

@@ -115,7 +115,7 @@ def connect(self, exe_path=None, **kwargs):
         self._app = pywinauto.Application().connect(path=connect_path, timeout=10)
         self._close_prompt_windows()
         self._main = self._app.top_window()
-        self._toolbar = self._main.child_window(class_name="ToolbarWindow32")
+        self._init_toolbar()
 
     @property
     def broker_type(self):
@@ -127,6 +127,9 @@ def balance(self):
 
         return self._get_balance_from_statics()
 
+    def _init_toolbar(self):
+        self._toolbar = self._main.child_window(class_name="ToolbarWindow32")
+
     def _get_balance_from_statics(self):
         result = {}
         for key, control_id in self._config.BALANCE_CONTROL_ID_GROUP.items():
@@ -536,3 +539,4 @@ def prepare(
             comm_password,
             **kwargs
         )
+        self._init_toolbar()