# -*- coding: utf-8 -*-
# Generated by Django 1.10.1 on 2016-09-12 13:04
from __future__ import unicode_literals

from django.db import migrations

from authentication.states import MEMBER_STATE, BLUE_STATE, NONE_STATE
from django.conf import settings

def determine_membership_by_character(char, apps):
    if settings.IS_CORP:
        if int(char.corporation_id) == int(settings.CORP_ID):
            return MEMBER_STATE
    else:
        if int(char.alliance_id) == int(settings.ALLIANCE_ID):
            return MEMBER_STATE
    EveCorporationInfo = apps.get_model('eveonline', 'EveCorporationInfo')
    if EveCorporationInfo.objects.filter(corporation_id=char.corporation_id).exists() is False:
         return NONE_STATE
    else:
         corp = EveCorporationInfo.objects.get(corporation_id=char.corporation_id)
         if corp.is_blue:
             return BLUE_STATE
         else:
             return NONE_STATE

def determine_membership_by_user(user, apps):
    AuthServicesInfo = apps.get_model('authentication', 'AuthServicesInfo')
    auth, c = AuthServicesInfo.objects.get_or_create(user=user)
    if auth.main_char_id:
        EveCharacter = apps.get_model('eveonline', 'EveCharacter')
        if EveCharacter.objects.filter(character_id=auth.main_char_id).exists():
            char = EveCharacter.objects.get(character_id=auth.main_char_id)
            return determine_membership_by_character(char, apps)
        else:
            return NONE_STATE
    else:
        return NONE_STATE

def set_state(user, apps):
    if user.is_active:
        state = determine_membership_by_user(user, apps)
    else:
        state = NONE_STATE
    AuthServicesInfo = apps.get_model('authentication', 'AuthServicesInfo')
    auth = AuthServicesInfo.objects.get_or_create(user=user)[0]
    if auth.state != state:
       auth.state = state
       auth.save()

def set_initial_state(apps, schema_editor):
    User = apps.get_model('auth', 'User')
    for u in User.objects.all():
        set_state(u, apps)

class Migration(migrations.Migration):

    dependencies = [
        ('authentication', '0007_remove_authservicesinfo_is_blue'),
        ('eveonline', '0001_initial'),
        ('auth', '0001_initial'),
    ]

    operations = [
        migrations.RunPython(set_initial_state, migrations.RunPython.noop)
    ]