Skip to content

Commit 3ea4cef

Browse files
committed
update user metadata table name
1 parent 0615b56 commit 3ea4cef

File tree

2 files changed

+18
-14
lines changed

2 files changed

+18
-14
lines changed

lib/pbench/server/api/resources/metadata_api.py

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
from flask_restful import Resource, abort
22
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
44
from 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: {}",

lib/pbench/server/database/models/metadata.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,11 @@
33
from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, Text
44

55

6-
class Metadata(Database.Base):
6+
class UserMetadata(Database.Base):
77
""" Metadata Model for storing user metadata details """
88

99
# TODO: Think about the better name
10-
__tablename__ = "metadata"
10+
__tablename__ = "user_metadata"
1111

1212
id = Column(Integer, primary_key=True, autoincrement=True)
1313
created = Column(DateTime, nullable=False, default=datetime.datetime.now())
@@ -28,9 +28,9 @@ def get_protected():
2828

2929
@staticmethod
3030
def query(**kwargs):
31-
query = Database.db_session.query(Metadata)
31+
query = Database.db_session.query(UserMetadata)
3232
for attr, value in kwargs.items():
33-
query = query.filter(getattr(Metadata, attr) == value)
33+
query = query.filter(getattr(UserMetadata, attr) == value)
3434
return query.all()
3535

3636
def add(self):
@@ -64,7 +64,7 @@ def delete(id):
6464
:return:
6565
"""
6666
try:
67-
metadata_query = Database.db_session.query(Metadata).filter_by(id=id)
67+
metadata_query = Database.db_session.query(UserMetadata).filter_by(id=id)
6868
metadata_query.delete()
6969
Database.db_session.commit()
7070
return True

0 commit comments

Comments
 (0)