Implemeted getters

model_service/model/__init__.py

@@ -1,4 +1,4 @@
 #!/usr/bin/env python3
 from .db import db
 from .default import Default
-from .details import Details
+from .aimodel import AIModel

model_service/model/aimodel.py

@@ -1,16 +1,18 @@
 #!/usr/bin/env python3
 from .db import db
 from sqlalchemy.sql import func
+from sqlalchemy.dialects.postgresql import UUID
+import uuid
 
 
-class Details(db.Model):
-    id = db.Column(db.Integer, primary_key=True, auto_increment=True)
-    name = db.Column(db.String(30), key=True)
+class AIModel(db.Model):
+    id = db.Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4, unique=True, nullable=False)
     timestamp = db.Column(db.TIMESTAMP, nullable=False, server_default=func.now())
 
+    # details
     mid_window = db.Column(db.Float)
     mid_step = db.Column(db.Float)
     short_window = db.Column(db.Float)
     short_step = db.Column(db.Float)
     compute_beat = db.Column(db.Float)
-    type = db.Column(db.String(15))
+    type = db.Column(db.String(15))

model_service/model/default.py

@@ -1,7 +1,8 @@
 #!/usr/bin/env python3
 from .db import db
+from sqlalchemy.dialects.postgresql import UUID
 
 
-class Default(db.Model):
-    default_id = db.Column(db.Integer, db.ForeignKey("details.id"), nullable=False)
-    default = db.relationship("Details")
+class Default(db.AIModel):
+    default_id = db.Column(UUID(as_uuid=True), db.ForeignKey("aimodel.id"), nullable=False)
+    default = db.relationship("AIModel")