from qgis.PyQt.QtWidgets import QToolBar
from .Resource import ResourceAction
from .icons import icons

class SiweiToolbar(QToolBar):
    def __init__(self, iface, parent=None) -> None:
        super().__init__("资源目录", parent)
        self.iface = iface
        self.icons = icons
        self.setToolTip("资源目录")
        self.add_button = None
        self.actions = []
        self.setup_action()

    def setup_action(self):
        action_ftp = ResourceAction(iface=self.iface, parent=self)
        self.actions.extend([action_ftp])
        self.addActions(self.actions)

    def remove_dock(self):
        print(self.actions)
        if self.actions is not None and len(self.actions) > 1:
            dock = self.actions[1]
            dock.unload()