+from django.conf import settings
+from django.template import Template, Context
+
+from django.db import models
+from django.contrib.contenttypes.models import ContentType
+
+import os.path
+
+import djangosphinx.apis.current as sphinxapi
+
+__all__ = ('generate_config_for_model', 'generate_config_for_models')
+
+def _get_database_engine():
+ if settings.DATABASE_ENGINE == 'mysql':
+ return settings.DATABASE_ENGINE
+ elif settings.DATABASE_ENGINE.startswith('postgresql'):
+ return 'pgsql'
+ raise ValueError, "Only MySQL and PostgreSQL engines are supported by Sphinx."
+
+def _get_template(name):
+ paths = (
+ os.path.join(os.path.dirname(__file__), '../apis/api%s/templates/' % (sphinxapi.VER_COMMAND_SEARCH,)),
+ os.path.join(os.path.dirname(__file__), '../templates/'),
+ )
+ for path in paths:
+ try:
+ fp = open(path + name, 'r')
+ except IOError:
+ continue
+ try:
+ t = Template(fp.read())
+ return t
+ finally:
+ fp.close()
+ raise ValueError, "Template matching name does not exist: %s." % (name,)
+
+def _is_sourcable_field(field):
+ # We can use float fields in 0.98
+ if sphinxapi.VER_COMMAND_SEARCH >= 0x113 and (isinstance(field, models.FloatField) or isinstance(field, models.DecimalField)):
+ return True
+ if isinstance(field, models.ForeignKey):
+ return True
+ if isinstance(field, models.IntegerField) and field.choices:
+ return True
+ if not field.rel:
+ return True
+ return False
+
+# No trailing slashes on paths
+DEFAULT_SPHINX_PARAMS = {
+ 'database_engine': _get_database_engine(),
+ 'database_host': settings.DATABASE_HOST,
+ 'database_port': settings.DATABASE_PORT,
+ 'database_name': settings.DATABASE_NAME,
+ 'database_user': settings.DATABASE_USER,
+ 'database_password': settings.DATABASE_PASSWORD,
+ 'log_file': '/var/log/sphinx/searchd.log',
+ 'data_path': '/var/data',
+}
+
+# Generate for single models
+
+def generate_config_for_model(model_class, index=None, sphinx_params={}):
+ """
+ Generates a sample configuration including an index and source for
+ the given model which includes all attributes and date fields.
+ """
+ return generate_source_for_model(model_class, index, sphinx_params) + "\n\n" + generate_index_for_model(model_class, index, sphinx_params)
+
+def generate_index_for_model(model_class, index=None, sphinx_params={}):
+ """Generates a source configmration for a model."""
+ t = _get_template('index.conf')
+
+ if index is None:
+ index = model_class._meta.db_table
+
+ params = DEFAULT_SPHINX_PARAMS
+ params.update(sphinx_params)
+ params.update({
+ 'index_name': index,
+ 'source_name': index,
+ })
+
+ c = Context(params)
+
+ return t.render(c)
+
+
+def generate_source_for_model(model_class, index=None, sphinx_params={}):
+ """Generates a source configmration for a model."""
+ t = _get_template('source.conf')
+
+ valid_fields = [f for f in model_class._meta.fields if _is_sourcable_field(f)]
+
+ # Hackish solution for a bug I've introduced into composite pks branch
+ pk = model_class._meta.get_field(model_class._meta.pk.name)
+
+ if pk not in valid_fields:
+ valid_fields.insert(0, model_class._meta.pk)
+
+ if index is None:
+ index = model_class._meta.db_table
+
+ params = DEFAULT_SPHINX_PARAMS
+ params.update(sphinx_params)
+ params.update({
+ 'source_name': index,
+ 'index_name': index,
+ 'table_name': index,
+ 'primary_key': pk.column,
+ 'field_names': [f.column for f in valid_fields],
+ 'group_columns': [f.column for f in valid_fields if (f.rel or isinstance(f, models.BooleanField) or isinstance(f, models.IntegerField)) and not f.primary_key],
+ 'date_columns': [f.column for f in valid_fields if isinstance(f, models.DateTimeField) or isinstance(f, models.DateField)],
+ 'float_columns': [f.column for f in valid_fields if isinstance(f, models.FloatField) or isinstance(f, models.DecimalField)],
+ })
+
+ c = Context(params)
+
+ return t.render(c)
+
+# Generate for multiple models (search UNIONs)
+
+def generate_config_for_models(model_classes, index=None, sphinx_params={}):
+ """
+ Generates a sample configuration including an index and source for
+ the given model which includes all attributes and date fields.
+ """
+ return generate_source_for_models(model_classes, index, sphinx_params) + "\n\n" + generate_index_for_models(model_classes, index, sphinx_params)
+
+def generate_index_for_models(model_classes, index=None, sphinx_params={}):
+ """Generates a source configmration for a model."""
+ t = _get_template('index-multiple.conf')
+
+ if index is None:
+ index = '_'.join(m._meta.db_table for m in model_classes)
+
+ params = DEFAULT_SPHINX_PARAMS
+ params.update(sphinx_params)
+ params.update({
+ 'index_name': index,
+ 'source_name': index,
+ })
+
+ c = Context(params)
+
+ return t.render(c)
+
+def generate_source_for_models(model_classes, index=None, sphinx_params={}):
+ """Generates a source configmration for a model."""
+ t = _get_template('source-multiple.conf')
+
+ # We need to loop through each model and find only the fields that exist *exactly* the
+ # same across models.
+ def _the_tuple(f):
+ return (f.__class__, f.column, getattr(f.rel, 'to', None), f.choices)
+
+ valid_fields = [_the_tuple(f) for f in model_classes[0]._meta.fields if _is_sourcable_field(f)]
+ for model_class in model_classes[1:]:
+ valid_fields = [_the_tuple(f) for f in model_class._meta.fields if _the_tuple(f) in valid_fields]
+
+ tables = []
+ for model_class in model_classes:
+ tables.append((model_class._meta.db_table, ContentType.objects.get_for_model(model_class)))
+
+ if index is None:
+ index = '_'.join(m._meta.db_table for m in model_classes)
+
+ params = DEFAULT_SPHINX_PARAMS
+ params.update(sphinx_params)
+ params.update({
+ 'tables': tables,
+ 'source_name': index,
+ 'index_name': index,
+ 'field_names': [f[1] for f in valid_fields],
+ 'group_columns': [f[1] for f in valid_fields if f[2] or isinstance(f[0], models.BooleanField) or isinstance(f[0], models.IntegerField)],
+ 'date_columns': [f[1] for f in valid_fields if issubclass(f[0], models.DateTimeField) or issubclass(f[0], models.DateField)],
+ 'float_columns': [f[1] for f in valid_fields if isinstance(f[0], models.FloatField) or isinstance(f[0], models.DecimalField)],
+ })
+
+ c = Context(params)
+
+ return t.render(c)
\ No newline at end of file