presik_pos/app/commons/menu_buttons.py

import os
from pathlib import Path

from decimal import Decimal
from PyQt5.QtCore import Qt, pyqtSignal, QSize, QRect
from PyQt5.QtGui import QIcon
from PyQt5.QtWidgets import (
    QWidget, QHBoxLayout, QScroller, QVBoxLayout, QPushButton, QGridLayout,
    QScrollArea, QFrame
)

from .custom_button import CustomButton

pkg_dir = str(Path(os.path.dirname(__file__)).parents[0])
file_back_icon = os.path.join(pkg_dir, 'share', 'back.svg')
file_close_icon = os.path.join(pkg_dir, 'share', 'close.svg')
file_menu_img = os.path.join(pkg_dir, 'share', 'menu.png')

__all__ = ['GridButtons', 'MenuDash']


def money(v):
    return '${:9,}'.format(int(v))


class GridButtons(QWidget):
    sigItem_selected = pyqtSignal(str)

    def __init__(self, parent, rows, num_cols, action, style='product_button'):
        """
            rows: a list of lists
            num_cols: number of columns?
        """
        QWidget.__init__(self)
        self.parent = parent
        self.layout = QGridLayout()
        self.setLayout(self.layout)
        self.button_size = parent.screen_size
        self.rows = rows
        self.action = action
        self.num_cols = num_cols
        self.style = style
        self.create_list_items()
        self.layout.setSpacing(2)
        self.set_items()

    def create_list_items(self):
        self.list_items = []
        for row in self.rows:
            if isinstance(row[3], Decimal):
                row[3] = money(int(row[3]))

            item_button = CustomButton(
                parent=self,
                id=row[0],
                title=row[2],
                desc=str(row[3]),
                method='action_selected',
                target=row[0],
                size=self.button_size,
                name_style=self.style,
            )
            self.list_items.append(item_button)

    def action_selected(self, idx):
        self.action(idx)

    def set_items(self):
        colx = 0
        rowy = 0
        for item_button in self.list_items:
            if colx >= self.num_cols:
                colx = 0
                rowy += 1
            self.layout.addWidget(item_button, rowy, colx)
            colx += 1
        self.layout.setRowStretch(rowy + 1, 1)


class MenuDash(QVBoxLayout):

    def __init__(self, parent, values, selected_method=None, title=None):
        """
            parent: parent window
            values: is to list of list/tuples values for data model
                [('a' 'b', 'c'), ('d', 'e', 'f')...]
            on_selected: method to call when triggered the selection
            title: title of window
        """
        super(MenuDash, self).__init__()

        self.parent = parent
        self.values = values
        self.current_view = '0'
        self._view = 0

        self.ctx_widgets = {}
        self.button_size = parent.screen_size
        self.method_on_selected = getattr(self.parent, selected_method)
        self.create_categories()

        # pixmap = QPixmap(file_menu_img)
        # new_pixmap = pixmap.scaled(180, 55)
        # push_menu.setPixmap(new_pixmap)
        self.is_expanded = False
        self.push_menu = QPushButton('MENU')
        self.push_menu.setObjectName('button_push_menu')
        self.push_menu.clicked.connect(self.expand_menu)

        self.close_menu = QPushButton()
        self.close_menu.setObjectName('button_close_menu')
        self.close_menu.setIcon(QIcon(file_close_icon))
        self.close_menu.setIconSize(QSize(40, 40))
        self.close_menu.clicked.connect(self.stretch_menu)
        self.close_menu.hide()

        widget_head = QWidget()
        widget_head.setStyleSheet("background-color: white;")
        self.layout_head = QHBoxLayout()
        widget_head.setLayout(self.layout_head)

        self.addWidget(widget_head, 0)
        self.pushButtonBack = QPushButton()
        self.pushButtonBack.setIcon(QIcon(file_back_icon))
        self.pushButtonBack.setIconSize(QSize(15, 15))
        self.pushButtonBack.setMaximumWidth(15)

        self.layout_head.addWidget(self.pushButtonBack, stretch=0)
        self.layout_head.addWidget(self.push_menu, stretch=1)
        self.layout_head.addWidget(self.close_menu, stretch=0)
        self.addWidget(self.menu_area, 0)
        self.pushButtonBack.clicked.connect(self.action_back)

        width = self.parent.screen_width
        self.start_width = int(width / 3)
        self.last_width = width - self.start_width - 10

    def change_view(self, view_id):
        view = self.ctx_widgets[self.current_view]['items']
        view.hide()
        if not view_id:
            view_id = '0'

        new_view = self.ctx_widgets[view_id]['items']
        new_view.show()
        self.addWidget(new_view)
        self.current_view = view_id

    def get_products(self, values, name=None):
        grid_buttons = GridButtons(
            self.parent,
            values,
            num_cols=4,
            action=self.method_on_selected
        )
        scroll_area = QScrollArea()
        scroll_area.setWidgetResizable(True)
        scroll_area.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
        scroll_area.setWidget(grid_buttons)
        QScroller.grabGesture(scroll_area, QScroller.LeftMouseButtonGesture)
        return scroll_area
        # print('Name ...', cat_id, name)
        # self.ctx_widgets[cat_id] = {
        #     'items': scroll_area,
        #     'parent': parent_id,
        # }

    def create_categories(self):
        # set the list model
        self.menu_area = QScrollArea()
        self.menu_area.setWidgetResizable(True)
        self.menu_area.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOn)
        QScroller.grabGesture(self.menu_area, QScroller.LeftMouseButtonGesture)

        menu = QWidget()
        self.menu_area.setWidget(menu)

        self.main_categories = QGridLayout()
        self.main_categories.setSpacing(4)

        menu.setLayout(self.main_categories)

        num_cat_cols = 2
        row = 0
        col = 0
        parent_id = '0'
        self.ctx_widgets = {
            '0': {
                'items': self.menu_area,
                'parent': None,
            },
        }
        for value in self.values:
            print('------------- VALORES ---------')
            print(value)
            print('  ')
            cat_id = str(value['id'])
            if not value:
                continue
            if col > num_cat_cols - 1:
                col = 0
                row += 1
            name_button = 'button_' + cat_id

            button = CustomButton(
                parent=self,
                id=name_button,
                icon=value['icon'],
                desc=value['name'],
                method='selected_method',
                target=cat_id,
                size=self.button_size,
                name_style='category_button'
            )
            self.main_categories.addWidget(button, row, col)
            items = value.get('items')
            childs = value.get('childs', [])
            col += 1
            if items:
                products = self.get_products(items)
                self.ctx_widgets[cat_id] = {
                    'items': products,
                    'parent': parent_id,
                }
            # if childs:
            #     subrow = 0
            #     subcol = 0
            #     sub_categories = QGridLayout()
            #     sub_categories.setSpacing(4)
            #     scroll_subarea = QScrollArea()
            #     scroll_subarea.setWidgetResizable(True)
            #     scroll_subarea.setLayout(sub_categories)
            #     self.ctx_widgets[cat_id] = {
            #         'items': scroll_subarea,
            #         'parent': parent_id,
            #     }
            #     for subcat in childs:
            #         sub_id = str(subcat['id'])
            #         if subcol > num_cat_cols - 1:
            #             subcol = 0
            #             subrow += 1
            #         name_button = 'button_' + sub_id
            #         button = CustomButton(
            #             parent=self,
            #             id=name_button,
            #             icon=None,
            #             desc=subcat['name'],
            #             method='selected_method',
            #             target=str(subcat['id']),
            #             size=self.button_size,
            #             name_style='category_button'
            #         )
            #         sub_categories.addWidget(button, row, col)
            #         subcol += 1
            #         products_list = subcat.get('items')
            #
            #         if products_list:
            #             products_items = self.get_products(products_list)
            #             print('===>', sub_id,  products_items)
            #             self.ctx_widgets[sub_id] = {
            #                 'items': products_items,
            #                 'parent': cat_id,
            #             }
            #         col += 1

    def action_back(self):
        parent_id = self.ctx_widgets[self.current_view]['parent']
        self.change_view(parent_id)
        self.expand_menu()

    def selected_method(self, args):
        self.change_view(args)
        self.expand_menu()

    def setDisabled(self, bool):
        self.menu_area.setDisabled(bool)
        # self.grid_buttons.setDisabled(bool)
        # self.scroll_area.setDisabled(bool)

    def stretch_menu(self):
        self.parent.show_right_panel(True)
        self.change_view({'category_invisible': False})
        self.close_menu.hide()

    def expand_menu(self):
        self.parent.show_right_panel(False)
        rect = QRect(self.start_width, 0, self.last_width, self.parent.screen_height)
        self.setGeometry(rect)
        self.close_menu.show()
        self.is_expanded = True