1 files changed, 97 insertions, 0 deletions
diff --git a/archaeological_operations/views.py b/archaeological_operations/views.py
new file mode 100644
index 000000000..27ebd60e9
--- /dev/null
+++ b/archaeological_operations/views.py
@@ -0,0 +1,97 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# Copyright (C) 2010-2012  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as
+# published by the Free Software Foundation, either version 3 of the
+# License, or (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Affero General Public License for more details.
+
+# You should have received a copy of the GNU Affero General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+# See the file COPYING for details.
+
+import json
+
+from django.db.models import Q
+from django.http import HttpResponse
+from django.shortcuts import render_to_response
+
+from ishtar_common.views import get_item, show_item, revert_item
+import models
+
+def autocomplete_operation(request, non_closed=True):
+    person_types = request.user.ishtaruser.person.person_type
+    if (not request.user.has_perm('ishtar_common.view_operation', models.Operation)\
+       and not request.user.has_perm('ishtar_common.view_own_operation',
+                                                              models.Operation)
+       and not person_types.rights.filter(wizard__url_name='operation_search'
+                                                                     ).count()):
+        return HttpResponse(mimetype='text/plain')
+    if not request.GET.get('term'):
+        return HttpResponse(mimetype='text/plain')
+    q = request.GET.get('term')
+    query = Q()
+    for q in q.split(' '):
+        extra = Q(towns__name__icontains=q)
+        try:
+            value = int(q)
+            extra = extra | Q(year=q) | Q(operation_code=q)
+        except ValueError:
+            pass
+        query = query & extra
+    if non_closed:
+        query = query & Q(end_date__isnull=True)
+    limit = 15
+    operations = models.Operation.objects.filter(query)[:limit]
+    data = json.dumps([{'id':operation.pk, 'value':unicode(operation)}
+                                          for operation in operations])
+    return HttpResponse(data, mimetype='text/plain')
+
+def get_available_operation_code(request, year=None):
+    if not request.user.has_perm('ishtar_common.view_operation', models.Operation)\
+       and not request.user.has_perm('ishtar_common.view_own_operation',
+                                                              models.Operation):
+        return HttpResponse(mimetype='text/plain')
+    data = json.dumps({'id':models.Operation.get_available_operation_code(year)})
+    return HttpResponse(data, mimetype='text/plain')
+
+get_operation = get_item(models.Operation, 'get_operation', 'operation',
+      bool_fields = ['end_date__isnull'],
+      extra_request_keys={'common_name':'common_name__icontains',
+                          'end_date':'end_date__isnull',
+                          'year_index':('year', 'operation_code')})
+show_operation = show_item(models.Operation, 'operation')
+revert_operation = revert_item(models.Operation)
+
+get_operationsource = get_item(models.OperationSource,
+        'get_operationsource', 'operationsource',
+      extra_request_keys={'operation__towns':'operation__towns__pk',
+                  'operation__operation_type':'operation__operation_type__pk',
+                  'operation__year':'operation__year'})
+
+get_administrativeactfile = get_item(models.AdministrativeAct,
+        'get_administrativeactfile', 'administrativeactfile',
+      extra_request_keys={'associated_file__towns':'associated_file__towns__pk',
+                          'operation__towns':'operation__towns__pk',
+                          'act_type__intented_to':'act_type__intented_to'})
+get_administrativeactop = get_item(models.AdministrativeAct,
+        'get_administrativeactop', 'administrativeactop',
+      extra_request_keys={'associated_file__towns':'associated_file__towns__pk',
+                          'operation__towns':'operation__towns__pk',
+                          'act_type__intented_to':'act_type__intented_to'})
+
+
+def dashboard_operation(request, dct, obj_id=None, *args, **kwargs):
+    """
+    Operation dashboard
+    """
+    dct = {'dashboard': models.OperationDashboard()}
+    return render_to_response('dashboard_operation.html', dct,
+                              context_instance=RequestContext(request))