def test_get(self):
mbid = 'f27ec8db-af05-4f36-916e-3d57f91ecf5e'
entity = Entity(mbid=mbid, type='test')
db.session.add(entity)
db.session.flush()
same_entity = Entity.get(mbid)
self.assertEqual(entity, same_entity)
missing_csrf = Entity.get('85d9c621-e30f-4788-a962-a089c0d34182')
self.assertIsNone(missing_csrf)
def entity_data(mbid):
use_cache = not request.args.get('no_cache', False)
type_hint = request.args.get('type_hint', None)
entity = None
if use_cache:
entity = Entity.get(mbid=mbid)
if not entity:
entity = Entity.get_remote(mbid=mbid, type_hint=type_hint)
db.session.commit()
if not entity:
return Response(json.dumps({'entity': None}), status=404, mimetype='application/json')
return jsonify(entity=entity.to_dict())
