#!/usr/bin/env python
# -*- coding: utf-8 -*-
# -*- Python -*-
#

import os, sys
basedir = os.path.abspath(os.path.join(os.path.dirname(__file__), "../.."))
sys.path = [ basedir ] + sys.path
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "ietf.settings")

import django
django.setup()

import argparse
import pprint
from django.contrib import admin
from django.contrib.auth.models import User
from ietf.person.models import Person

parser = argparse.ArgumentParser()
parser.add_argument("source_id",type=int)
parser.add_argument("target_id",type=int)
args = parser.parse_args()

source = Person.objects.get(pk=args.source_id)
target = Person.objects.get(pk=args.target_id)

print "Merging person {}({}) to {}({})".format(source.ascii,source.pk,target.ascii,target.pk)
response = raw_input('Ok to continue y/n? ')
if response.lower() != 'y':
    sys.exit()

# merge emails
for email in source.email_set.all():
    print "Merging email: {}".format(email.address)
    email.person = target
    email.save()

# merge aliases
target_aliases = [ a.name for a in target.alias_set.all() ]
for alias in source.alias_set.all():
    if alias.name in target_aliases:
        alias.delete()
    else:
        print "Merging alias: {}".format(alias.name)
        alias.person = target
        alias.save()

# merge DocEvents
for docevent in source.docevent_set.all():
    docevent.by = target
    docevent.save()
    
# merge SubmissionEvents
for subevent in source.submissionevent_set.all():
    subevent.by = target
    subevent.save()

# merge Messages
for message in source.message_set.all():
    message.by = target
    message.save()

# merge Constraints
for constraint in source.constraint_set.all():
    constraint.person = target
    constraint.save()

# merge Roles
for role in source.role_set.all():
    role.person = target
    role.save()

# check for any remaining relationships and delete if none
objs = [source]
opts = Person._meta
user = User.objects.filter(is_superuser=True).first()
admin_site = admin.site
using = 'default'

deletable_objects, perms_needed, protected = admin.utils.get_deleted_objects(
    objs, opts, user, admin_site, using)
    
if len(deletable_objects) > 1:
    print "Not Deleting Person: {}({})".format(source.ascii,source.pk)
    print "Related objects remain:"
    pprint.pprint(deletable_objects[1])

else:
    print "Deleting Person: {}({})".format(source.ascii,source.pk)
    source.delete()