from qgis.PyQt.QtWidgets import QToolBar
from .FTP.main import FTPAction
from .icons import icons


class SiweiToolbar(QToolBar):
    def __init__(self, iface, parent=None) -> None:
        super().__init__("Siwei Tools 工具栏", parent)
        self.iface = iface
        self.icons = icons
        self.setToolTip("四维 Tools 工具栏")
        self.add_button = None
        self.actions = []
        self.setup_action()

    def setup_action(self):
        self.addWidget(self.add_button)
        # 添加 Action
        ftpAction = FTPAction(iface=self.iface, parent=self)
        self.addActions(self.actions)

    def remove_dock(self):
        dock = self.actions[1]
        dock.unload()