mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2025-07-11 05:20:16 +02:00
4ea7fdeaf2
Now allows users with corp_apis permission to see some APIs when searching, if result is in corp. Correct type mismatch when determining if user with corp_apis can see member list. Correctly pull EveWho memberlist in corp mode when API missing from settings.py Closes #552
302 lines
14 KiB
Python
302 lines
14 KiB
Python
from __future__ import unicode_literals
|
|
from django.conf import settings
|
|
from django.contrib.auth.decorators import login_required
|
|
from django.shortcuts import render, redirect
|
|
|
|
from collections import namedtuple
|
|
|
|
from authentication.models import AuthServicesInfo
|
|
from services.managers.eve_api_manager import EveApiManager
|
|
from services.managers.evewho_manager import EveWhoManager
|
|
from eveonline.models import EveCorporationInfo
|
|
from eveonline.models import EveAllianceInfo
|
|
from eveonline.models import EveCharacter
|
|
from eveonline.models import EveApiKeyPair
|
|
from fleetactivitytracking.models import Fat
|
|
from corputils.forms import CorputilsSearchForm
|
|
from evelink.api import APIError
|
|
|
|
import logging
|
|
import datetime
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Player(object):
|
|
def __init__(self, main, user, maincorp, maincorpid, altlist, apilist, n_fats):
|
|
self.main = main
|
|
self.user = user
|
|
self.maincorp = maincorp
|
|
self.maincorpid = maincorpid
|
|
self.altlist = altlist
|
|
self.apilist = apilist
|
|
self.n_fats = n_fats
|
|
|
|
|
|
def first_day_of_next_month(year, month):
|
|
if month == 12:
|
|
return datetime.datetime(year + 1, 1, 1)
|
|
else:
|
|
return datetime.datetime(year, month + 1, 1)
|
|
|
|
|
|
def first_day_of_previous_month(year, month):
|
|
if month == 1:
|
|
return datetime.datetime(year - 1, 12, 1)
|
|
else:
|
|
return datetime.datetime(year, month - 1, 1)
|
|
|
|
|
|
@login_required
|
|
def corp_member_view(request, corpid=None, year=datetime.date.today().year, month=datetime.date.today().month):
|
|
year = int(year)
|
|
month = int(month)
|
|
start_of_month = datetime.datetime(year, month, 1)
|
|
start_of_next_month = first_day_of_next_month(year, month)
|
|
start_of_previous_month = first_day_of_previous_month(year, month)
|
|
logger.debug("corp_member_view called by user %s" % request.user)
|
|
|
|
try:
|
|
user_main = EveCharacter.objects.get(
|
|
character_id=AuthServicesInfo.objects.get_or_create(user=request.user)[0].main_char_id)
|
|
user_corp_id = user_main.corporation_id
|
|
except (ValueError, EveCharacter.DoesNotExist):
|
|
user_corp_id = settings.CORP_ID
|
|
|
|
if not settings.IS_CORP:
|
|
alliance = EveAllianceInfo.objects.get(alliance_id=settings.ALLIANCE_ID)
|
|
alliancecorps = EveCorporationInfo.objects.filter(alliance=alliance)
|
|
membercorplist = [(int(membercorp.corporation_id), str(membercorp.corporation_name)) for membercorp in
|
|
alliancecorps]
|
|
membercorplist.sort(key=lambda tup: tup[1])
|
|
membercorp_id_list = [int(membercorp.corporation_id) for membercorp in alliancecorps]
|
|
|
|
bluecorps = EveCorporationInfo.objects.filter(is_blue=True)
|
|
bluecorplist = [(int(bluecorp.corporation_id), str(bluecorp.corporation_name)) for bluecorp in bluecorps]
|
|
bluecorplist.sort(key=lambda tup: tup[1])
|
|
bluecorp_id_list = [int(bluecorp.corporation_id) for bluecorp in bluecorps]
|
|
|
|
if not (user_corp_id in membercorp_id_list or user_corp_id not in bluecorp_id_list):
|
|
user_corp_id = None
|
|
|
|
if not corpid:
|
|
if settings.IS_CORP:
|
|
corpid = settings.CORP_ID
|
|
elif user_corp_id:
|
|
corpid = user_corp_id
|
|
else:
|
|
corpid = membercorplist[0][0]
|
|
|
|
corp = EveCorporationInfo.objects.get(corporation_id=corpid)
|
|
if request.user.has_perm('auth.alliance_apis') or (request.user.has_perm('auth.corp_apis') and user_corp_id == corpid):
|
|
logger.debug("Retreiving and sending API-information")
|
|
|
|
if settings.IS_CORP:
|
|
try:
|
|
member_list = EveApiManager.get_corp_membertracking(settings.CORP_API_ID, settings.CORP_API_VCODE)
|
|
except APIError:
|
|
logger.debug("Corp API does not have membertracking scope, using EveWho data instead.")
|
|
member_list = EveWhoManager.get_corporation_members(corpid)
|
|
else:
|
|
member_list = EveWhoManager.get_corporation_members(corpid)
|
|
|
|
characters_with_api = {}
|
|
characters_without_api = {}
|
|
|
|
num_registered_characters = 0
|
|
for char_id, member_data in member_list.items():
|
|
try:
|
|
char = EveCharacter.objects.get(character_id=char_id)
|
|
char_owner = char.user
|
|
try:
|
|
mainid = int(AuthServicesInfo.objects.get_or_create(user=char_owner)[0].main_char_id)
|
|
mainchar = EveCharacter.objects.get(character_id=mainid)
|
|
mainname = mainchar.character_name
|
|
maincorp = mainchar.corporation_name
|
|
maincorpid = mainchar.corporation_id
|
|
api_pair = EveApiKeyPair.objects.get(api_id=char.api_id)
|
|
except (ValueError, EveCharacter.DoesNotExist, EveApiKeyPair.DoesNotExist):
|
|
logger.info("No main character seem to be set for character %s" % char.character_name)
|
|
mainname = "User: " + char_owner.username
|
|
mainchar = char
|
|
maincorp = "Not set."
|
|
maincorpid = None
|
|
api_pair = None
|
|
num_registered_characters += 1
|
|
characters_with_api.setdefault(mainname, Player(main=mainchar,
|
|
user=char_owner,
|
|
maincorp=maincorp,
|
|
maincorpid=maincorpid,
|
|
altlist=[],
|
|
apilist=[],
|
|
n_fats=0)
|
|
).altlist.append(char)
|
|
if api_pair:
|
|
characters_with_api[mainname].apilist.append(api_pair)
|
|
|
|
except EveCharacter.DoesNotExist:
|
|
characters_without_api.update({member_data["name"]: member_data["id"]})
|
|
|
|
for char in EveCharacter.objects.filter(corporation_id=corpid):
|
|
if not int(char.character_id) in member_list:
|
|
logger.info("Character '%s' does not exist in EveWho dump." % char.character_name)
|
|
char_owner = char.user
|
|
try:
|
|
mainid = int(AuthServicesInfo.objects.get_or_create(user=char_owner)[0].main_char_id)
|
|
mainchar = EveCharacter.objects.get(character_id=mainid)
|
|
mainname = mainchar.character_name
|
|
maincorp = mainchar.corporation_name
|
|
maincorpid = mainchar.corporation_id
|
|
api_pair = EveApiKeyPair.objects.get(api_id=char.api_id)
|
|
except (ValueError, EveCharacter.DoesNotExist, EveApiKeyPair.DoesNotExist):
|
|
logger.info("No main character seem to be set for character %s" % char.character_name)
|
|
mainname = "User: " + char_owner.username
|
|
mainchar = char
|
|
maincorp = "Not set."
|
|
maincorpid = None
|
|
api_pair = None
|
|
num_registered_characters += 1
|
|
characters_with_api.setdefault(mainname, Player(main=mainchar,
|
|
user=char_owner,
|
|
maincorp=maincorp,
|
|
maincorpid=maincorpid,
|
|
altlist=[],
|
|
apilist=[],
|
|
n_fats=0)
|
|
).altlist.append(char)
|
|
if api_pair:
|
|
characters_with_api[mainname].apilist.append(api_pair)
|
|
|
|
n_unacounted = corp.member_count - (num_registered_characters + len(characters_without_api))
|
|
|
|
for mainname, player in characters_with_api.items():
|
|
fats_this_month = Fat.objects.filter(user=player.user).filter(
|
|
fatlink__fatdatetime__gte=start_of_month).filter(fatlink__fatdatetime__lt=start_of_next_month)
|
|
characters_with_api[mainname].n_fats = len(fats_this_month)
|
|
|
|
if start_of_next_month > datetime.datetime.now():
|
|
start_of_next_month = None
|
|
|
|
if not settings.IS_CORP:
|
|
context = {"membercorplist": membercorplist,
|
|
"corp": corp,
|
|
"characters_with_api": sorted(characters_with_api.items()),
|
|
'n_registered': num_registered_characters,
|
|
'n_unacounted': n_unacounted,
|
|
"characters_without_api": sorted(characters_without_api.items()),
|
|
"search_form": CorputilsSearchForm()}
|
|
else:
|
|
logger.debug("corp_member_view running in corportation mode")
|
|
context = {"corp": corp,
|
|
"characters_with_api": sorted(characters_with_api.items()),
|
|
'n_registered': num_registered_characters,
|
|
'n_unacounted': n_unacounted,
|
|
"characters_without_api": sorted(characters_without_api.items()),
|
|
"search_form": CorputilsSearchForm()}
|
|
|
|
context["next_month"] = start_of_next_month
|
|
context["previous_month"] = start_of_previous_month
|
|
context["this_month"] = start_of_month
|
|
|
|
return render(request, 'registered/corputils.html', context=context)
|
|
else:
|
|
logger.warn('User %s (%s) not authorized to view corp stats for corp id %s' % (request.user, user_corp_id, corpid))
|
|
return redirect("auth_dashboard")
|
|
|
|
|
|
def can_see_api(user, character):
|
|
if user.has_perm('auth.alliance_apis'):
|
|
return True
|
|
try:
|
|
user_main = EveCharacter.objects.get(
|
|
character_id=AuthServicesInfo.objects.get_or_create(user=user)[0].main_char_id)
|
|
if user.has_perm('auth.corp_apis') and user_main.corporation_id == character.corporation_id:
|
|
return True
|
|
except EveCharacter.DoesNotExist:
|
|
return False
|
|
return False
|
|
|
|
|
|
@login_required
|
|
def corputils_search(request, corpid=settings.CORP_ID):
|
|
logger.debug("corputils_search called by user %s" % request.user)
|
|
|
|
corp = EveCorporationInfo.objects.get(corporation_id=corpid)
|
|
|
|
authorized = False
|
|
try:
|
|
user_main = EveCharacter.objects.get(
|
|
character_id=AuthServicesInfo.objects.get_or_create(user=request.user)[0].main_char_id)
|
|
if request.user.has_perm('auth.alliance_apis') or (
|
|
request.user.has_perm('auth.corp_apis') and (user_main.corporation_id == corpid)):
|
|
logger.debug("Retreiving and sending API-information")
|
|
authorized = True
|
|
except (ValueError, EveCharacter.DoesNotExist):
|
|
if request.user.has_perm('auth.alliance_apis'):
|
|
logger.debug("Retrieving and sending API-information")
|
|
authorized = True
|
|
|
|
if authorized:
|
|
if request.method == 'POST':
|
|
form = CorputilsSearchForm(request.POST)
|
|
logger.debug("Request type POST contains form valid: %s" % form.is_valid())
|
|
if form.is_valid():
|
|
# Really dumb search and only checks character name
|
|
# This can be improved but it does the job for now
|
|
searchstring = form.cleaned_data['search_string']
|
|
logger.debug("Searching for player with character name %s for user %s" % (searchstring, request.user))
|
|
|
|
member_list = {}
|
|
if settings.IS_CORP:
|
|
member_list = EveApiManager.get_corp_membertracking(settings.CORP_API_ID, settings.CORP_API_VCODE)
|
|
if not member_list:
|
|
logger.debug('Unable to fetch members from API. Pulling from EveWho')
|
|
member_list = EveWhoManager.get_corporation_members(corpid)
|
|
|
|
SearchResult = namedtuple('SearchResult',
|
|
['name', 'id', 'main', 'api_registered', 'character', 'apiinfo'])
|
|
|
|
searchresults = []
|
|
for memberid, member_data in member_list.items():
|
|
if searchstring.lower() in member_data["name"].lower():
|
|
try:
|
|
char = EveCharacter.objects.get(character_name=member_data["name"])
|
|
user = char.user
|
|
mainid = int(AuthServicesInfo.objects.get_or_create(user=user)[0].main_char_id)
|
|
main = EveCharacter.objects.get(character_id=mainid)
|
|
if can_see_api(request.user, char):
|
|
api_registered = True
|
|
apiinfo = EveApiKeyPair.objects.get(api_id=char.api_id)
|
|
else:
|
|
api_registered = False
|
|
apiinfo = None
|
|
except EveCharacter.DoesNotExist:
|
|
api_registered = False
|
|
char = None
|
|
main = ""
|
|
apiinfo = None
|
|
|
|
searchresults.append(SearchResult(name=member_data["name"], id=memberid, main=main,
|
|
api_registered=api_registered,
|
|
character=char, apiinfo=apiinfo))
|
|
|
|
logger.info("Found %s members for user %s matching search string %s" % (
|
|
len(searchresults), request.user, searchstring))
|
|
|
|
context = {'corp': corp, 'results': searchresults, 'search_form': CorputilsSearchForm(),
|
|
"year": datetime.datetime.now().year, "month": datetime.datetime.now().month}
|
|
|
|
return render(request, 'registered/corputilssearchview.html',
|
|
context=context)
|
|
else:
|
|
logger.debug("Form invalid - returning for user %s to retry." % request.user)
|
|
context = {'corp': corp, 'members': None, 'search_form': CorputilsSearchForm()}
|
|
return render(request, 'registered/corputilssearchview.html', context=context)
|
|
|
|
else:
|
|
logger.debug("Returning empty search form for user %s" % request.user)
|
|
return redirect("auth_corputils")
|
|
else:
|
|
logger.warn('User %s not authorized to view corp stats for corp ID %s' % (request.user, corpid))
|
|
return redirect("auth_dashboard")
|