Start refactoring for images management

gns3server/db/models/__init__.py

@@ -20,6 +20,7 @@ from .users import User, UserGroup
 from .roles import Role
 from .permissions import Permission
 from .computes import Compute
+from .images import Image
 from .templates import (
     Template,
     CloudTemplate,

gns3server/db/models/images.py

@@ -0,0 +1,34 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2021 GNS3 Technologies Inc.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+from sqlalchemy import Column, String, Integer
+from sqlalchemy.orm import relationship
+
+from .base import BaseTable
+
+
+class Image(BaseTable):
+
+    __tablename__ = "images"
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    filename = Column(String, unique=True, index=True)
+    image_type = Column(String)
+    path = Column(String)
+    checksum = Column(String)
+    checksum_algorithm = Column(String)
+    templates = relationship("Template")

gns3server/db/models/templates.py

@@ -35,6 +35,8 @@ class Template(BaseTable):
     usage = Column(String)
     template_type = Column(String)
 
+    image_id = Column(Integer, ForeignKey('images.id', ondelete="CASCADE"))
+
     __mapper_args__ = {
         "polymorphic_identity": "templates",
         "polymorphic_on": template_type,