bin/djais/views.py

   1 # -*- coding: utf-8 -*-
   2
   3 import os
   4 from datetime import *
   5 from time import time as ctime
   6 import crack
   7 import struct
   8 import rrdtool
   9 from django.http import *
  10 from django.template import loader, RequestContext
  11 from django import forms
  12 from django.shortcuts import render_to_response, get_object_or_404
  13 from django.db import IntegrityError
  14
  15 from decoratedstr import remove_decoration
  16
  17 from basicauth import http_authenticate
  18 from models import *
  19 from show_targets_ships import *
  20 from ais import Nmea, NmeaFeeder, strmmsi_to_mmsi, SHIP_TYPES, STATUS_CODES, AIS_STATUS_NOT_AVAILABLE, AIS_ROT_NOT_AVAILABLE, AIS_LATLON_SCALE, AIS_LON_NOT_AVAILABLE, AIS_LAT_NOT_AVAILABLE, AIS_COG_SCALE, AIS_COG_NOT_AVAILABLE, AIS_NO_HEADING, AIS_SOG_SCALE, AIS_SOG_NOT_AVAILABLE, AIS_SOG_MAX_SPEED, add_nmea1, add_nmea5_partial, load_fleet_to_uset
  21 from ntools import datetime_to_timestamp, clean_ais_charset
  22
  23 STATS_DIR = '/var/lib/ais/stats'
  24
  25 def auth(username, raw_password):
  26     try:
  27         user = User.objects.get(login=username)
  28     except User.DoesNotExist:
  29         return None
  30     if not user.check_password(raw_password):
  31         return None
  32     return user
  33
  34
  35 @http_authenticate(auth, 'ais')
  36 def index(request):
  37     return render_to_response('index.html', {}, RequestContext(request))
  38
  39
  40 class VesselSearchForm(forms.Form):
  41     mmsi = forms.CharField(max_length=9, required=False)
  42     name = forms.CharField(max_length=20, required=False)
  43     imo = forms.IntegerField(required=False)
  44     callsign = forms.CharField(max_length=7, required=False)
  45     destination = forms.CharField(max_length=20, required=False)
  46     def clean(self):
  47         cleaned_data = self.cleaned_data
  48         for value in cleaned_data.values():
  49             if value:
  50                 return cleaned_data
  51         raise forms.ValidationError("You must enter at least one criteria")
  52
  53
  54
  55 @http_authenticate(auth, 'ais')
  56 def vessel_search(request):
  57     if request.method == 'POST' or request.META['QUERY_STRING']:
  58         form = VesselSearchForm(request.REQUEST)
  59         if form.is_valid():
  60             data = form.cleaned_data
  61             vessels = Vessel.objects
  62             if data['mmsi']:
  63                 vessels = vessels.filter(mmsi=strmmsi_to_mmsi(data['mmsi']))
  64             if data['name']:
  65                 vessels = vessels.filter(name__contains=data['name'].upper())
  66             if data['imo']:
  67                 vessels = vessels.filter(imo=data['imo'])
  68             if data['callsign']:
  69                 vessels = vessels.filter(callsign__contains=data['callsign'].upper())
  70             if data['destination']:
  71                 vessels = vessels.filter(destination__contains=data['destination'].upper())
  72             return render_to_response('vessels.html', {'vessels': vessels}, RequestContext(request))
  73     else: # GET
  74         form = VesselSearchForm()
  75
  76     return render_to_response('vessel_index.html', {'form': form}, RequestContext(request))
  77
  78 @http_authenticate(auth, 'ais')
  79 def vessel(request, strmmsi):
  80     os.chdir('/home/nirgal/ais.nirgal.com/ais/') # FIXME
  81     mmsi = strmmsi_to_mmsi(strmmsi)
  82     vessel = get_object_or_404(Vessel, pk=mmsi)
  83     nmea = Nmea.new_from_lastinfo(strmmsi)
  84     #if not nmea.timestamp_1 and not nmea.timestamp_5:
  85     #    raise Http404
  86     return render_to_response('vessel.html', {'nmea': nmea}, RequestContext(request))
  87
  88
  89 class VesselManualInputForm(forms.Form):
  90     timestamp = forms.DateTimeField(label=u'When', help_text=u'When was the observation made in GMT. Use YYYY-MM-DD HH:MM:SS format')
  91     imo = forms.IntegerField(required=False, min_value=1000000, max_value=9999999)
  92     name = forms.CharField(max_length=20, required=False)
  93     callsign = forms.CharField(max_length=7, required=False)
  94     type = forms.TypedChoiceField(required=False, choices = [ kv for kv in SHIP_TYPES.iteritems() if 'reserved' not in kv[1].lower()], coerce=int, empty_value=0, initial=0)
  95     status = forms.TypedChoiceField(required=False, choices = [ kv for kv in STATUS_CODES.iteritems() if 'reserved' not in kv[1].lower()], coerce=int, empty_value=AIS_STATUS_NOT_AVAILABLE, initial=AIS_STATUS_NOT_AVAILABLE)
  96     sog = forms.FloatField(label='Speed', help_text='Over ground, in knots', required=False, min_value=0, max_value=AIS_SOG_MAX_SPEED/AIS_SOG_SCALE)
  97     latitude = forms.CharField(required=False)
  98     longitude = forms.CharField(required=False)
  99     cog = forms.FloatField(label='Course', help_text='Over ground', required=False, min_value=0.0, max_value=359.9)
 100     heading = forms.IntegerField(required=False, min_value=0, max_value=359)
 101
 102     @staticmethod
 103     def _clean_ais_charset(ustr):
 104         ustr = remove_decoration(ustr) # benign cleaning, but can increase size (œ->oe)
 105         ustr = ustr.upper() # benign cleaning
 106         str = clean_ais_charset(ustr.encode('ascii', 'replace'))
 107         if unicode(str) != ustr:
 108             raise forms.ValidationError('Invalid character: AIS alphabet is @ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^- !"#$%&\'()*+,-./0123456789:;<=>?')
 109         return str
 110
 111     def clean_timestamp(self):
 112         data = self.cleaned_data['timestamp']
 113         if data is None:
 114             return None
 115         if data < datetime.utcnow() - timedelta(365):
 116             raise forms.ValidationError('Date is too much is the past.')
 117         if data > datetime.utcnow():
 118             raise forms.ValidationError('Date is be in the future. This form is only for observed results.')
 119         return datetime_to_timestamp(data)
 120
 121     def clean_imo(self):
 122         data = self.cleaned_data['imo']
 123         if data is None:
 124             return 0
 125         return data
 126
 127     def clean_name(self):
 128         name = self.cleaned_data['name']
 129         if name is None:
 130             return ''
 131         name = VesselManualInputForm._clean_ais_charset(name)
 132         if len(name)>20:
 133             raise forms.ValidationError('Ensure this value has at most 20 characters (it has %s).' % len(name))
 134         return name
 135
 136     def clean_callsign(self):
 137         callsign = self.cleaned_data['callsign']
 138         if callsign is None:
 139             return ''
 140         callsign = VesselManualInputForm._clean_ais_charset(callsign)
 141         if len(callsign)>7:
 142             raise forms.ValidationError('Ensure this value has at most 7 characters (it has %s).' % len(callsign))
 143         return callsign
 144
 145     def clean_sog(self):
 146         sog = self.cleaned_data['sog']
 147         if sog is None:
 148             return AIS_SOG_NOT_AVAILABLE
 149         return int(sog*AIS_SOG_SCALE)
 150
 151     def clean_latitude(self):
 152         data = self.cleaned_data['latitude']
 153         data = data.replace(u"''", u'"') # commong mistake
 154         data = data.replace(u' ', u'') # remove spaces
 155         sides = u'SN'
 156         if not data:
 157             return AIS_LAT_NOT_AVAILABLE
 158         tmp, side = data[:-1], data[-1]
 159         if side == sides[0]:
 160             side = -1
 161         elif side == sides[1]:
 162             side = 1
 163         else:
 164             raise forms.ValidationError(u'Last character must be either %s or %s.' % (sides[0], sides[1]))
 165         spl = tmp.split(u'°')
 166         if len(spl) == 1:
 167             raise forms.ValidationError(u'You need to use the ° character.')
 168         d, tmp = spl
 169         try:
 170             d = float(d)
 171         except ValueError:
 172             raise forms.ValidationError(u'Degrees must be an number. It\'s %s.' % d)
 173         spl = tmp.split(u"'", 1)
 174         if len(spl) == 1:
 175             # no ' sign: ok only if there is nothing but the side after °
 176             # we don't accept seconds if there is no minutes:
 177             # It might be an entry mistake
 178             tmp = spl[0]
 179             if len(tmp) == 0:
 180                 m = s = 0
 181             else:
 182                 raise forms.ValidationError(u'You must use the \' character between ° and %s.' % data[-1])
 183         else:
 184             m, tmp = spl
 185             try:
 186                 m = float(m)
 187             except ValueError:
 188                 raise forms.ValidationError(u'Minutes must be an number. It\'s %s.' % m)
 189             if len(tmp) == 0:
 190                 s = 0
 191             else:
 192                 if tmp[-1] != '"':
 193                     raise forms.ValidationError(u'You must use the " character between seconds and %s.' % data[-1])
 194                 s = tmp[:-1]
 195                 try:
 196                     s = float(s)
 197                 except ValueError:
 198                     raise forms.ValidationError(u'Seconds must be an number. It\'s %s.' % s)
 199         data = side * ( d + m / 60 + s / 3600)
 200
 201         if data < -90 or data > 90:
 202             raise forms.ValidationError(u'%s in not in -90..90 range' % data)
 203         return int(data * AIS_LATLON_SCALE)
 204
 205     def clean_longitude(self):
 206         data = self.cleaned_data['longitude']
 207         data = data.replace(u"''", u'"') # commong mistake
 208         data = data.replace(u' ', u'') # remove spaces
 209         sides = u'WE'
 210         if not data:
 211             return AIS_LON_NOT_AVAILABLE
 212         tmp, side = data[:-1], data[-1]
 213         if side == sides[0]:
 214             side = -1
 215         elif side == sides[1]:
 216             side = 1
 217         else:
 218             raise forms.ValidationError(u'Last character must be either %s or %s.' % (sides[0], sides[1]))
 219         spl = tmp.split(u'°')
 220         if len(spl) == 1:
 221             raise forms.ValidationError(u'You need to use the ° character.')
 222         d, tmp = spl
 223         try:
 224             d = float(d)
 225         except ValueError:
 226             raise forms.ValidationError(u'Degrees must be an number. It\'s %s.' % d)
 227         spl = tmp.split(u"'", 1)
 228         if len(spl) == 1:
 229             # no ' sign: ok only if there is nothing but the side after °
 230             # we don't accept seconds if there is no minutes:
 231             # It might be an entry mistake
 232             tmp = spl[0]
 233             if len(tmp) == 0:
 234                 m = s = 0
 235             else:
 236                 raise forms.ValidationError(u'You must use the \' character between ° and %s.' % data[-1])
 237         else:
 238             m, tmp = spl
 239             try:
 240                 m = float(m)
 241             except ValueError:
 242                 raise forms.ValidationError(u'Minutes must be an number. It\'s %s.' % m)
 243             if len(tmp) == 0:
 244                 s = 0
 245             else:
 246                 if tmp[-1] != '"':
 247                     raise forms.ValidationError(u'You must use the " character between seconds and %s.' % data[-1])
 248                 s = tmp[:-1]
 249                 try:
 250                     s = float(s)
 251                 except ValueError:
 252                     raise forms.ValidationError(u'Seconds must be an number. It\'s %s.' % s)
 253         data = side * ( d + m / 60 + s / 3600)
 254
 255         if data < -180 or data > 180:
 256             raise forms.ValidationError(u'%s in not in -180..180 range' % data)
 257         return int(data * AIS_LATLON_SCALE)
 258
 259     def clean_cog(self):
 260         data = self.cleaned_data['cog']
 261         if data is None:
 262             return AIS_COG_NOT_AVAILABLE
 263         return int(data * AIS_COG_SCALE)
 264
 265     def clean_heading(self):
 266         #raise forms.ValidationError(u'clean_heading called')
 267         data = self.cleaned_data['heading']
 268         if data is None:
 269             return AIS_NO_HEADING
 270         return data
 271
 272     def clean(self):
 273         cleaned_data = self.cleaned_data
 274         if (cleaned_data.get('latitude', AIS_LAT_NOT_AVAILABLE) == AIS_LAT_NOT_AVAILABLE ) ^ ( cleaned_data.get('longitude', AIS_LON_NOT_AVAILABLE) == AIS_LON_NOT_AVAILABLE):
 275             raise forms.ValidationError('It makes no sense to enter just a latitude or a longitude. Enter both or none.')
 276         if cleaned_data.get('latitude', AIS_LAT_NOT_AVAILABLE) == AIS_LAT_NOT_AVAILABLE:
 277             if cleaned_data.get('status', AIS_STATUS_NOT_AVAILABLE) != AIS_STATUS_NOT_AVAILABLE:
 278                 raise forms.ValidationError('It makes no sense to enter a status without coordinates. Please enter latitute and longitude too.')
 279             if cleaned_data.get('sog', AIS_SOG_NOT_AVAILABLE) != AIS_SOG_NOT_AVAILABLE:
 280                 raise forms.ValidationError('It makes no sense to enter a speed without coordinates. Please enter latitute and longitude too.')
 281             if cleaned_data.get('cog', AIS_COG_NOT_AVAILABLE) != AIS_COG_NOT_AVAILABLE:
 282                 raise forms.ValidationError('It makes no sense to enter a course without coordinates. Please enter latitute and longitude too.')
 283             if cleaned_data.get('heading', AIS_NO_HEADING) != AIS_NO_HEADING:
 284                 raise forms.ValidationError('It makes no sense to enter a heading without coordinates. Please enter latitute and longitude too.')
 285
 286         if cleaned_data.get('timestamp', None) \
 287         and cleaned_data.get('imo', 0) == 0 \
 288         and cleaned_data.get('name', '') == '' \
 289         and cleaned_data.get('callsign', '') == '' \
 290         and cleaned_data.get('type', 0) == 0 \
 291         and cleaned_data.get('status', AIS_STATUS_NOT_AVAILABLE) == AIS_STATUS_NOT_AVAILABLE \
 292         and cleaned_data.get('sog', AIS_SOG_NOT_AVAILABLE) == AIS_SOG_NOT_AVAILABLE \
 293         and cleaned_data.get('latitude', AIS_LAT_NOT_AVAILABLE) == AIS_LAT_NOT_AVAILABLE \
 294         and cleaned_data.get('longitude', AIS_LON_NOT_AVAILABLE) == AIS_LON_NOT_AVAILABLE \
 295         and cleaned_data.get('cog', AIS_COG_NOT_AVAILABLE) == AIS_COG_NOT_AVAILABLE \
 296         and cleaned_data.get('heading', AIS_NO_HEADING) == AIS_NO_HEADING:
 297             raise forms.ValidationError("You must enter some data, beside when.")
 298         return cleaned_data
 299
 300 @http_authenticate(auth, 'ais')
 301 def vessel_manual_input(request, strmmsi):
 302     strmmsi = strmmsi.encode('utf-8')
 303     nmea = Nmea.new_from_lastinfo(strmmsi)
 304     if request.method == 'POST' or request.META['QUERY_STRING']:
 305         form = VesselManualInputForm(request.REQUEST)
 306         if form.is_valid():
 307             data = form.cleaned_data
 308             source = 'U' +  struct.pack('<I', request.user.id)[0:3]
 309             result = u''
 310             if data['imo'] != 0 \
 311             or data['name'] != '' \
 312             or data['callsign'] != '' \
 313             or data['type'] != 0:
 314                 toto = (strmmsi, data['timestamp'], data['imo'], data['name'], data['callsign'], data['type'], 0,0,0,0, 0,0,24,60, 0, '', source)
 315                 result += 'UPDATING NMEA 5: '+repr(toto)+'<br>'
 316                 add_nmea5_partial(*toto)
 317             if data['status'] != AIS_STATUS_NOT_AVAILABLE \
 318             or data['sog'] != AIS_SOG_NOT_AVAILABLE \
 319             or data['latitude'] != AIS_LAT_NOT_AVAILABLE \
 320             or data['longitude'] != AIS_LON_NOT_AVAILABLE \
 321             or data['cog'] != AIS_COG_NOT_AVAILABLE \
 322             or data['heading'] != AIS_NO_HEADING:
 323
 324                 toto = (strmmsi, data['timestamp'], data['status'], AIS_ROT_NOT_AVAILABLE, data['sog'], data['latitude'], data['longitude'], data['cog'], data['heading'], source)
 325                 result += 'UPDATING NMEA 1: '+repr(toto)+'<br>'
 326                 add_nmea1(*toto)
 327             return HttpResponse('Not fully implemented: '+repr(data) + '<br>' + result)
 328     else: # GET
 329         form = VesselManualInputForm()
 330     return render_to_response('vessel_manual_input.html', {'form': form, 'nmea': nmea}, RequestContext(request))
 331
 332 @http_authenticate(auth, 'ais')
 333 def vessel_track(request, strmmsi):
 334     ndays = request.REQUEST.get('ndays', 90)
 335     try:
 336         ndays = int(ndays)
 337     except ValueError:
 338         ndays = 90
 339     grain = request.REQUEST.get('grain', 3600)
 340     try:
 341         grain = int(grain)
 342     except ValueError:
 343         grain = 3600
 344     os.chdir('/home/nirgal/ais.nirgal.com/ais/') # FIXME
 345     date_end = datetime.utcnow()
 346     date_start = date_end - timedelta(ndays)
 347     nmea_iterator = NmeaFeeder(strmmsi, date_end, date_start, granularity=grain)
 348     value = kml_to_kmz(format_boat_track(nmea_iterator))
 349     response = HttpResponse(value, mimetype="application/vnd.google-earth.kml")
 350     response['Content-Disposition'] = 'attachment; filename=%s.kmz' % strmmsi
 351     return response
 352
 353
 354 @http_authenticate(auth, 'ais')
 355 def vessel_animation(request, strmmsi):
 356     ndays = request.REQUEST.get('ndays', 90)
 357     try:
 358         ndays = int(ndays)
 359     except ValueError:
 360         ndays = 90
 361     grain = request.REQUEST.get('grain', 3600)
 362     try:
 363         grain = int(grain)
 364     except ValueError:
 365         grain = 3600
 366     os.chdir('/home/nirgal/ais.nirgal.com/ais/') # FIXME
 367     date_end = datetime.utcnow()
 368     date_start = date_end - timedelta(ndays)
 369     nmea_iterator = NmeaFeeder(strmmsi, date_end, date_start, granularity=grain)
 370     value = kml_to_kmz(format_boat_intime(nmea_iterator))
 371     response = HttpResponse(value, mimetype="application/vnd.google-earth.kml")
 372     response['Content-Disposition'] = 'attachment; filename=%s.kmz' % strmmsi
 373     return response
 374
 375
 376 @http_authenticate(auth, 'ais')
 377 def fleets(request):
 378     fleetusers = request.user.fleetuser_set.all()
 379     return render_to_response('fleets.html', {'fleetusers':fleetusers}, RequestContext(request))
 380
 381
 382 @http_authenticate(auth, 'ais')
 383 def fleet(request, fleetname):
 384     fleet = get_object_or_404(Fleet, pk=fleetname)
 385     if not FleetUser.objects.filter(fleet=fleetname, user=request.user.id).all():
 386         return HttpResponseForbidden('<h1>Forbidden</h1>')
 387     return render_to_response('fleet.html', {'fleet':fleet}, RequestContext(request))
 388
 389
 390 @http_authenticate(auth, 'ais')
 391 def fleet_vessels(request, fleetname):
 392     fleet = get_object_or_404(Fleet, pk=fleetname)
 393     if not FleetUser.objects.filter(fleet=fleetname, user=request.user.id).all():
 394         return HttpResponseForbidden('<h1>Forbidden</h1>')
 395     vessels = fleet.vessel.all()
 396     return render_to_response('fleet_vessels.html', {'fleet':fleet, 'vessels': vessels}, RequestContext(request))
 397
 398
 399 @http_authenticate(auth, 'ais')
 400 def fleet_vessel_add(request, fleetname):
 401     fleet = get_object_or_404(Fleet, pk=fleetname)
 402     if not FleetUser.objects.filter(fleet=fleetname, user=request.user.id).all():
 403         return HttpResponseForbidden('<h1>Forbidden</h1>')
 404     strmmsi = request.REQUEST['mmsi']
 405     mmsi = strmmsi_to_mmsi(strmmsi)
 406     try:
 407         vessel = Vessel.objects.get(pk=mmsi)
 408     except Vessel.DoesNotExist:
 409         return HttpResponse('No such vessel')
 410     try:
 411         FleetVessel(fleet=fleet, vessel=vessel).save()
 412     except IntegrityError:
 413         return HttpResponse('Integrity error: Is the ship already in that fleet?')
 414     return HttpResponse('Done')
 415
 416
 417 class FleetAddVessel(forms.Form):
 418     mmsi = forms.CharField(help_text=u'Enter one MMSI per line', required=False, widget=forms.Textarea)
 419     #name = forms.CharField(max_length=20, required=False)
 420     #imo = forms.IntegerField(required=False)
 421     #callsign = forms.CharField(max_length=7, required=False)
 422     #destination = forms.CharField(max_length=20, required=False)
 423     def clean(self):
 424         cleaned_data = self.cleaned_data
 425         for value in cleaned_data.values():
 426             if value:
 427                 return cleaned_data
 428         raise forms.ValidationError("You must enter at least one criteria")
 429
 430 @http_authenticate(auth, 'ais')
 431 def fleet_vessel_add2(request, fleetname):
 432     fleet = get_object_or_404(Fleet, pk=fleetname)
 433     if not FleetUser.objects.filter(fleet=fleetname, user=request.user.id).all():
 434         return HttpResponseForbidden('<h1>Forbidden</h1>')
 435     if request.method == 'POST' or request.META['QUERY_STRING']:
 436         form = FleetAddVessel(request.REQUEST)
 437         if form.is_valid():
 438             data = form.cleaned_data
 439             result = []
 440             a_strmmsi = request.REQUEST['mmsi']
 441             if a_strmmsi:
 442                 for strmmsi in a_strmmsi.split('\n'):
 443                     strmmsi = strmmsi.strip('\r')
 444                     if not strmmsi:
 445                         continue
 446                     try:
 447                         sqlmmsi = strmmsi_to_mmsi(strmmsi)
 448                     except AssertionError:
 449                         result.append('Invalid mmsi %s' % strmmsi)
 450                         continue
 451                     try:
 452                         vessel = Vessel.objects.get(pk=sqlmmsi)
 453                     except Vessel.DoesNotExist:
 454                         result.append('No vessel with MMSI '+strmmsi)
 455                         continue
 456                     try:
 457                         fv = FleetVessel.objects.get(fleet=fleet, vessel=vessel)
 458                         result.append('Vessel with MMSI %s is already in that fleet' % strmmsi)
 459                     except FleetVessel.DoesNotExist:
 460                         FleetVessel(fleet=fleet, vessel=vessel).save()
 461                         result.append('Vessel with MMSI %s added' % strmmsi)
 462
 463             return HttpResponse('<br>'.join(result))
 464     else: # GET
 465         form = FleetAddVessel()
 466
 467     return render_to_response('fleet_vessel_add.html', {'form': form, 'fleet': fleet}, RequestContext(request))
 468
 469
 470 @http_authenticate(auth, 'ais')
 471 def fleet_users(request, fleetname):
 472     fleet = get_object_or_404(Fleet, pk=fleetname)
 473     if not FleetUser.objects.filter(fleet=fleetname, user=request.user.id).all():
 474         return HttpResponseForbidden('<h1>Forbidden</h1>')
 475
 476     message = u''
 477     if request.method == 'POST' or request.META['QUERY_STRING']:
 478         action = request.REQUEST['action']
 479         userlogin = request.REQUEST['user']
 480         try:
 481             user = User.objects.get(login=userlogin)
 482         except User.DoesNotExist:
 483             message = u'User %s does not exist.' % userlogin
 484         else:
 485             if action == u'add':
 486                 try:
 487                     fu = FleetUser.objects.get(fleet=fleet, user=user)
 488                     message = u'User %s already has access.' % user.login
 489                 except FleetUser.DoesNotExist:
 490                     FleetUser(fleet=fleet, user=user).save()
 491                     message = u'Granted access to user %s.' % user.login
 492             elif action == u'revoke':
 493                 try:
 494                     fu = FleetUser.objects.get(fleet=fleet, user=user)
 495                     fu.delete()
 496                     message = u'Revoked access to user %s.' % user.login
 497                 except FleetUser.DoesNotExist:
 498                     message = u'User %s didn\'t have access.' % user.login
 499             else:
 500                 message = u'Unknown action %s' % action
 501
 502     fleetusers = fleet.fleetuser_set.all()
 503     otherusers = User.objects.exclude(id__in=[fu.user.id for fu in fleetusers]).order_by('name')
 504     return render_to_response('fleet_users.html', {'fleet':fleet, 'fleetusers': fleetusers, 'otherusers': otherusers, 'message': message }, RequestContext(request))
 505
 506
 507 @http_authenticate(auth, 'ais')
 508 def fleet_lastpos(request, fleetname):
 509     fleet = get_object_or_404(Fleet, pk=fleetname)
 510     if not FleetUser.objects.filter(fleet=fleetname, user=request.user.id).all():
 511         return HttpResponseForbidden('<h1>Forbidden</h1>')
 512     fleet_uset = load_fleet_to_uset(fleetname)
 513     # = set([mmsi_to_strmmsi(vessel.mmsi) for vessel in fleet.vessel.all()])
 514     os.chdir('/home/nirgal/ais.nirgal.com/ais/') # FIXME
 515     value = kml_to_kmz(format_fleet(fleet_uset, document_name=fleetname+' fleet').encode('utf-8'))
 516     response = HttpResponse(value, mimetype="application/vnd.google-earth.kml")
 517     response['Content-Disposition'] = 'attachment; filename=%s.kmz' % fleetname
 518     return response
 519
 520
 521 @http_authenticate(auth, 'ais')
 522 def users(request):
 523     users = User.objects.all()
 524     for user in users:
 525         user.admin_ok = user.is_admin_by(request.user.id)
 526     return render_to_response('users.html', {'users':users}, RequestContext(request))
 527
 528
 529 class UserEditForm(forms.Form):
 530     login = forms.RegexField(regex=r'^[a-zA-Z0-9_]+$', max_length=16,
 531         error_message ='Login must only contain letters, numbers and underscores')
 532     name = forms.CharField(max_length=50)
 533     email = forms.EmailField()
 534     def __init__(self, *args, **kargs):
 535         forms.Form.__init__(self, *args, **kargs)
 536         self.old_login = kargs['initial']['login']
 537     def clean_login(self):
 538         new_login = self.cleaned_data['login']
 539         if new_login != self.old_login:
 540             if  User.objects.filter(login=new_login).count():
 541                 raise forms.ValidationError("Sorry that login is already in use. Try another one.")
 542         return new_login
 543
 544 @http_authenticate(auth, 'ais')
 545 def user_detail(request, login):
 546     user = get_object_or_404(User, login=login)
 547     user.admin_ok = user.is_admin_by(request.user.id)
 548     return render_to_response('user_detail.html', {'auser': user}, RequestContext(request))
 549
 550 @http_authenticate(auth, 'ais')
 551 def user_edit(request, login):
 552     initial = {}
 553     if login:
 554         user = get_object_or_404(User, login=login)
 555         if not user.is_admin_by(request.user.id):
 556             return HttpResponseForbidden('403 Forbidden')
 557     else:
 558         user = User()
 559         user.father_id = request.user.id;
 560     initial['login'] = user.login
 561     initial['name'] = user.name
 562     initial['email'] = user.email
 563     if request.method == 'POST':
 564         form = UserEditForm(request.POST, initial=initial)
 565         if form.is_valid():
 566             user.login = form.cleaned_data['login']
 567             user.name = form.cleaned_data['name']
 568             user.email = form.cleaned_data['email']
 569             user.save()
 570             return HttpResponseRedirect('/user/')
 571     else: # GET
 572         form = UserEditForm(initial=initial)
 573
 574     return render_to_response('user_edit.html', {'form':form, 'auser': user}, RequestContext(request))
 575
 576
 577 class ChangePasswordForm(forms.Form):
 578     new_password = forms.CharField(max_length=16, widget=forms.PasswordInput())
 579     new_password_check = forms.CharField(max_length=16, widget=forms.PasswordInput())
 580     def clean_generic_password(self, field_name):
 581         password = self.cleaned_data[field_name]
 582         try:
 583             crack.FascistCheck(password)
 584         except ValueError, err:
 585             raise forms.ValidationError(err.message)
 586         return password
 587
 588     def clean_new_password(self):
 589         return self.clean_generic_password('new_password')
 590     def clean_new_password_check(self):
 591         return self.clean_generic_password('new_password_check')
 592     def clean(self):
 593         cleaned_data = self.cleaned_data
 594         pass1 = cleaned_data.get('new_password')
 595         pass2 = cleaned_data.get('new_password_check')
 596         if pass1 != pass2:
 597             self._errors['new_password_check'] = forms.util.ErrorList(['Passwords check must match'])
 598             del cleaned_data['new_password_check']
 599         return cleaned_data
 600
 601
 602 @http_authenticate(auth, 'ais')
 603 def user_change_password(request, login):
 604     user = get_object_or_404(User, login=login)
 605     if not user.is_admin_by(request.user.id):
 606         return HttpResponseForbidden('403 Forbidden')
 607     if request.method == 'POST':
 608         form = ChangePasswordForm(request.POST)
 609         if form.is_valid():
 610             user.set_password(form.cleaned_data['new_password'])
 611             user.save()
 612             return HttpResponseRedirect('/user/')
 613     else: # GET
 614         form = ChangePasswordForm()
 615     return render_to_response('user_change_password.html', {'form':form, 'auser':user}, RequestContext(request))
 616
 617
 618 @http_authenticate(auth, 'ais')
 619 def user_delete(request, login):
 620     user = get_object_or_404(User, login=login)
 621     if not user.is_admin_by(request.user.id):
 622         return HttpResponseForbidden('403 Forbidden')
 623     if request.REQUEST.get('confirm', None):
 624         user.delete()
 625         return HttpResponseRedirect('/user/')
 626     return render_to_response('user_delete.html', {'form':None, 'auser':user}, RequestContext(request))
 627
 628
 629 def logout(request):
 630     # TODO
 631     return HttpResponse('Not implemented')
 632     #response = render_to_response('logout.html', {}, RequestContext(request))
 633     #return response
 634
 635 @http_authenticate(auth, 'ais')
 636 def sources(request):
 637     os.chdir('/home/nirgal/ais.nirgal.com/ais/') # FIXME
 638     png_stat_dir = '/home/nirgal/ais.nirgal.com/www/stats'
 639     sources = ( 'NMMT', 'NMKT', 'NMRW', 'NMEZ', 'NMAS', 'NMAH' )
 640     now = int(ctime())
 641     periods = ({
 642         'name_tiny': '2h',
 643         'name_long': '2 hours',
 644         'seconds': 2*60*60
 645         }, {
 646         'name_tiny': '6h',
 647         'name_long': '6 hours',
 648         'seconds': 6*60*60,
 649         }, {
 650         'name_tiny': '2d',
 651         'name_long': '2 days',
 652         'seconds': 2*24*60*60
 653         #}, {
 654         #'name_tiny': '7d',
 655         #'name_long': '1 week',
 656         #'seconds': 7*24*60*60
 657         })
 658
 659     for source in sources:
 660         for period in periods:
 661             args = os.path.join(STATS_DIR, source+'-'+period['name_tiny']+'-bytes.png'), \
 662                 '--lazy', \
 663                 '-l', '0', \
 664                 '--title', source+' - Bandwidth usage - '+period['name_long'], \
 665                 '--start', '%d' % (now-period['seconds']), \
 666                 '--end', '%d' % now, \
 667                 '--vertical-label', 'bps', \
 668                 'DEF:bytes=%s:bytes:AVERAGE' % os.path.join(STATS_DIR, source+'.rrd'), \
 669                 'DEF:rawbytes=%s:rawbytes:AVERAGE' % os.path.join(STATS_DIR, source+'.rrd'), \
 670                 'CDEF:bits=bytes,8,*', \
 671                 'CDEF:rawbits=rawbytes,8,*', \
 672                 'LINE:bits#00FF00:IP payload', \
 673                 'LINE:rawbits#FF0000:IP with headers'
 674             rrdtool.graph(*args)
 675             args = os.path.join(STATS_DIR, source+'-'+period['name_tiny']+'-counts.png'), \
 676                 '--lazy', \
 677                 '-l', '0', \
 678                 '--title', source+' - Packet\'izer stats - '+period['name_long'], \
 679                 '--start', '%d' % (now-period['seconds']), \
 680                 '--end', '%d' % now, \
 681                 '--vertical-label', 'Hz', \
 682                 'DEF:packets=%s:packets:AVERAGE' % os.path.join(STATS_DIR, source+'.rrd'), \
 683                 'DEF:lines=%s:lines:AVERAGE' % os.path.join(STATS_DIR, source+'.rrd'), \
 684                 'LINE:packets#FF0000:input packets', \
 685                 'LINE:lines#00FF00:AIVDM lines'
 686             rrdtool.graph(*args)
 687
 688     return render_to_response('sources.html', {'sources':sources, 'periods': periods}, RequestContext(request))