You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
2.0 KiB
67 lines
2.0 KiB
from pyramid.security import Allow, Authenticated, Deny, Everyone |
|
|
|
from .base import BaseResource, PaginationResourceMixin |
|
from ordr2.models import User, Role |
|
|
|
|
|
|
|
|
|
class UserAccount(BaseResource): |
|
def __acl__(self): |
|
return [ (Allow, 'role:admin', 'view') ] |
|
|
|
|
|
class UserList(BaseResource, PaginationResourceMixin): |
|
sql_model_class = User |
|
child_resource_class = UserAccount |
|
default_sorting = 'user.asc' |
|
default_items_per_page = 12 |
|
|
|
def __acl__(self): |
|
return [ (Allow, 'role:admin', 'view') ] |
|
|
|
|
|
def prepare_filtered_query(self, dbsession, filter_params): |
|
''' setup the base filtered query ''' |
|
query = dbsession.query(self.sql_model_class) |
|
role_name = filter_params.get('role', None) |
|
try: |
|
role_name = role_name.lower() |
|
role = Role(role_name) |
|
query = query.filter_by(role=role) |
|
except (AttributeError, ValueError): |
|
role_name = None |
|
self.filters['role'] = role_name |
|
return query |
|
|
|
|
|
def prepare_sorted_query(self, query, sorting): |
|
''' setup the base filtered query ''' |
|
available_fields = { |
|
'user': 'user_name', |
|
'first': 'first_name', |
|
'last': 'last_name', |
|
'email': 'email', |
|
'role': 'role' |
|
} |
|
name = available_fields.get(sorting.field, None) |
|
model_field = getattr(self.sql_model_class, name, None) |
|
if model_field: |
|
sort_func = sorting.func(model_field) |
|
query = query.order_by(sort_func) |
|
default_sort = self.parse_sort_parameters(self.default_sorting) |
|
if sorting.field != default_sort.field: |
|
default_sort = self.parse_sort_parameters(self.default_sorting) |
|
query = self.prepare_sorted_query(query, default_sort) |
|
return query |
|
|
|
|
|
|
|
class Admin(BaseResource): |
|
|
|
nodes = { |
|
'users': UserList |
|
} |
|
|
|
def __acl__(self): |
|
return [ (Allow, 'role:admin', 'view') ]
|
|
|