# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2008 - 2009  Douglas S. Blank <doug.blank@gmail.com>
# Copyright (C) 2009         B. Malengier <benny.malengier@gmail.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#
# $Id$
#

"""
Export to the Django Models on the configured database backend


"""

#------------------------------------------------------------------------
#
# Standard Python Modules
#
#------------------------------------------------------------------------
import sys
import os
import time

#------------------------------------------------------------------------
#
# Set up logging
#
#------------------------------------------------------------------------
import logging
log = logging.getLogger(".ExportDjango")

#------------------------------------------------------------------------
#
# GRAMPS modules
#
#------------------------------------------------------------------------
import ExportOptions
from Utils import create_id, probably_alive
import const
import gen.lib

from TransUtils import get_addon_translator
translator = get_addon_translator(__file__)
_ = translator.gettext
ngettext = translator.ngettext

from django.conf import settings
import web.settings as default_settings
try:
    settings.configure(default_settings)
except RuntimeError:
    # already configured; ignore
    pass

from web.libdjango import DjangoInterface

def export_all(database, filename, option_box=None, callback=None):
    if not callable(callback): 
        callback = lambda (percent): None # dummy

    start = time.time()
    total = (database.get_number_of_notes() + 
             database.get_number_of_people() +
             database.get_number_of_events() + 
             database.get_number_of_families() +
             database.get_number_of_repositories() +
             database.get_number_of_places() +
             database.get_number_of_media_objects() +
             database.get_number_of_sources()) * 2 # steps
    count = 0.0
    dji = DjangoInterface()
    dji.clear_tables("primary", "secondary", "ref")

    for step in [0, 1]:
        print >> sys.stderr, "Exporting Step %d..." % (step + 1)
        # ---------------------------------
        # Person
        # ---------------------------------
        for person_handle in database.person_map.keys():
            data = database.person_map[person_handle]
            if step == 0:
                dji.add_person(data)
            elif step == 1:
                djperson = dji.add_person_detail(data)
                person = database.get_person_from_handle(person_handle)
                djperson.probably_alive = probably_alive(person, database)
                djperson.save()
            count += 1
            callback(100 * count/total)

        # ---------------------------------
        # Notes
        # ---------------------------------
        for note_handle in database.note_map.keys():
            data = database.note_map[note_handle]
            if step == 0:
                dji.add_note(data)
            count += 1
            callback(100 * count/total)

        # ---------------------------------
        # Family
        # ---------------------------------
        for family_handle in database.family_map.keys():
            data = database.family_map[family_handle]
            if step == 0:
                dji.add_family(data)
            elif step == 1:
                dji.add_family_detail(data)
            count += 1
            callback(100 * count/total)

        # ---------------------------------
        # Source
        # ---------------------------------
        for source_handle in database.source_map.keys():
            data = database.source_map[source_handle]
            if step == 0:
                dji.add_source(data)
            elif step == 1:
                dji.add_source_detail(data)
            count += 1
            callback(100 * count/total)

        # ---------------------------------
        # Event
        # ---------------------------------
        for event_handle in database.event_map.keys():
            data = database.event_map[event_handle]
            if step == 0:
                dji.add_event(data)
            elif step == 1:
                dji.add_event_detail(data)
            count += 1
            callback(100 * count/total)

        # ---------------------------------
        # Repository
        # ---------------------------------
        for repository_handle in database.repository_map.keys():
            data = database.repository_map[repository_handle]
            if step == 0:
                dji.add_repository(data)
            elif step == 1:
                dji.add_repository_detail(data)
            count += 1
            callback(100 * count/total)
    
        # ---------------------------------
        # Place 
        # ---------------------------------
        for place_handle in database.place_map.keys():
            data = database.place_map[place_handle]
            if step == 0:
                dji.add_place(data)
            elif step == 1:
                dji.add_place_detail(data)
            count += 1
            callback(100 * count/total)
    
        # ---------------------------------
        # Media
        # ---------------------------------
        for media_handle in database.media_map.keys():
            data = database.media_map[media_handle]
            if step == 0:
                dji.add_media(data)
            elif step == 1:
                dji.add_media_detail(data)
            count += 1
            callback(100 * count/total)

    total_time = time.time() - start
    msg = ngettext('Export Complete: %d second','Export Complete: %d seconds', total_time ) % total_time
    print >> sys.stderr, msg
    return True

class NoFilenameOptions(ExportOptions.WriterOptionBox):
    no_fileselect = True