1
1
from flask_restful import Resource , abort
2
2
from flask import request , make_response , jsonify
3
- from pbench .server .database .models .metadata import Metadata
3
+ from pbench .server .database .models .metadata import UserMetadata
4
4
from pbench .server .api .auth import Auth
5
5
6
6
@@ -64,7 +64,7 @@ def post(self):
64
64
65
65
try :
66
66
# Create a new metadata object
67
- metadata_object = Metadata (
67
+ metadata_object = UserMetadata (
68
68
value = value , key = metadata_key .lower (), user_id = current_user_id
69
69
)
70
70
# insert the metadata object for a user
@@ -125,7 +125,9 @@ def get(self, key):
125
125
current_user_id = None
126
126
try :
127
127
# Query the metadata object with a given key
128
- metadata_objects = Metadata .query (user_id = current_user_id , key = key .lower ())
128
+ metadata_objects = UserMetadata .query (
129
+ user_id = current_user_id , key = key .lower ()
130
+ )
129
131
data = [
130
132
metadata .get_json (include = ["id" , "created" , "updated" , "value" ])
131
133
for metadata in metadata_objects
@@ -201,7 +203,7 @@ def get(self, id=None):
201
203
202
204
try :
203
205
# Fetch the metadata object
204
- metadata_objects = Metadata .query (id = id )
206
+ metadata_objects = UserMetadata .query (id = id )
205
207
except Exception :
206
208
self .logger .exception (
207
209
"Exception occurred in the GET request while querying the Metadata model, id: {}" ,
@@ -263,7 +265,7 @@ def put(self, id=None):
263
265
abort (400 , message = "Invalid json object in request" )
264
266
265
267
try :
266
- metadata_objects = Metadata .query (id = id )
268
+ metadata_objects = UserMetadata .query (id = id )
267
269
except Exception :
268
270
self .logger .exception (
269
271
"Exception occurred in the PUT request while querying the Metadata model, id: {}" ,
@@ -284,15 +286,17 @@ def put(self, id=None):
284
286
# not present in the metadata db. If any key in the payload does not match
285
287
# with the column name we will abort the update request.
286
288
non_existent = set (post_data .keys ()).difference (
287
- set (Metadata .__table__ .columns .keys ())
289
+ set (UserMetadata .__table__ .columns .keys ())
288
290
)
289
291
if non_existent :
290
292
self .logger .warning (
291
293
"User trying to update fields that are not present in the metadata database. Fields: {}" ,
292
294
non_existent ,
293
295
)
294
296
abort (400 , message = "Invalid fields in update request payload" )
295
- protected = set (post_data .keys ()).intersection (set (Metadata .get_protected ()))
297
+ protected = set (post_data .keys ()).intersection (
298
+ set (UserMetadata .get_protected ())
299
+ )
296
300
for field in protected :
297
301
if getattr (metadata_object , field ) != post_data [field ]:
298
302
self .logger .warning (
@@ -335,7 +339,7 @@ def delete(self, id=None):
335
339
336
340
try :
337
341
# Fetch the metadata object
338
- metadata_objects = Metadata .query (id = id )
342
+ metadata_objects = UserMetadata .query (id = id )
339
343
except Exception :
340
344
self .logger .exception (
341
345
"Exception occurred in the Delete request while querying the Metadata model, id: {}" ,
@@ -354,7 +358,7 @@ def delete(self, id=None):
354
358
355
359
try :
356
360
# Delete the metadata object
357
- Metadata .delete (id = id )
361
+ UserMetadata .delete (id = id )
358
362
except Exception :
359
363
self .logger .exception (
360
364
"Exception occurred in the while deleting the metadata entry, id: {}" ,
0 commit comments