djangoappengine / db /

Full commit
from .db_settings import get_indexes
from djangotoolbox.db.creation import NonrelDatabaseCreation

class StringType(object):
    def __init__(self, internal_type):
        self.internal_type = internal_type

    def __mod__(self, field):
        # Hack to save URLField and EmailField to the native
        # GAE data types Link and Email.
        # Overriding the data_types dictionary has no effect because the
        # get_internal_type() is not overridden in URLField and EmailField
        # and returns "CharField". Hence the Field.db_type() method returns
        # the mapping of the CharField. 
        field_type = field['model']._meta.get_field(field['name'])
        field_type_class_name = field_type.__class__.__name__
        if field_type_class_name == 'URLField':
            return 'gae_link'
        elif field_type_class_name == 'EmailField':
            return 'gae_email'
        indexes = get_indexes().get(field['model'], {})
        if field['name'] in indexes.get('indexed', ()):
            return 'text'
        elif field['name'] in indexes.get('unindexed', ()):
            return 'longtext'
        return self.internal_type

def get_data_types():
    # TODO: Add GAEKeyField and a corresponding db_type
    string_types = ('text', 'longtext')
    data_types = NonrelDatabaseCreation.data_types.copy()
    for name, field_type in data_types.items():
        if field_type in string_types:
            data_types[name] = StringType(field_type)
    return data_types

class DatabaseCreation(NonrelDatabaseCreation):
    # This dictionary maps Field objects to their associated GAE column
    # types, as strings. Column-type strings can contain format strings; they'll
    # be interpolated against the values of Field.__dict__ before being output.
    # If a column type is set to None, it won't be included in the output.

    data_types = get_data_types()

    def create_test_db(self, *args, **kw):
        """Destroys the test datastore. A new store will be recreated on demand"""
        self.connection.use_test_datastore = True

    def destroy_test_db(self, *args, **kw):
        """Destroys the test datastore files."""
        from .base import destroy_datastore, get_test_datastore_paths