# Generated by Django 2.0 on 2018-04-01 15:11

from django.db import migrations
from django.utils import timezone

import pytz

def localize_event_datetimes(apps, schema_editor):
    Event = MyModel = apps.get_model('events', 'Event')
    for event in Event.objects.all():
        utc_tz = pytz.timezone('UTC')
        event_tz = get_event_timezone(event)
        print("Converting event %s to %s" % (event.id, event_tz))
        event.start_time = utc_tz.localize(timezone.make_naive(event_tz.localize(timezone.make_naive(event.start_time, pytz.utc))))
        event.end_time = utc_tz.localize(timezone.make_naive(event_tz.localize(timezone.make_naive(event.end_time, pytz.utc))))
        event.save()

def localize_commonevent_datetimes(apps, schema_editor):
    CommonEvent = MyModel = apps.get_model('events', 'CommonEvent')
    for event in CommonEvent.objects.all():
        utc_tz = pytz.timezone('UTC')
        event_tz = get_event_timezone(event)
        print("Converting common event %s to %s" % (event.id, event_tz))
        event.start_time = utc_tz.localize(timezone.make_naive(event_tz.localize(timezone.make_naive(event.start_time, pytz.utc))))
        event.end_time = utc_tz.localize(timezone.make_naive(event_tz.localize(timezone.make_naive(event.end_time, pytz.utc))))
        event.save()

def get_event_timezone(event):
    if event.place is not None:
        return pytz.timezone(event.place.tz)
    elif hasattr(event, 'team') and event.team is not None:
        return pytz.timezone(event.team.tz)
    elif hasattr(event, 'city') and event.city is not None:
        return pytz.timezone(event.city.tz)
    else:
        return pytz.timezone("UTC")

class Migration(migrations.Migration):

    dependencies = [
        ('events', '0021_add_account_secret'),
    ]

    operations = [
        migrations.RunPython(localize_event_datetimes, reverse_code=migrations.RunPython.noop),
        migrations.RunPython(localize_commonevent_datetimes, reverse_code=migrations.RunPython.noop),
    ]