# -*- coding: utf-8 -*-

'''
# $Id: django_db_mockup.py 375 2007-07-25 10:45:14Z akhavr $
'''

import copy
import django.conf
from django.core.management import syncdb
from django.db import backend, connection
import django.db.backends.sqlite3.base as django_sqlite
from django.db.backends import util
from pysqlite2 import dbapi2 as sqlite
    
def temp_django_db_conn():
    'create temporary django db connection'
    sqlite_conn = django_sqlite.DatabaseWrapper()
    sqlite_conn.connection = sqlite.connect(
        database=':memory:',
        detect_types=sqlite.PARSE_DECLTYPES | sqlite.PARSE_COLNAMES)
    # define some functions that are missing from sqlite
    sqlite_conn.connection.create_function('concat', 2,
                                           lambda *args: ''.join(args))
    return sqlite_conn

def subst_django_db_conn(new_conn):
    '''
    substitute django db connection by <new_conn>
    warning: black magic involved
    '''
    # black magic ;)
    klass = connection.__class__
    orig_conn = klass.__new__(klass)
    orig_conn.__dict__.update(connection.__dict__)

    connection.__class__ = new_conn.__class__
    connection.__dict__.update(new_conn.__dict__)
    return orig_conn

class DbMock:
    def setup(self):
        'db mockup setup'
        self.sqlite_conn = temp_django_db_conn()
        self.orig_conn = subst_django_db_conn(self.sqlite_conn)

        self.orig_db_engine = django.conf.settings.DATABASE_ENGINE
        django.conf.settings.DATABASE_ENGINE = 'sqlite3'
    
        self.orig_db_name = django.conf.settings.DATABASE_NAME
        django.conf.settings.DATABASE_NAME = ':memory:'
    
        self.supports_constraints = backend.supports_constraints
        backend.supports_constraints = False

        self.operator_mapping = copy.deepcopy(backend.OPERATOR_MAPPING)
        backend.OPERATOR_MAPPING.update(django_sqlite.OPERATOR_MAPPING)
    
        syncdb(verbosity=0, interactive=False)
        return

    def teardown(self):
        'db mockup teardown'
        subst_django_db_conn(self.orig_conn)
        django.conf.settings.DATABASE_ENGINE = self.orig_db_engine
        django.conf.settings.DATABASE_NAME = self.orig_db_name
        backend.supports_constraints = self.supports_constraints
        backend.OPERATOR_MAPPING = self.operator_mapping
        return