Add meme count endpoint

This commit is contained in:
Tyler Perkins 2023-02-24 11:40:01 -05:00
parent f903f4e10c
commit 65cde7fcc9
2 changed files with 24 additions and 0 deletions

View File

@ -1,6 +1,7 @@
from flask_restx import Api from flask_restx import Api
from .search import api as searchNamespace from .search import api as searchNamespace
from .get import api as getNamespace from .get import api as getNamespace
from .util import api as utilNamespace
api = Api( api = Api(
title='Memes', title='Memes',
@ -10,3 +11,4 @@ api = Api(
api.add_namespace(searchNamespace) api.add_namespace(searchNamespace)
api.add_namespace(getNamespace) api.add_namespace(getNamespace)
api.add_namespace(utilNamespace)

22
src/api/util.py Normal file
View File

@ -0,0 +1,22 @@
from flask_restx import Namespace, Resource, fields
from flask_restx import reqparse
from api.clientGetter import getClientSafely
import logging
# Exported namespace
api = Namespace('util', description='Misc Utilities')
@api.route('/count')
@api.doc(description="Get number of memes in store")
class getCount(Resource):
def get(self):
client = getClientSafely()
if client is None:
return {
"message": "Error connecting to S3"
}, 500
return {
"count" : len(client.getCurrentMemeList())
}, 200