3
0
Fork 0

fix: improve support for random objects with grid, master view

thus far we expected either dict or "native" ORM object which can
essentially behave like a dict when needed.  but a "non-native" object
may not behave like a dict and this hopefully fixes the logic to allow
for those anyway..
This commit is contained in:
Lance Edgar 2024-11-25 19:11:41 -06:00
parent ba9021c990
commit dcdc0e7dab
4 changed files with 70 additions and 3 deletions

View file

@ -1940,6 +1940,15 @@ class Grid:
}) })
return filters return filters
def object_to_dict(self, obj):
""" """
try:
dct = dict(obj)
except TypeError:
dct = dict(obj.__dict__)
dct.pop('_sa_instance_state', None)
return dct
def get_vue_context(self): def get_vue_context(self):
""" """
Returns a dict of context for the grid, for use with the Vue Returns a dict of context for the grid, for use with the Vue
@ -1976,7 +1985,7 @@ class Grid:
original_record = record original_record = record
# convert record to new dict # convert record to new dict
record = dict(record) record = self.object_to_dict(record)
# make all values safe for json # make all values safe for json
record = make_json_safe(record, warn=False) record = make_json_safe(record, warn=False)

View file

@ -2018,8 +2018,12 @@ class MasterView(View):
:param obj: Model instance object. :param obj: Model instance object.
""" """
route_prefix = self.get_route_prefix() route_prefix = self.get_route_prefix()
kw = dict([(key, obj[key]) try:
for key in self.get_model_key()]) kw = dict([(key, obj[key])
for key in self.get_model_key()])
except TypeError:
kw = dict([(key, getattr(obj, key))
for key in self.get_model_key()])
kw.update(kwargs) kw.update(kwargs)
return self.request.route_url(f'{route_prefix}.{action}', **kw) return self.request.route_url(f'{route_prefix}.{action}', **kw)

View file

@ -1373,6 +1373,25 @@ class TestGrid(WebTestCase):
filters = grid.get_vue_filters() filters = grid.get_vue_filters()
self.assertEqual(len(filters), 2) self.assertEqual(len(filters), 2)
def test_object_to_dict(self):
grid = self.make_grid()
setting = {'name': 'foo', 'value': 'bar'}
# new dict but with same values
dct = grid.object_to_dict(setting)
self.assertIsInstance(dct, dict)
self.assertIsNot(dct, setting)
self.assertEqual(dct, setting)
# random object, not iterable
class MockSetting:
def __init__(self, **kw):
self.__dict__.update(kw)
mock = MockSetting(**setting)
dct = grid.object_to_dict(mock)
self.assertIsInstance(dct, dict)
self.assertEqual(dct, setting)
def test_get_vue_context(self): def test_get_vue_context(self):
# empty if no columns defined # empty if no columns defined

View file

@ -734,6 +734,41 @@ class TestMasterView(WebTestCase):
self.request.matchdict = {'name': 'blarg'} self.request.matchdict = {'name': 'blarg'}
self.assertRaises(HTTPNotFound, view.get_instance, session=self.session) self.assertRaises(HTTPNotFound, view.get_instance, session=self.session)
def test_get_action_url_for_dict(self):
model = self.app.model
setting = {'name': 'foo', 'value': 'bar'}
with patch.multiple(mod.MasterView, create=True,
model_class=model.Setting):
mod.MasterView.defaults(self.pyramid_config)
view = self.make_view()
url = view.get_action_url_view(setting, 0)
self.assertEqual(url, self.request.route_url('settings.view', name='foo'))
def test_get_action_url_for_orm_object(self):
model = self.app.model
setting = model.Setting(name='foo', value='bar')
self.session.add(setting)
self.session.commit()
with patch.multiple(mod.MasterView, create=True,
model_class=model.Setting):
mod.MasterView.defaults(self.pyramid_config)
view = self.make_view()
url = view.get_action_url_view(setting, 0)
self.assertEqual(url, self.request.route_url('settings.view', name='foo'))
def test_get_action_url_for_adhoc_object(self):
model = self.app.model
class MockSetting:
def __init__(self, **kw):
self.__dict__.update(kw)
setting = MockSetting(name='foo', value='bar')
with patch.multiple(mod.MasterView, create=True,
model_class=model.Setting):
mod.MasterView.defaults(self.pyramid_config)
view = self.make_view()
url = view.get_action_url_view(setting, 0)
self.assertEqual(url, self.request.route_url('settings.view', name='foo'))
def test_get_action_url_view(self): def test_get_action_url_view(self):
model = self.app.model model = self.app.model
setting = model.Setting(name='foo', value='bar') setting = model.Setting(name='foo', value='bar')