models

zodman · zodman · commit 3518a7cd951f · 2018-03-23T08:28:01.000-06:00
diff --git a/app/urls.py b/app/urls.py
@@ -1,30 +1,9 @@
 from django.conf.urls import url
 from django.contrib import admin
-from django.views.generic import TemplateView
 from django.contrib.staticfiles.urls import staticfiles_urlpatterns
 from django.conf import settings
 from django.conf.urls.static import static
-from tasks.models import Task
-import json 
-
-class TaskView(TemplateView):
-    template_name ="index.html"
-    def process(self, t):
-        task = t.to_dict()
-        for i in ('path', 'depth', 'path', 'numchild'):
-            if i in task:
-                del task[i]
-        task['id'] = str(task['id'])
-        childrens = [c.id for c in t.get_children()]
-        task["dependencies"] = ",".join([str(i) for i in childrens])
-        return task
-    def get_context_data(self, **kwargs):
-        ctx = super(TaskView, self).get_context_data(**kwargs)
-        tasks = Task.objects.all()
-
-        ctx["tasks"] = json.dumps([ self.process(t) for t in tasks])    
-        return ctx
-
+from tasks.views import TaskView
 
 urlpatterns = [
     url(r'^admin/', admin.site.urls),
diff --git a/tasks/migrations/0003_task_custom_class.py b/tasks/migrations/0003_task_custom_class.py
@@ -0,0 +1,19 @@
+# Generated by Django 2.0.3 on 2018-03-23 01:04
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('tasks', '0002_auto_20180322_0444'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='task',
+            name='custom_class',
+            field=models.CharField(default='', max_length=20),
+            preserve_default=False,
+        ),
+    ]
diff --git a/tasks/migrations/0004_auto_20180323_0104.py b/tasks/migrations/0004_auto_20180323_0104.py
@@ -0,0 +1,18 @@
+# Generated by Django 2.0.3 on 2018-03-23 01:04
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('tasks', '0003_task_custom_class'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='task',
+            name='custom_class',
+            field=models.CharField(blank=True, max_length=20, null=True),
+        ),
+    ]
diff --git a/tasks/models.py b/tasks/models.py
@@ -10,6 +10,10 @@ class Task(MP_Node, ToDictMixin):
     start = models.DateField()
     end = models.DateField()
     progress = models.PositiveIntegerField(default=0)
+    custom_class = models.CharField(max_length=20, null=True, blank=True)
+
+    def __str__(self):
+        return self.__unicode__()
 
     def __unicode__(self):
         return u"{}".format(self.name)
diff --git a/tasks/templates/index.html b/tasks/templates/index.html
@@ -19,6 +19,9 @@
             color:white;
             font-size:1rem;
         }
+        .gantt .red .bar-progress {
+                        fill: tomato;
+                                    }
     </style>
   </head>
   <body>
diff --git a/tasks/views.py b/tasks/views.py
@@ -1,6 +1,26 @@
 # -*- coding: utf-8 -*-
 from __future__ import unicode_literals
-
 from django.shortcuts import render
+from django.views.generic import TemplateView
+from tasks.models import Task
+import json 
+
+class TaskView(TemplateView):
+    template_name ="index.html"
+    def process(self, t):
+        task = t.to_dict()
+        for i in ('path', 'depth', 'path', 'numchild'):
+            if i in task:
+                del task[i]
+        task['id'] = str(task['id'])
+        childrens = [c.id for c in t.get_children()]
+        task["dependencies"] = ",".join([str(i) for i in childrens])
+        return task
+
+    def get_context_data(self, **kwargs):
+        ctx = super(TaskView, self).get_context_data(**kwargs)
+        tasks = Task.objects.all()
+
+        ctx["tasks"] = json.dumps([ self.process(t) for t in tasks])    
+        return ctx
 
-# Create your views here.
