it/itf/app/models.py
2011-03-06 18:05:55 +05:30

173 lines
4.5 KiB
Python

from django.db import models
import operator
from django.db.models import Q
from ox.text import smartSplit
from ox.django.fields import DictField
from django.core.paginator import Paginator, InvalidPage, EmptyPage
from django.contrib.contenttypes.models import ContentType
def splitSearch(string):
ret = []
for s in smartSplit(string):
word = s.replace("'", "").replace('"', '')
ret.append(word)
return ret
class ItfModel(models.Model):
fts_fields = []
fk_filters = []
sort_fields = []
hasComments = True
class Meta:
abstract = True
def list_dict(self):
return {}
def info_dict(self):
return {}
@classmethod
def fts(kls, qset, search):
terms = splitSearch(search)
qobjects = []
for t in terms:
for f in kls.fts_fields:
qstring = f + '__icontains'
qobject = Q(**{qstring:t})
qobjects.append(qobject)
return qset.filter(reduce(operator.or_, qobjects))
'''
eg. fks = {
'somefield': [1, 5, 7],
'someotherfield': [3]
}
'''
@classmethod
def filter_by_fks(kls, qset, fks):
qobjects = []
for key in fks.keys():
field = getField(kls._meta.fields, key)
if field:
# rel_class = field.related.parent_model
for i in fks[key]:
qobject = Q(**{field.name: i})
qobjects.append(qobject)
return qset.filter(reduce(operator.or_, qobjects))
@classmethod
def get_fk_objects(kls):
ret = {}
for f in kls.fk_filters:
ret[f] = []
field = getField(kls._meta.fields, f)
rel_class = field.related.parent_model
for o in rel_class.objects.all():
ret[f].append({
'id': o.id,
'title': unicode(o)
})
return ret
@classmethod
def get_list(kls, data):
options = {
'page_no': 1,
'list_size': 8,
'search': '',
'sort': [],
'range': [0,50]
}
options.update(data)
ret = []
page_no = options['page_no']
list_size = options['list_size']
qset = kls.objects.all()
search = options['search']
if search != '':
qset = kls.fts(qset, search)
sort = options['sort']
if sort != []:
for s in sort:
if s['operator'] == '-':
operator = '-'
else:
operator = ''
sort = operator + s['key']
qset = qset.order_by(sort)
r0 = options['range'][0]
r1 = options['range'][1]
results = qset[r0:r1]
for r in results:
ret.append(r.list_dict())
return ret
def getField(fields, name):
for f in fields:
if f.name == name:
return f
return False
class ModelsModel(models.Model):
model = models.ForeignKey(ContentType)
friendly_name = models.CharField(max_length=256)
friendly_name_singular = models.CharField(max_length=256, blank=True, null=True)
info = models.TextField(blank=True)
sort_options = models.ManyToManyField("ModelSort", blank=True, null=True)
# filter_fields = DictField(blank=True)
is_visible = models.BooleanField(default=True)
@property
def module(self):
return self.model.app_label
def __unicode__(self):
return self.friendly_name
def get_dict(self):
# sort_options = map(lambda x: {'key': x.key, 'text': x.text}, self.sort_options.all())
return {
'module': self.module,
'model': self.model.model,
'info': self.info,
'friendly_name': self.friendly_name,
'friendly_name_singular': self.friendly_name_singular,
'fk_filters': self.model.model_class().fk_filters,
'fts_fields': self.model.model_class().fts_fields,
'sort_options': map(lambda x: {'key': x.key, 'text': x.text}, self.sort_options.all())
}
class ModelSort(models.Model):
key = models.CharField(max_length=64)
text = models.CharField(max_length=512)
def __unicode__(self):
return "%s: %s" % (self.key, self.text,)
class ModelExtraButton(models.Model):
icon = models.CharField(max_length=64)
mouseover = models.CharField(max_length=512, blank=True, null=True)
dialog_text = models.TextField()
model = models.ForeignKey(ModelsModel)
def __unicode__(self):
return self.mouseover
def site_config():
with open(settings.SITE_CONFIG) as f:
site_config = json.load(f)
site_config['keys'] = {}
for key in site_config['itemKeys']:
site_config['keys'][key['id']] = key
site_config['_findKeys'] = {}
for key in site_config['findKeys']:
site_config['_findKeys'][key['id']] = key
return site_config