status-desktop/test/e2e/gui/screens/community_settings.py

import time
import typing

import allure

import configs
import driver
from driver.objects_access import walk_children
from gui.components.community.color_select_popup import ColorSelectPopup
from gui.components.community.kick_member_popup import KickMemberPopup
from gui.components.community.tags_select_popup import TagsSelectPopup
from gui.components.os.open_file_dialogs import OpenFileDialog
from gui.components.picture_edit_popup import PictureEditPopup
from gui.elements.button import Button
from gui.elements.check_box import CheckBox
from gui.elements.object import QObject
from gui.elements.scroll import Scroll
from gui.elements.text_edit import TextEdit
from gui.elements.text_label import TextLabel
from gui.objects_map import communities_names, names
from gui.screens.community_settings_tokens import TokensView
from scripts.tools.image import Image


class CommunitySettingsScreen(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_communityLoader_Loader)
        self.left_panel = LeftPanel()


class LeftPanel(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_communityColumnView_CommunityColumnView)
        self._back_to_community_button = Button(
            communities_names.mainWindow_communitySettingsBackToCommunityButton_StatusBaseText)
        self._overview_button = Button(communities_names.overview_StatusNavigationListItem)
        self._members_button = Button(communities_names.members_StatusNavigationListItem)
        self._permissions_button = Button(communities_names.permissions_StatusNavigationListItem)
        self._tokens_button = Button(communities_names.tokens_StatusNavigationListItem)
        self._airdrops_button = Button(communities_names.airdrops_StatusNavigationListItem)

    @allure.step('Open community main view')
    def back_to_community(self):
        self._back_to_community_button.click()

    @allure.step('Open community overview')
    def open_overview(self) -> 'OverviewView':
        if not self._overview_button.is_selected:
            self._overview_button.click()
        return OverviewView()

    @allure.step('Open community members')
    def open_members(self) -> 'MembersView':
        if not self._members_button.is_selected:
            self._members_button.click()
        return MembersView()

    @allure.step('Open permissions')
    def open_permissions(self) -> 'PermissionsIntroView':
        self._permissions_button.click()
        return PermissionsIntroView()

    @allure.step('Open tokens')
    def open_tokens(self) -> 'TokensView':
        self._tokens_button.click()
        return TokensView()

    @allure.step('Open airdrops')
    def open_airdrops(self) -> 'AirdropsView':
        self._airdrops_button.click()
        return AirdropsView()


class OverviewView(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_OverviewSettingsPanel)
        self._name_text_label = TextLabel(communities_names.communityOverviewSettingsCommunityName_StatusBaseText)
        self._description_text_label = TextLabel(
            communities_names.communityOverviewSettingsCommunityDescription_StatusBaseText)
        self._edit_button = Button(communities_names.mainWindow_Edit_Community_StatusButton)

    @property
    @allure.step('Get community name')
    def name(self) -> str:
        return self._name_text_label.text

    @property
    @allure.step('Get community description')
    def description(self) -> str:
        return self._description_text_label.text

    @allure.step('Open edit community view')
    def open_edit_community_view(self, attempts: int = 2) -> 'EditCommunityView':
        time.sleep(0.5)
        self._edit_button.click()
        try:
            return EditCommunityView()
        except Exception as ex:
            if attempts:
                self.open_edit_community_view(attempts - 1)
            else:
                raise ex


class EditCommunityView(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_communityEditPanelScrollView_EditSettingsPanel)
        self._scroll = Scroll(communities_names.communityEditPanelScrollView_Flickable)
        self._name_text_edit = TextEdit(communities_names.communityEditPanelScrollView_communityNameInput_TextEdit)
        self._description_text_edit = TextEdit(
            communities_names.communityEditPanelScrollView_communityDescriptionInput_TextEdit)
        self._logo = QObject(communities_names.communityEditPanelScrollView_image_StatusImage)
        self._add_logo_button = Button(communities_names.communityEditPanelScrollView_editButton_StatusRoundButton)
        self._banner = QObject(communities_names.communityEditPanelScrollView_image_StatusImage_2)
        self._add_banner_button = Button(communities_names.communityEditPanelScrollView_editButton_StatusRoundButton_2)
        self._select_color_button = Button(communities_names.communityEditPanelScrollView_StatusPickerButton)
        self._choose_tag_button = Button(communities_names.communityEditPanelScrollView_Choose_StatusPickerButton)
        self._tag_item = QObject(communities_names.communityEditPanelScrollView_StatusCommunityTag)
        self._archive_support_checkbox = CheckBox(
            communities_names.communityEditPanelScrollView_archiveSupportToggle_StatusCheckBox)
        self._request_to_join_checkbox = CheckBox(
            communities_names.communityEditPanelScrollView_requestToJoinToggle_StatusCheckBox)
        self._pin_messages_checkbox = CheckBox(
            communities_names.communityEditPanelScrollView_pinMessagesToggle_StatusCheckBox)
        self._intro_text_edit = TextEdit(
            communities_names.communityEditPanelScrollView_editCommunityIntroInput_TextEdit)
        self._outro_text_edit = TextEdit(
            communities_names.communityEditPanelScrollView_editCommunityOutroInput_TextEdit)
        self._save_changes_button = Button(names.mainWindow_Save_changes_StatusButton)
        self._cropped_image_edit_logo_item = QObject(communities_names.croppedImageEditLogo)
        self._cropped_image_edit_banner_item = QObject(communities_names.croppedImageEditBanner)

    @property
    @allure.step('Get community name')
    def name(self) -> str:
        return self._name_text_edit.text

    @name.setter
    @allure.step('Set community name')
    def name(self, value: str):
        self._name_text_edit.text = value

    @property
    @allure.step('Get community description')
    def description(self) -> str:
        return self._description_text_edit.text

    @description.setter
    @allure.step('Set community description')
    def description(self, value: str):
        self._description_text_edit.text = value

    @property
    @allure.step('Get community logo')
    def logo(self) -> Image:
        return self._logo.image

    @allure.step('Set community logo')
    def logo(self, kwargs: dict):
        self._add_logo_button.click()
        OpenFileDialog().wait_until_appears().open_file(kwargs['fp'])
        PictureEditPopup().wait_until_appears().set_zoom_shift_for_picture(kwargs.get('zoom', None),
                                                                           kwargs.get('shift', None))

    @property
    @allure.step('Get community banner')
    def banner(self) -> Image:
        return self._banner.image

    @allure.step('Set community banner')
    def banner(self, kwargs: dict):
        self._add_banner_button.click()
        OpenFileDialog().wait_until_appears().open_file(kwargs['fp'])
        PictureEditPopup().wait_until_appears().set_zoom_shift_for_picture(kwargs.get('zoom', None),
                                                                           kwargs.get('shift', None))

    @allure.step('Set community logo without file upload dialog')
    def set_logo_without_file_upload_dialog(self, path):
        self._cropped_image_edit_logo_item.object.cropImage('file://' + str(path))
        return PictureEditPopup()

    @allure.step('Set community banner without file upload dialog')
    def set_banner_without_file_upload_dialog(self, path):
        self._cropped_image_edit_banner_item.object.cropImage('file://' + str(path))
        return PictureEditPopup()

    @property
    @allure.step('Get community color')
    def color(self) -> str:
        return str(self._select_color_button.object.text)

    @color.setter
    @allure.step('Set community color')
    def color(self, value: str):
        self._scroll.vertical_scroll_to(self._select_color_button)
        self._select_color_button.click()
        ColorSelectPopup().wait_until_appears().select_color(value)

    @property
    @allure.step('Get community tags')
    def tags(self):
        self._scroll.vertical_scroll_to(self._choose_tag_button)
        return [str(tag.title) for tag in driver.fiandAllObjects(self._tag_item.real_name)]

    @tags.setter
    @allure.step('Set community tags')
    def tags(self, values: typing.List[str]):
        self._scroll.vertical_scroll_to(self._choose_tag_button)
        self._choose_tag_button.click()
        TagsSelectPopup().wait_until_appears().select_tags(values)

    @property
    @allure.step('Get community intro')
    def intro(self) -> str:
        self._scroll.vertical_scroll_to(self._intro_text_edit)
        return self._intro_text_edit.text

    @intro.setter
    @allure.step('Set community intro')
    def intro(self, value: str):
        self._scroll.vertical_scroll_to(self._intro_text_edit)
        self._intro_text_edit.text = value

    @property
    @allure.step('Get community outro')
    def outro(self) -> str:
        self._scroll.vertical_scroll_to(self._outro_text_edit)
        return self._outro_text_edit.text

    @outro.setter
    @allure.step('Set community outro')
    def outro(self, value: str):
        self._scroll.vertical_scroll_to(self._outro_text_edit)
        self._outro_text_edit.text = value

    @property
    @allure.step('Get pin message checkbox state')
    def pin_message_checkbox_state(self) -> bool:
        self._scroll.vertical_scroll_to(self._pin_messages_checkbox)
        return self._pin_messages_checkbox.object.checked

    @allure.step('Edit community')
    def edit(self, name, description, intro, outro, logo, banner):
        self._scroll.vertical_scroll_to(self._name_text_edit)
        self.name = name
        self.description = description
        self.set_logo_without_file_upload_dialog(logo)
        PictureEditPopup().set_zoom_shift_for_picture(None, None)
        self.set_banner_without_file_upload_dialog(banner)
        PictureEditPopup().set_zoom_shift_for_picture(None, None)
        self.intro = intro
        self.outro = outro
        self._save_changes_button.click()
        self.wait_until_hidden()


class MembersView(QObject):

    def __init__(self):
        super().__init__(communities_names.mainWindow_MembersSettingsPanel)
        self._member_list_item = QObject(communities_names.memberItem_StatusMemberListItem)
        self._kick_member_button = Button(communities_names.communitySettings_MembersTab_Member_Kick_Button)

    @property
    @allure.step('Get community members')
    def members(self) -> typing.List[str]:
        return [str(member.title) for member in driver.findAllObjects(self._member_list_item.real_name)]

    @allure.step('Get community member objects')
    def get_member_objects(self) -> typing.List:
        member_objects = []
        for item in driver.findAllObjects(self._member_list_item.real_name):
            member_objects.append(item)
        return member_objects

    @allure.step('Find community member by name')
    def get_member_object(self, member_name: str):
        member = None
        started_at = time.monotonic()
        while member is None:
            for _member in self.get_member_objects():
                if member_name in str(_member.userName):
                    member = _member
                    break
            if time.monotonic() - started_at > configs.timeouts.MESSAGING_TIMEOUT_SEC:
                raise LookupError(f'Member not found')
        return member

    @allure.step('Kick community member')
    def kick_member(self, member_name: str):
        member = self.get_member_object(member_name)
        QObject(real_name=driver.objectMap.realName(member)).hover()
        self._kick_member_button.click()
        kick_member_popup = KickMemberPopup()
        assert kick_member_popup.exists
        kick_member_popup.confirm_kicking()


class AirdropsView(QObject):
    def __init__(self):
        super(AirdropsView, self).__init__(communities_names.mainWindow_airdropPanel_AirdropsSettingsPanel)
        self._new_airdrop_button = Button(communities_names.mainWindow_New_Airdrop_StatusButton)
        self._welcome_image = QObject(communities_names.welcomeSettingsAirdrops_Image)
        self._welcome_title = TextLabel(communities_names.welcomeSettingsAirdrops_Title)
        self._welcome_subtitle = TextLabel(communities_names.welcomeSettingsAirdrops_Subtitle)
        self._welcome_checklist_1 = TextLabel(communities_names.checkListText_0_Airdrops)
        self._welcome_checklist_2 = TextLabel(communities_names.checkListText_1_Airdrops)
        self._welcome_checklist_3 = TextLabel(communities_names.checkListText_2_Airdrops)
        self._get_started_infobox = QObject(communities_names.infoBox_StatusInfoBoxPanel)
        self._mint_owner_token_button = Button(communities_names.mint_Owner_token_Airdrops_StatusButton)

    @property
    @allure.step('Get new airdrop button enable state')
    def is_new_airdrop_button_present(self) -> bool:
        return self._new_airdrop_button.exists

    @property
    @allure.step('Get airdrops welcome image path')
    def airdrops_welcome_image_path(self) -> str:
        return self._welcome_image.object.source.path

    @property
    @allure.step('Get airdrops welcome title')
    def airdrops_welcome_title(self) -> str:
        return self._welcome_title.text

    @property
    @allure.step('Get airdrops welcome subtitle')
    def airdrops_welcome_subtitle(self) -> str:
        return self._welcome_subtitle.text

    @property
    @allure.step('Get airdrops checklist')
    def airdrops_checklist(self) -> typing.List[str]:
        airdrops_checklist = [str(self._welcome_checklist_1.object.text), str(self._welcome_checklist_2.object.text),
                              str(self._welcome_checklist_3.object.text)]
        return airdrops_checklist

    @property
    @allure.step('Get airdrops info box title')
    def airdrops_infobox_title(self) -> str:
        return self._get_started_infobox.object.title

    @property
    @allure.step('Get airdrops info box text')
    def airdrops_infobox_text(self) -> str:
        return self._get_started_infobox.object.text

    @property
    @allure.step('Get airdrops mint owner token button visibility state')
    def is_airdrops_owner_token_button_visible(self) -> bool:
        return self._mint_owner_token_button.is_visible


class PermissionsIntroView(QObject):
    def __init__(self):
        super(PermissionsIntroView, self).__init__(communities_names.o_IntroPanel)
        self._add_new_permission_button = Button(communities_names.add_new_permission_button)
        self._welcome_image = QObject(communities_names.community_welcome_screen_image)
        self._welcome_title = TextLabel(communities_names.community_welcome_screen_title)
        self._welcome_subtitle = TextLabel(communities_names.community_welcome_screen_subtitle)
        self._welcome_checklist_1 = TextLabel(communities_names.community_welcome_screen_checkList_element1)
        self._welcome_checklist_2 = TextLabel(communities_names.community_welcome_screen_checkList_element2)
        self._welcome_checklist_3 = TextLabel(communities_names.community_welcome_screen_checkList_element3)
        self._edit_permission_button = QObject(communities_names.edit_pencil_icon_StatusIcon)
        self._delete_permission_button = QObject(communities_names.delete_icon_StatusIcon)
        self._hide_icon = QObject(communities_names.hide_icon_StatusIcon)

    @property
    @allure.step('Get hide icon visibility')
    def is_hide_icon_visible(self) -> bool:
        return self._hide_icon.is_visible

    @property
    @allure.step('Get permission welcome image path')
    def permission_welcome_image_source(self) -> str:
        return self._welcome_image.object.source.path

    @property
    @allure.step('Get permission welcome title')
    def permission_welcome_title(self) -> str:
        return self._welcome_title.text

    @property
    @allure.step('Get permission welcome subtitle')
    def permission_welcome_subtitle(self) -> str:
        return self._welcome_subtitle.text

    @property
    @allure.step('Get permission checklist')
    def permission_checklist(self) -> typing.List[str]:
        permission_checklist = [str(self._welcome_checklist_1.object.text), str(self._welcome_checklist_2.object.text),
                                str(self._welcome_checklist_3.object.text)]
        return permission_checklist

    @allure.step('Click add new permission button')
    def add_new_permission(self) -> 'PermissionsSettingsView':
        self._add_new_permission_button.click()
        return PermissionsSettingsView()

    @property
    @allure.step('Is add new permission button visible')
    def is_add_new_permission_button_present(self) -> bool:
        return self._add_new_permission_button.exists

    @allure.step('Open edit permission view')
    def open_edit_permission_view(self):
        self._edit_permission_button.click()
        return PermissionsSettingsView().wait_until_appears()

    @allure.step('Click delete permission button')
    def click_delete_permission(self):
        self._delete_permission_button.click()


class PermissionsSettingsView(QObject):
    def __init__(self):
        super(PermissionsSettingsView, self).__init__(communities_names.mainWindow_PermissionsSettingsPanel)
        self._who_holds_checkbox = CheckBox(communities_names.editPermissionView_whoHoldsSwitch_StatusSwitch)
        self._who_holds_asset_field = TextEdit(communities_names.edit_TextEdit)
        self._who_holds_amount_field = TextEdit(communities_names.inputValue_StyledTextField)
        self._asset_item = QObject(communities_names.o_TokenItem)
        self._is_allowed_to_option_button = Button(communities_names.customPermissionListItem)
        self._in_general_checkbox = Button(communities_names.checkBox_StatusCheckBox)
        self._hide_permission_checkbox = CheckBox(communities_names.editPermissionView_switchItem_StatusSwitch)
        self._create_permission_button = Button(communities_names.editPermissionView_Create_permission_StatusButton)
        self._add_button = Button(communities_names.add_StatusButton)
        self._who_holds_list_item = QObject(communities_names.editPermissionView_Who_holds_StatusItemSelector)
        self._is_allowed_to_list_item = QObject(communities_names.editPermissionView_Is_allowed_to_StatusFlowSelector)
        self._in_list_item = QObject(communities_names.editPermissionView_In_StatusItemSelector)
        self._tag_item = QObject(communities_names.o_StatusListItemTag)
        self._who_holds_tag = QObject(communities_names.whoHoldsTagListItem)
        self._is_allowed_tag = QObject(communities_names.isAllowedTagListItem)
        self._in_community_in_channel_tag = QObject(communities_names.inCommunityTagListItem)
        self._is_allowed_to_edit_tag = QObject(communities_names.isAllowedToEditPermissionView_StatusListItemTag)
        self._member_role_limit_warning = QObject(communities_names.memberRoleLimitWarning)

    @allure.step('Verify member role limit warning is present')
    def is_member_role_warning_text_present(self):
        return self._member_role_limit_warning.exists

    @allure.step('Get warning text')
    def get_member_role_limit_warning_text(self):
        return str(self._member_role_limit_warning.object.text)

    @allure.step('Get titles of Who holds tags')
    def get_who_holds_tags_titles(self, attempt: int = 2) -> typing.List[str]:
        try:
            return [str(tag.title) for tag in driver.findAllObjects(self._who_holds_tag.real_name)]
        except AttributeError as er:
            if attempt:
                time.sleep(1)
                return self.get_who_holds_tags_titles(attempt - 1)
            else:
                raise er

    @allure.step('Get titles of Is Allowed tags')
    def get_is_allowed_tags_titles(self, attempt: int = 2) -> typing.List[str]:
        try:
            return [str(tag.title) for tag in driver.findAllObjects(self._is_allowed_tag.real_name)]
        except AttributeError as er:
            if attempt:
                time.sleep(1)
                return self.get_is_allowed_tags_titles(attempt - 1)
            else:
                raise er

    @allure.step('Get title of inCommunity tag')
    def get_in_community_in_channel_tags_titles(self, attempt: int = 2) -> typing.List[str]:
        try:
            return [str(tag.title) for tag in driver.findAllObjects(self._in_community_in_channel_tag.real_name)]
        except AttributeError as er:
            if attempt:
                time.sleep(1)
                return self.get_in_community_in_channel_tags_titles(attempt - 1)
            else:
                raise er

    @allure.step('Set state of who holds checkbox')
    def set_who_holds_checkbox_state(self, state):
        if state is False:
            self._who_holds_checkbox.set(state)

    @allure.step('Set asset and amount')
    def set_who_holds_asset_and_amount(self, asset: str, amount: str):
        if asset is not False:
            self.open_who_holds_context_menu()
            self._who_holds_asset_field.clear().text = asset
            self._asset_item.click()
            self._who_holds_asset_field.wait_until_hidden()
            self._who_holds_amount_field.text = amount
            self.click_add_button()

    @allure.step('Choose option from Is allowed to context menu')
    def set_is_allowed_to(self, name):
        self.open_is_allowed_to_context_menu()
        self._is_allowed_to_option_button.real_name['objectName'] = name
        self._is_allowed_to_option_button.wait_until_appears().click()
        self.click_add_button()

    @allure.step('Choose channel from In context menu')
    def set_in(self, in_general):
        if in_general == '#general':
            self.open_in_context_menu()
            self._in_general_checkbox.wait_until_appears().click()
            self.click_add_button()

    @allure.step('Click add button')
    def click_add_button(self, attempt: int = 2):
        self._add_button.click()
        try:
            self._add_button.wait_until_hidden()
        except AssertionError as err:
            if attempt:
                self.click_add_button(attempt - 1)
            else:
                raise err

    @allure.step('Click create permission')
    def create_permission(self):
        self._create_permission_button.click()
        self._create_permission_button.wait_until_hidden()

    @allure.step('Open Who holds context menu')
    def open_who_holds_context_menu(self, attempt: int = 2):
        try:
            for child in walk_children(self._who_holds_list_item.object):
                if getattr(child, 'objectName', '') == 'addItemButton':
                    driver.mouseClick(child)
                    return
        except LookupError:
            if attempt:
                return self.open_who_holds_context_menu(attempt - 1)
            else:
                raise LookupError('Add button for who holds not found')

    @allure.step('Open Is allowed to context menu')
    def open_is_allowed_to_context_menu(self, attempt: int = 2):
        try:
            for child in walk_children(self._is_allowed_to_list_item.object):
                if getattr(child, 'objectName', '') == 'addItemButton':
                    driver.mouseClick(child)
                    return
        except LookupError:
            if attempt:
                return self.open_is_allowed_to_context_menu(attempt - 1)
            else:
                raise LookupError('Add button for allowed to not found')

    @allure.step('Open In context menu')
    def open_in_context_menu(self, attempt: int = 2):
        try:
            for child in walk_children(self._in_list_item.object):
                if getattr(child, 'objectName', '') == 'addItemButton':
                    driver.mouseClick(child)
                    return
        except LookupError:
            if attempt:
                return self.open_in_context_menu(attempt - 1)
            else:
                raise LookupError('Add button for in not found')

    @allure.step('Switch hide permission checkbox')
    def switch_hide_permission_checkbox(self, state):
        self._hide_permission_checkbox.set(state)

    @allure.step('Change allowed to option from permission')
    def set_allowed_to_from_permission(self, name):
        self._is_allowed_to_edit_tag.click()
        self._is_allowed_to_option_button.real_name['objectName'] = name
        self._is_allowed_to_option_button.wait_until_appears().click()
        self.click_add_button()