Germano Gabbianelli avatar Germano Gabbianelli committed 96b8d41

Added autocompleteform_factory.

Comments (0)

Files changed (1)

autocomplete/utils.py

 from django import forms
+from django.forms.models import modelform_factory
 from django.db import models
 
 from autocomplete import widgets
     kwargs.pop('request', None) # request can be passed by contrib.admin
     db = kwargs.get('using')
     settings = view.get_settings(ac_id)
+    if widget_class is None:
+        widget_class = widgets.AutocompleteWidget
+    if multiple_widget_class is None:
+        multiple_widget_class = widgets.MultipleAutocompleteWidget
 
     if formfield is None:
         formfield = getattr(settings.field, 'formfield', forms.CharField)
     return formfield(**kwargs)
 
 
-def autocompleteform_factory():
-    """Implement me!"""
+def _formfield_callback(autocomplete_fields, **ac_options):
+    view = ac_options.get('view', default_view)
+
+    def autocomplete_callback(field, **kwargs):
+        kwargs = dict(ac_options, **kwargs)
+        if field.name in autocomplete_fields:
+            ac_id = autocomplete_fields[field.name]
+            return autocomplete_formfield(ac_id, **kwargs)
+        elif view.has_settings(field):
+            return autocomplete_formfield(field, **kwargs)
+        return field.formfield(**kwargs)
+    return autocomplete_callback
+
+
+def autocompleteform_factory(model, autocomplete_fields={},
+        form=forms.ModelForm, fields=None, exclude=None, **ac_options):
+    """
+    autocompleteform_factory(MyModel, {'friends':'myapp.friends'})
+    """
+    # XXX maybe we should make formfield_callback customizable
+    callback = _formfield_callback(autocomplete_fields, **ac_options)
+    return modelform_factory(model, form, fields, exclude, callback)
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.