mirror of
https://github.com/ae-utbm/sith.git
synced 2024-11-16 03:03:21 +00:00
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
|
from ajax_select import register, LookupChannel
|
||
|
|
|
||
|
|
from core.views.site import search_user
|
||
|
|
from core.models import User
|
||
|
|
from counter.models import Product, Counter
|
||
|
|
|
||
|
|
@register('users')
|
||
|
|
class UsersLookup(LookupChannel):
|
||
|
|
model = User
|
||
|
|
|
||
|
|
def get_query(self, q, request):
|
||
|
|
return search_user(q)
|
||
|
|
|
||
|
|
def format_match(self, obj):
|
||
|
|
return obj.get_mini_item()
|
||
|
|
|
||
|
|
def format_item_display(self, item):
|
||
|
|
return item.get_display_name()
|
||
|
|
|
||
|
|
@register('counters')
|
||
|
|
class CountersLookup(LookupChannel):
|
||
|
|
model = Counter
|
||
|
|
|
||
|
|
def get_query(self, q, request):
|
||
|
|
return self.model.objects.filter(name__icontains=q)[:50]
|
||
|
|
|
||
|
|
def format_item_display(self, item):
|
||
|
|
return item.name
|
||
|
|
|
||
|
|
@register('products')
|
||
|
|
class ProductsLookup(LookupChannel):
|
||
|
|
model = Product
|
||
|
|
|
||
|
|
def get_query(self, q, request):
|
||
|
|
print(request.__dict__)
|
||
|
|
return (self.model.objects.filter(name__icontains=q) | self.model.objects.filter(code__icontains=q))[:50]
|
||
|
|
|
||
|
|
def format_item_display(self, item):
|
||
|
|
return item.name
|