class TagMaker(Resource):
success_marshaller = {
"success": fields.Integer,
}
def __init__(self, app, jwt):
self.db = Database()
self.db.connect()
self.jwt = jwt
@marshal_with(success_marshaller)
@jwt_required
def post(self):
parser = reqparse.RequestParser()
parser.add_argument("tag", type=str)
args = parser.parse_args()
tag = args["tag"]
username = get_jwt_identity()
userId = self.db.nameToId(username)
self.db.addTag(username, tag)
return {"success": 1}
class TagEndpoint(Resource):
success_marshaller = {
"success": fields.Integer,
}
tag_marshaller = {
"success": fields.Integer,
"tags": fields.List,
}
def __init__(self, app, jwt):
self.db = Database()
self.db.connect()
self.jwt = jwt
@marshal_with(success_marshaller)
@jwt_required
def post(self):
parser = reqparse.RequestParser()
parser.add_argument("tag", type=str)
parser.add_argument(
"remove", type=int
) #user can choose the remove option, which will delete the specified tag for the user if it exists
args = parser.parse_args()
tag = args["tag"]
rem = args["remove"]
username = get_jwt_identity()
userId = self.db.nameToId(username)
if rem is None or rem == 0:
self.db.addTag(username, tag)
else:
self.db.removeTag(username, tag)
return {"success": 1}
#@marshal_with(tag_marshaller)
@jwt_required
def get(self):
username = get_jwt_identity()
tags = self.db.getTags(username)
resp = dict()
if tags:
resp["success"] = 1
resp["tags"] = tags
else:
resp["success"] = 0
resp["tags"] = None
print(resp)
return resp
