11from flask_restful import Resource , abort
22from flask import request , make_response , jsonify
3- from pbench .server .database .models .metadata import Metadata
3+ from pbench .server .database .models .metadata import UserMetadata
44from pbench .server .api .auth import Auth
55
66
@@ -64,7 +64,7 @@ def post(self):
6464
6565 try :
6666 # Create a new metadata object
67- metadata_object = Metadata (
67+ metadata_object = UserMetadata (
6868 value = value , key = metadata_key .lower (), user_id = current_user_id
6969 )
7070 # insert the metadata object for a user
@@ -125,7 +125,9 @@ def get(self, key):
125125 current_user_id = None
126126 try :
127127 # 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+ )
129131 data = [
130132 metadata .get_json (include = ["id" , "created" , "updated" , "value" ])
131133 for metadata in metadata_objects
@@ -201,7 +203,7 @@ def get(self, id=None):
201203
202204 try :
203205 # Fetch the metadata object
204- metadata_objects = Metadata .query (id = id )
206+ metadata_objects = UserMetadata .query (id = id )
205207 except Exception :
206208 self .logger .exception (
207209 "Exception occurred in the GET request while querying the Metadata model, id: {}" ,
@@ -263,7 +265,7 @@ def put(self, id=None):
263265 abort (400 , message = "Invalid json object in request" )
264266
265267 try :
266- metadata_objects = Metadata .query (id = id )
268+ metadata_objects = UserMetadata .query (id = id )
267269 except Exception :
268270 self .logger .exception (
269271 "Exception occurred in the PUT request while querying the Metadata model, id: {}" ,
@@ -284,15 +286,17 @@ def put(self, id=None):
284286 # not present in the metadata db. If any key in the payload does not match
285287 # with the column name we will abort the update request.
286288 non_existent = set (post_data .keys ()).difference (
287- set (Metadata .__table__ .columns .keys ())
289+ set (UserMetadata .__table__ .columns .keys ())
288290 )
289291 if non_existent :
290292 self .logger .warning (
291293 "User trying to update fields that are not present in the metadata database. Fields: {}" ,
292294 non_existent ,
293295 )
294296 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+ )
296300 for field in protected :
297301 if getattr (metadata_object , field ) != post_data [field ]:
298302 self .logger .warning (
@@ -335,7 +339,7 @@ def delete(self, id=None):
335339
336340 try :
337341 # Fetch the metadata object
338- metadata_objects = Metadata .query (id = id )
342+ metadata_objects = UserMetadata .query (id = id )
339343 except Exception :
340344 self .logger .exception (
341345 "Exception occurred in the Delete request while querying the Metadata model, id: {}" ,
@@ -354,7 +358,7 @@ def delete(self, id=None):
354358
355359 try :
356360 # Delete the metadata object
357- Metadata .delete (id = id )
361+ UserMetadata .delete (id = id )
358362 except Exception :
359363 self .logger .exception (
360364 "Exception occurred in the while deleting the metadata entry, id: {}" ,
0 commit comments