# Eve W-Space
# Copyright (C) 2013 Andrew Austin and other contributors
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version. An additional term under section
# 7 of the GPL is included in the LICENSE file.
#
# 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
# Create your views here.
from models import Fleet, UserLog, SiteType, SiteRecord, UserSite
from Map.models import System
from django.core.exceptions import PermissionDenied
from django.http import HttpResponse
from django.template.response import TemplateResponse
from django.contrib.auth import get_user_model
from django.contrib.auth.decorators import login_required, permission_required
from django.shortcuts import get_object_or_404, get_list_or_404
from django.conf import settings
from datetime import datetime
import pytz
import csv
User = get_user_model()
def require_boss():
def _dec(view_func):
def _view(request, fleetID, *args, **kwargs):
fleet = get_object_or_404(Fleet, pk=fleetID)
if fleet.current_boss != request.user or fleet.ended:
raise PermissionDenied
else:
return view_func(request, fleetID, *args, **kwargs)
_view.__name__ = view_func.__name__
_view.__doc__ = view_func.__doc__
_view.__dict__ = view_func.__dict__
return _view
return _dec
@permission_required('SiteTracker.can_sitetracker')
[docs]def status_bar(request):
"""
Return a template with just the ST status bar tag.
"""
return TemplateResponse(request, 'st_bar_refresh.html')
@permission_required('SiteTracker.can_sitetracker')
[docs]def create_fleet(request):
"""
Takes a system ID via POST and makes a sitetracker fleet with the
requesting user as initial and current boss. Then adds the requestor
to the fleet as a member.
"""
if not request.is_ajax():
raise PermissionDenied
sysid = request.POST.get('sysID', None)
system = get_object_or_404(System, pk=sysid)
fleet = Fleet(initial_boss=request.user, current_boss=request.user,
system=system)
fleet.save()
UserLog(fleet=fleet, user=request.user).save()
return HttpResponse()
@permission_required('SiteTracker.can_sitetracker')
[docs]def leave_fleet(request, fleetID=None):
"""
Leaves the given fleet. If fleetID is not provided, leave all fleets.
"""
if not request.is_ajax():
raise PermissionDenied
if fleetID:
fleet = get_object_or_404(Fleet, pk=fleetID)
fleet.leave_fleet(request.user)
return HttpResponse()
else:
for log in request.user.sitetrackerlogs.filter(leavetime=None).all():
log.fleet.leave_fleet(request.user)
return HttpResponse()
@require_boss()
[docs]def kick_member(request, fleetID, memberID):
"""
Removes a member from the fleet.
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
member = get_list_or_404(fleet.members, leavetime=None, user__pk=memberID)
for mem in member:
fleet.leave_fleet(mem.user)
return HttpResponse()
@permission_required('SiteTracker.can_sitetracker')
@permission_required('SiteTracker.can_sitetracker')
[docs]def join_fleet(request, fleetID):
"""
Join the current user to the fleet.
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
fleet.join_fleet(request.user)
return HttpResponse()
@require_boss()
[docs]def disband_fleet(request, fleetID):
"""
Disband the fleet.
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
fleet.close_fleet()
return HttpResponse()
@require_boss()
[docs]def credit_site(request, fleetID):
"""
Credit a site to the given fleet. Takes POST input:
type = short_name of site type
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
site_type = get_object_or_404(SiteType, shortname=request.POST.get('type', None))
fleet.credit_site(site_type, fleet.system, request.user)
return HttpResponse()
@require_boss()
[docs]def remove_site(request, fleetID, siteID):
"""
Uncredit a site to the given fleet.
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
site = get_object_or_404(SiteRecord, pk=siteID)
site.delete()
return HttpResponse()
@require_boss()
[docs]def approve_fleet_site(request, fleetID, siteID, memberID):
"""
Approve a pending site while the fleet is active.
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
site = get_object_or_404(SiteRecord, pk=siteID)
member = get_object_or_404(User, pk=memberID)
if fleet.ended:
raise PermissionDenied
site.members.get(user=member).approve()
return HttpResponse()
[docs]def claim_site(request, fleetID, siteID, memberID):
"""
Claims the site (making it pending) if called by member.
Fully grants credit for the site if called by current boss.
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
site = get_object_or_404(SiteRecord, pk=siteID)
member = get_object_or_404(User, pk=memberID)
if fleet.ended:
raise PermissionDenied
if member in site:
if fleet.current_boss == request.user:
site.members.get(user=member).approve()
return HttpResponse()
else:
raise PermissionDenied
else:
if fleet.current_boss == request.user:
site.members.add(UserSite(site=site, user=member, pending=False))
return HttpResponse()
elif member == request.user:
site.members.add(UserSite(site=site, user=member, pending=True))
return HttpResponse()
raise PermissionDenied
[docs]def unclaim_site(request, fleetID, siteID, memberID):
"""
Unclaims a site during a running fleet.
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
site = get_object_or_404(SiteRecord, pk=siteID)
member = get_object_or_404(User, pk=memberID)
if fleet.ended:
raise PermissionDenied
if fleet.current_boss == request.user or member == request.user:
site.members.filter(user=member).delete()
return HttpResponse()
raise PermissionDenied
@permission_required('SiteTracker.can_sitetracker')
[docs]def refresh_fleets(request):
"""
Return a template with tags for the myfleets and availfleets lists.
"""
if not request.is_ajax():
raise PermissionDenied
myfleets = request.user.sitetrackerlogs.filter(leavetime=None)
return TemplateResponse(request, "st_fleet_refresh.html", {'myfleets': myfleets})
@require_boss()
[docs]def boss_panel(request, fleetID):
"""
Return the HTML for the boss control panel popup.
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
return TemplateResponse(request, "st_boss_panel.html", {'fleet': fleet})
@require_boss()
[docs]def refresh_boss_member(request, fleetID, memberID):
"""
Returns an updated details view for a boss panel member.
"""
if not request.is_ajax():
raise PermissionDenied
fleet = get_object_or_404(Fleet, pk=fleetID)
member = get_object_or_404(User, pk=memberID)
return TemplateResponse(request, "st_boss_member_refresh.html",
{'member': fleet.members.filter(user=member).latest('jointime')})