refactor DB

euclid/database/models/components/base_models.py

@@ -0,0 +1,24 @@
+from peewee import *
+from datetime import datetime
+from uuid import uuid4
+
+from const import db
+
+
+# Establish connection to the database
+database = PostgresqlDatabase(
+    db.DB_NAME,
+    user=db.DB_USER,
+    password=db.DB_PASSWORD,
+    host=db.DB_HOST,
+    port=db.DB_PORT
+)
+
+
+class BaseModel(Model):
+    id = UUIDField(primary_key=True, default=uuid4)
+    created_at = DateTimeField(default=datetime.now)
+    updated_at = DateTimeField(default=datetime.now)
+
+    class Meta:
+        database = database

euclid/database/models/components/progress_step.py

@@ -0,0 +1,16 @@
+from peewee import *
+
+from playhouse.postgres_ext import BinaryJSONField
+
+from database.models.components.base_models import BaseModel
+from database.models.app import App
+
+
+class ProgressStep(BaseModel):
+    app = ForeignKeyField(App, primary_key=True)
+    step = CharField()
+    data = BinaryJSONField(null=True)
+    messages = BinaryJSONField(null=True)
+    app_data = BinaryJSONField()
+    completed = BooleanField(default=False)
+    completed_at = DateTimeField(null=True)