Add default normalize logic for API views
and use common logic for getting field list in traditional Form class
This commit is contained in:
parent
960d6279a9
commit
35728e20be
|
@ -28,7 +28,10 @@ from __future__ import unicode_literals, absolute_import
|
|||
|
||||
import json
|
||||
|
||||
import six
|
||||
|
||||
from rattail.config import parse_bool
|
||||
from rattail.db.util import get_fieldnames
|
||||
|
||||
from cornice import resource, Service
|
||||
|
||||
|
@ -268,6 +271,21 @@ class APIMasterView(APIView):
|
|||
query = self.Session.query(cls)
|
||||
return query
|
||||
|
||||
def get_fieldnames(self):
|
||||
if not hasattr(self, '_fieldnames'):
|
||||
self._fieldnames = get_fieldnames(
|
||||
self.rattail_config, self.model_class,
|
||||
columns=True, proxies=True, relations=False)
|
||||
return self._fieldnames
|
||||
|
||||
def normalize(self, obj):
|
||||
data = {'_str': six.text_type(obj)}
|
||||
|
||||
for field in self.get_fieldnames():
|
||||
data[field] = getattr(obj, field)
|
||||
|
||||
return data
|
||||
|
||||
def _collection_get(self):
|
||||
from sqlalchemy_filters import apply_filters, apply_sort, apply_pagination
|
||||
|
||||
|
|
|
@ -49,21 +49,16 @@ class WorkOrderView(APIMasterView):
|
|||
self.workorder_handler = app.get_workorder_handler()
|
||||
|
||||
def normalize(self, workorder):
|
||||
return {
|
||||
'_str': six.text_type(workorder),
|
||||
'uuid': workorder.uuid,
|
||||
'id': workorder.id,
|
||||
'customer_uuid': workorder.customer.uuid,
|
||||
data = super(WorkOrderView, self).normalize(workorder)
|
||||
data.update({
|
||||
'customer_name': workorder.customer.name,
|
||||
'estimated_total': workorder.estimated_total,
|
||||
'notes': workorder.notes,
|
||||
'status_code': workorder.status_code,
|
||||
'status_label': self.enum.WORKORDER_STATUS[workorder.status_code],
|
||||
'date_submitted': six.text_type(workorder.date_submitted or ''),
|
||||
'date_received': six.text_type(workorder.date_received or ''),
|
||||
'date_released': six.text_type(workorder.date_released or ''),
|
||||
'date_delivered': six.text_type(workorder.date_delivered or ''),
|
||||
}
|
||||
})
|
||||
return data
|
||||
|
||||
def create_object(self, data):
|
||||
|
||||
|
|
|
@ -37,6 +37,7 @@ from sqlalchemy.ext.associationproxy import AssociationProxy, ASSOCIATION_PROXY
|
|||
from rattail.time import localtime
|
||||
from rattail.util import prettify, pretty_boolean, pretty_quantity
|
||||
from rattail.core import UNSPECIFIED
|
||||
from rattail.db.util import get_fieldnames
|
||||
|
||||
import colander
|
||||
import deform
|
||||
|
@ -396,19 +397,8 @@ class Form(object):
|
|||
if not self.model_class:
|
||||
raise ValueError("Must define model_class to use make_fields()")
|
||||
|
||||
mapper = orm.class_mapper(self.model_class)
|
||||
|
||||
# first add primary column fields
|
||||
fields = FieldList([prop.key for prop in mapper.iterate_properties
|
||||
if not prop.key.startswith('_')
|
||||
and prop.key != 'versions'])
|
||||
|
||||
# then add association proxy fields
|
||||
for key, desc in sa.inspect(self.model_class).all_orm_descriptors.items():
|
||||
if desc.extension_type == ASSOCIATION_PROXY:
|
||||
fields.append(key)
|
||||
|
||||
return fields
|
||||
return get_fieldnames(self.request.rattail_config, self.model_class,
|
||||
columns=True, proxies=True, relations=True)
|
||||
|
||||
def make_renderers(self):
|
||||
"""
|
||||
|
|
Loading…
Reference in a new issue