# Copyright (c) 2021 Ultimaker B.V.
# Cura is released under the terms of the LGPLv3 or higher.

from UM.Settings.SQLQueryFactory import SQLQueryFactory
from UM.Settings.DatabaseContainerMetadataController import DatabaseMetadataContainerController
from UM.Settings.InstanceContainer import InstanceContainer


class VariantDatabaseHandler(DatabaseMetadataContainerController):
    """The Database handler for Variant containers"""

    def __init__(self):
        super().__init__(SQLQueryFactory(
            table="variant",
            fields={
                "id": "text",
                "name": "text",
                "hardware_type": "text",
                "definition": "text",
                "version": "text",
                "setting_version": "text",
                "reference_extruder_id": "text",
                "build_plate_name": "text",
            },
        ))
        self._container_type = InstanceContainer
