ProjectFlask/FlaskMigrateExtension/MigrateConfigDialog.py

# -*- coding: utf-8 -*-

# Copyright (c) 2020 - 2021 Detlev Offenbach <detlev@die-offenbachs.de>
#

"""
Module implementing a dialog to edit the flask-migrate configuration.
"""

from PyQt6.QtCore import Qt
from PyQt6.QtWidgets import QDialog

from EricWidgets.EricPathPicker import EricPathPickerModes
from EricWidgets.EricApplication import ericApp

from .Ui_MigrateConfigDialog import Ui_MigrateConfigDialog


class MigrateConfigDialog(QDialog, Ui_MigrateConfigDialog):
    """
    Class implementing a dialog to edit the flask-migrate configuration.
    """
    def __init__(self, configuration, parent=None):
        """
        Constructor
        
        @param configuration current pybabel configuration
        @type dict
        @param parent reference to the parent widget
        @type QWidget
        """
        super().__init__(parent)
        self.setupUi(self)
        
        self.__ericProject = ericApp().getObject("Project")
        
        self.migrationsDirectoryPicker.setMode(
            EricPathPickerModes.DIRECTORY_MODE)
        self.migrationsDirectoryPicker.setDefaultDirectory(
            self.__ericProject.getProjectPath())
        
        self.migrationsDirectoryPicker.setFocus(
            Qt.FocusReason.OtherFocusReason)
        
        if (
            "migrationsDirectory" in configuration and
            configuration["migrationsDirectory"]
        ):
            self.migrationsDirectoryPicker.setText(
                self.__ericProject.getAbsoluteUniversalPath(
                    configuration["migrationsDirectory"]))
        
        msh = self.minimumSizeHint()
        self.resize(max(self.width(), msh.width()), msh.height())
    
    def getConfiguration(self):
        """
        Public method to get the entered configuration data.
        
        @return pybabel configuration
        @rtype dict
        """
        configuration = {
            "migrationsDirectory": self.__ericProject.getRelativeUniversalPath(
                self.migrationsDirectoryPicker.text()),
        }
        
        return configuration