GitHub-Proxy/Cura
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/Ultimaker/Cura synced 2025-08-11 15:08:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add variants to the database

Browse Source 
CURA-6096
This commit is contained in:
Jaime van Kessel 2021-08-20 11:05:37 +02:00
parent 48ab21ac68
commit 5311c949f0
1 changed files with 33 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
cura/Settings/CuraContainerRegistry.py
View File

@ -46,6 +46,7 @@ class CuraContainerRegistry(ContainerRegistry):
self._add_to_database_handlers["variant"] = self._addVariantToDatabase self._add_to_database_handlers["variant"] = self._addVariantToDatabase
self._add_to_database_handlers["quality"] = self._addQualityToDatabase self._add_to_database_handlers["quality"] = self._addQualityToDatabase
self._add_to_database_handlers["intent"] = self._addIntentToDatabase
self._get_from_database_handlers["variant"] = self._getVariantFromDatabase self._get_from_database_handlers["variant"] = self._getVariantFromDatabase
self._get_from_database_handlers["quality"] = self._getQualityFromDatabase self._get_from_database_handlers["quality"] = self._getQualityFromDatabase
@ -60,7 +61,14 @@ class CuraContainerRegistry(ContainerRegistry):
connection = self._getDatabaseConnection() connection = self._getDatabaseConnection()
result = connection.cursor().execute("SELECT * FROM variants where id = ?", (container_id,)) result = connection.cursor().execute("SELECT * FROM variants where id = ?", (container_id,))
data = result.fetchone() data = result.fetchone()
return {"id": data[0], "name": data[1], "hardware_type": data[2], "definition": data[3] } return {"id": data[0], "name": data[1], "hardware_type": data[2], "definition": data[3]}
def _getIntentFromDatabase(self, container_id):
connection = self._getDatabaseConnection()
result = connection.cursor().execute("SELECT * FROM intents where id = ?", (container_id,))
data = result.fetchone()
return {"id": data[0], "name": data[1], "quality_type": data[2], "intent_category": data[3], "variant": data[4], "definition": data[5]}
def _addVariantToDatabase(self, metadata) -> None: def _addVariantToDatabase(self, metadata) -> None:
connection = self._getDatabaseConnection() connection = self._getDatabaseConnection()
@ -84,12 +92,22 @@ class CuraContainerRegistry(ContainerRegistry):
"INSERT INTO qualities (id, name, quality_type, material, variant, global_quality, definition) VALUES (?, ?, ? ,?, ?, ?, ?)", "INSERT INTO qualities (id, name, quality_type, material, variant, global_quality, definition) VALUES (?, ?, ? ,?, ?, ?, ?)",
(metadata["id"], metadata["name"], metadata["quality_type"], material, variant, global_quality, metadata["definition"])) (metadata["id"], metadata["name"], metadata["quality_type"], material, variant, global_quality, metadata["definition"]))
def _addIntentToDatabase(self, metadata) -> None:
connection = self._getDatabaseConnection()
connection.cursor().execute(
"INSERT INTO intents (id, name, quality_type, intent_category, variant, definition) VALUES (?, ?, ? ,?, ?, ?)",
(metadata["id"], metadata["name"], metadata["quality_type"], metadata["intent_category"], metadata["variant"],
metadata["definition"]))
@override(ContainerRegistry) @override(ContainerRegistry)
def _createDatabaseFile(self, db_path: str) -> None: def _createDatabaseFile(self, db_path: str) -> None:
connection = super()._createDatabaseFile(db_path) connection = super()._createDatabaseFile(db_path)
cursor = connection.cursor() cursor = connection.cursor()
cursor.execute(""" cursor.execute("""
CREATE TABLE qualities( CREATE TABLE qualities
(
id text, id text,
name text, name text,
quality_type text, quality_type text,
@ -100,13 +118,25 @@ class CuraContainerRegistry(ContainerRegistry):
); );
CREATE UNIQUE INDEX idx_qualities_id on qualities (id); CREATE UNIQUE INDEX idx_qualities_id on qualities (id);
CREATE TABLE variants( CREATE TABLE variants
(
id text, id text,
name text, name text,
hardware_type text, hardware_type text,
definition text definition text
); );
CREATE UNIQUE INDEX idx_variants_id on variants (id); CREATE UNIQUE INDEX idx_variants_id on variants (id);
CREATE TABLE intents
(
id text,
name text,
quality_type text,
intent_category text,
variant text,
definition text
);
CREATE UNIQUE INDEX idx_intents_id on intents (id);
""") """)
return connection return connection