# Generated by Django 2.0.4 on 2018-05-08 07:42

from django.core.exceptions import ObjectDoesNotExist
from django.db import migrations


def add_teams_to_categories(apps, schema_editor):
    InfoCategory = apps.get_model("info", "InfoCategory")
    Team = apps.get_model("teams", "Team")
    Camp = apps.get_model("camps", "Camp")

    try:
        # 2016 - Everything is orga team
        camp2016 = Camp.objects.get(slug="bornhack-2016")
        orga2016 = Team.objects.get(camp=camp2016, name="Orga")
        InfoCategory.objects.filter(camp=camp2016).update(team=orga2016)

        # 2017 - Everything is orga team
        camp2017 = Camp.objects.get(slug="bornhack-2017")
        orga2017 = Team.objects.get(camp=camp2017, name="Orga")
        InfoCategory.objects.filter(camp=camp2017).update(team=orga2017)

        # 2018 - Map categories to teams
        camp2018 = Camp.objects.get(slug="bornhack-2018")
        team2018 = Team.objects.filter(camp=camp2018)
        infocategories2018 = InfoCategory.objects.filter(camp=camp2018)

        # Info team
        infoteam = team2018.get(name="Info")
        info_anchors = [
            "what",
            "when",
            "travel",
            "where",
            "sleep",
            "bicycles",
            "infodesk-and-cert",
            "shower-and-toilets",
            "venue-map",
            "villages",
        ]
        infocategories2018.filter(anchor__in=info_anchors).update(team=infoteam)

        # Food team
        food = team2018.get(name="Food")
        infocategories2018.filter(anchor__in=["food-and-groceries"]).update(team=food)

        # NOC team
        noc = team2018.get(name="NOC")
        infocategories2018.filter(anchor__in=["network"]).update(team=noc)

        # Power team
        power = team2018.get(name="Power")
        infocategories2018.filter(anchor__in=["power"]).update(team=power)

        # Shuttle bus
        shuttle_bus = team2018.get(name="Shuttle Bus")
        infocategories2018.filter(anchor__in=["shuttle-bus"]).update(team=shuttle_bus)

        # Bar
        bar = team2018.get(name="Bar")
        infocategories2018.filter(anchor__in=["bar"]).update(team=bar)

        # Make info team catch all remaining
        infocategories2018.filter(team__isnull=True).update(team=infoteam)
    except ObjectDoesNotExist:
        pass


class Migration(migrations.Migration):

    dependencies = [("info", "0004_infocategory_team")]

    operations = [migrations.RunPython(add_teams_to_categories)]