From f2e8eabda2a0c01ffe2462ff504ddb54c4d2d11e Mon Sep 17 00:00:00 2001
From: Piero Toffanin <pt@masseranolabs.com>
Date: Sat, 10 Sep 2016 11:24:16 -0400
Subject: [PATCH] Defined models

---
 app/models.py      | 53 ++++++++++++++++++++++++++++++++++++++++------
 webodm/settings.py |  5 ++++-
 2 files changed, 51 insertions(+), 7 deletions(-)

diff --git a/app/models.py b/app/models.py
index d8f8874b..d6a99c14 100644
--- a/app/models.py
+++ b/app/models.py
@@ -1,6 +1,12 @@
 from django.db import models
 from django.utils import timezone
 from django.contrib.auth.models import User
+from django.contrib.postgres import fields
+
+def assets_directory_path(taskId, projectId, filename):
+    # files will be uploaded to MEDIA_ROOT/project_<id>/task_<id>/<filename>
+    return 'project_{0}/task_{1}/{2}'.format(taskId, projectId, filename)
+
 
 class Project(models.Model):
     owner = models.ForeignKey(User, on_delete=models.PROTECT, help_text="The person who created the project")
@@ -11,15 +17,39 @@ class Project(models.Model):
     def __str__(self):
         return self.name
 
+
+class ProcessingNode(models.Model):
+    hostname = models.CharField(max_length=255, help_text="Hostname where the node is located (can be an internal hostname as well)")
+    port = models.PositiveIntegerField(help_text="Port that connects to the node's API")
+    api_version = models.CharField(max_length=32, help_text="API version used by the node")
+    last_refreshed = models.DateTimeField(null=True, help_text="When was the information about this node last retrieved?")
+    queue_count = models.PositiveIntegerField(default=0, help_text="Number of tasks currently being processed by this node (as reported by the node itself)")
+    available_options = fields.JSONField(default=dict(), help_text="Description of the options that can be used for processing")
+    def __str__(self):
+            return '%s %s (%s)'.format(self.hostname, self.port, self.api_version)
+
+
+def gcp_directory_path(task, filename):
+    return assets_directory_path(task.id, task.project.id, filename)
+
 class Task(models.Model):
+    STATUS_CODES = (
+        (10, 'QUEUED'),
+        (20, 'RUNNING'),
+        (30, 'FAILED'),
+        (40, 'COMPLETED'),
+        (50, 'CANCELED')
+    )
+
     uuid = models.CharField(max_length=255, primary_key=True, help_text="Unique identifier of the task (as returned by OpenDroneMap's REST API)")
-    project = models.ForeignKey(Project, on_delete=models.CASCADE)
+    project = models.ForeignKey(Project, on_delete=models.CASCADE, help_text="Project that this task belongs to")
     name = models.CharField(max_length=255, help_text="A label for the task")
     processing_time = models.IntegerField(default=-1, help_text="Number of milliseconds that elapsed since the beginning of this task (-1 indicates that no information is available)")
-    # options
-    console_output = models.TextField(null=True, help_text="Console output of OpenDroneMap's process")
-    # ground_control_points
-    # input_images
+    processing_node = models.ForeignKey(ProcessingNode, null=True, help_text="Processing node assigned to this task (or null if this task has not been associated yet)")
+    status = models.IntegerField(choices=STATUS_CODES, null=True, help_text="Current status of the task")
+    options = fields.JSONField(default=dict(), help_text="Options that are being used to process this task")
+    console_output = models.TextField(null=True, help_text="Console output of the OpenDroneMap's process")
+    ground_control_points = models.FileField(null=True, upload_to=gcp_directory_path, help_text="Optional Ground Control Points file to use for processing")
     # georeferenced_model
     # orthophoto
     # textured_model
@@ -27,4 +57,15 @@ class Task(models.Model):
     created_at = models.DateTimeField(default=timezone.now, help_text="Creation date")
 
     def __str__(self):
-        return '%s %s' % (self.name, self.uuid)
\ No newline at end of file
+        return '%s %s'.format(self.name, self.uuid)
+
+
+def image_directory_path(task, filename):
+    return assets_directory_path(imageUpload.task.id, imageUpload.task.project.id, filename)
+
+class ImageUpload(models.Model):
+    task = models.ForeignKey(Task, on_delete=models.CASCADE, help_text="Task this image belongs to")
+    image = models.ImageField(upload_to=image_directory_path, help_text="File uploaded by a user")
+    
+    def __str__(self):
+        return self.image.name
diff --git a/webodm/settings.py b/webodm/settings.py
index 4d36cf1d..a94b0744 100644
--- a/webodm/settings.py
+++ b/webodm/settings.py
@@ -130,4 +130,7 @@ STATICFILES_DIRS = [
 
 # Auth
 LOGIN_REDIRECT_URL = '/dashboard/'
-LOGIN_URL = '/login/'
\ No newline at end of file
+LOGIN_URL = '/login/'
+
+# File uploads
+MEDIA_ROOT = os.path.join(BASE_DIR, 'media')
\ No newline at end of file