| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980 |
- from flask_admin._compat import as_unicode, string_types
- from flask_admin.model.ajax import AjaxModelLoader, DEFAULT_PAGE_SIZE
- from .tools import get_primary_key
- class QueryAjaxModelLoader(AjaxModelLoader):
- def __init__(self, name, model, **options):
- """
- Constructor.
- :param fields:
- Fields to run query against
- """
- super(QueryAjaxModelLoader, self).__init__(name, options)
- self.model = model
- self.fields = options.get('fields')
- if not self.fields:
- raise ValueError('AJAX loading requires `fields` to be specified for %s.%s' % (model, self.name))
- self._cached_fields = self._process_fields()
- self.pk = get_primary_key(model)
- def _process_fields(self):
- remote_fields = []
- for field in self.fields:
- if isinstance(field, string_types):
- attr = getattr(self.model, field, None)
- if not attr:
- raise ValueError('%s.%s does not exist.' % (self.model, field))
- remote_fields.append(attr)
- else:
- remote_fields.append(field)
- return remote_fields
- def format(self, model):
- if not model:
- return None
- return (getattr(model, self.pk), as_unicode(model))
- def get_one(self, pk):
- return self.model.get(**{self.pk: pk})
- def get_list(self, term, offset=0, limit=DEFAULT_PAGE_SIZE):
- query = self.model.select()
- stmt = None
- for field in self._cached_fields:
- q = field ** (u'%%%s%%' % term)
- if stmt is None:
- stmt = q
- else:
- stmt |= q
- query = query.where(stmt)
- if offset:
- query = query.offset(offset)
- return list(query.limit(limit).execute())
- def create_ajax_loader(model, name, field_name, options):
- prop = getattr(model, field_name, None)
- if prop is None:
- raise ValueError('Model %s does not have field %s.' % (model, field_name))
- # TODO: Check for field
- remote_model = prop.rel_model
- return QueryAjaxModelLoader(name, remote_model, **options)
|
