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())