mirror of
https://github.com/ilri/dspace-statistics-api.git
synced 2024-11-14 18:37:03 +01:00
Alan Orth
4cf8656b35
I think it's more obvious if the "all items" route is plural. Also, this will allow me to eventually put documentation at the root.
66 lines
2.0 KiB
Python
66 lines
2.0 KiB
Python
from database import database_connection
|
|
import falcon
|
|
from solr import solr_connection
|
|
|
|
db = database_connection()
|
|
db.set_session(readonly=True)
|
|
solr = solr_connection()
|
|
|
|
class AllItemsResource:
|
|
def on_get(self, req, resp):
|
|
"""Handles GET requests"""
|
|
# Return HTTPBadRequest if id parameter is not present and valid
|
|
limit = req.get_param_as_int("limit", min=0, max=100) or 100
|
|
page = req.get_param_as_int("page", min=0) or 0
|
|
offset = limit * page
|
|
|
|
cursor = db.cursor()
|
|
|
|
# get total number of items so we can estimate the pages
|
|
cursor.execute('SELECT COUNT(id) FROM items')
|
|
pages = round(cursor.fetchone()[0] / limit)
|
|
|
|
# get statistics, ordered by id, and use limit and offset to page through results
|
|
cursor.execute('SELECT id, views, downloads FROM items ORDER BY id ASC LIMIT {} OFFSET {}'.format(limit, offset))
|
|
results = cursor.fetchmany(limit)
|
|
cursor.close()
|
|
|
|
# create a list to hold dicts of item stats
|
|
statistics = list()
|
|
|
|
# iterate over results and build statistics object
|
|
for item in results:
|
|
statistics.append({ 'id': item['id'], 'views': item['views'], 'downloads': item['downloads'] })
|
|
|
|
message = {
|
|
'currentPage': page,
|
|
'totalPages': pages,
|
|
'limit': limit,
|
|
'statistics': statistics
|
|
}
|
|
|
|
resp.media = message
|
|
|
|
class ItemResource:
|
|
def on_get(self, req, resp, item_id):
|
|
"""Handles GET requests"""
|
|
|
|
cursor = db.cursor()
|
|
cursor.execute('SELECT views, downloads FROM items WHERE id={}'.format(item_id))
|
|
results = cursor.fetchone()
|
|
cursor.close()
|
|
|
|
statistics = {
|
|
'id': item_id,
|
|
'views': results['views'],
|
|
'downloads': results['downloads']
|
|
}
|
|
|
|
resp.media = statistics
|
|
|
|
api = falcon.API()
|
|
api.add_route('/items', AllItemsResource())
|
|
api.add_route('/item/{item_id:int}', ItemResource())
|
|
|
|
# vim: set sw=4 ts=4 expandtab:
|