26 lines
1.1 KiB
Python
26 lines
1.1 KiB
Python
|
import driver
|
||
|
from gui.elements.object import QObject
|
||
|
from gui.elements.text_edit import TextEdit
|
||
|
from gui.objects_map import names
|
||
|
|
||
|
|
||
|
class TokenSelectorPopup(QObject):
|
||
|
def __init__(self):
|
||
|
super().__init__(names.tokenSelectorPanel_TokenSelectorNew)
|
||
|
self.token_selector_panel = QObject(names.tokenSelectorPanel_TokenSelectorNew)
|
||
|
self.tab_bar = QObject(names.tokensTabBar_StatusTabBar)
|
||
|
self.assets_tab = QObject(names.tokenSelectorPanel_AssetsTab)
|
||
|
self.collectibles_tab = QObject(names.tokenSelectorPanel_CollectiblesTab)
|
||
|
self.asset_list_item = QObject(names.tokenSelectorAssetDelegate_template)
|
||
|
self.amount_to_send_field = TextEdit(names.amountInput_TextEdit)
|
||
|
|
||
|
def select_asset_from_list(self, asset_name: str):
|
||
|
self.assets_tab.click()
|
||
|
assets_list = driver.findAllObjects(self.asset_list_item.real_name)
|
||
|
assert assets_list, f'Assets are not displayed'
|
||
|
for item in assets_list:
|
||
|
if getattr(item, 'tokensKey', '') == asset_name:
|
||
|
QObject(item).click()
|
||
|
break
|
||
|
return self
|