diff options
Diffstat (limited to 'snakeplan/projects/views/projects.py')
-rw-r--r-- | snakeplan/projects/views/projects.py | 17 |
1 files changed, 14 insertions, 3 deletions
diff --git a/snakeplan/projects/views/projects.py b/snakeplan/projects/views/projects.py index 78de64c..7cc4017 100644 --- a/snakeplan/projects/views/projects.py +++ b/snakeplan/projects/views/projects.py | |||
@@ -1,6 +1,10 @@ | |||
1 | from django.views.generic import list_detail | 1 | from django.views.generic import list_detail, create_update |
2 | from django.core.urlresolvers import reverse | ||
3 | |||
2 | from snakeplan.projects.models import Project | 4 | from snakeplan.projects.models import Project |
3 | from snakeplan.projects.models import Iteration | 5 | from snakeplan.projects.models import Iteration |
6 | from snakeplan.projects.forms import ProjectForm | ||
7 | |||
4 | 8 | ||
5 | 9 | ||
6 | def index(request): | 10 | def index(request): |
@@ -12,8 +16,8 @@ def index(request): | |||
12 | 16 | ||
13 | 17 | ||
14 | def project_iterations(request, project_id): | 18 | def project_iterations(request, project_id): |
15 | iterations = Iteration.objects.filter(project=project_id) | 19 | project = Project.objects.get(id=project_id) |
16 | project = iterations[0].project | 20 | iterations = project.iteration_set.all() |
17 | 21 | ||
18 | return list_detail.object_list( | 22 | return list_detail.object_list( |
19 | request=request, | 23 | request=request, |
@@ -21,3 +25,10 @@ def project_iterations(request, project_id): | |||
21 | extra_context={'project_name': project}, | 25 | extra_context={'project_name': project}, |
22 | allow_empty=True | 26 | allow_empty=True |
23 | ) | 27 | ) |
28 | |||
29 | |||
30 | def create_project(request): | ||
31 | post_save_redirect = '/project/%(id)s/' | ||
32 | return create_update.create_object(request, | ||
33 | form_class=ProjectForm, | ||
34 | post_save_redirect=post_save_redirect) | ||