mirror of
https://github.com/ilri/dspace-statistics-api.git
synced 2024-11-22 06:15:02 +01:00
Alan Orth
9e942736b1
It is much more efficient to cache view and download statistics in a database than to query Solr on demand (not to mention that it is not possible to page easily with facets in Solr). I decided to use SQLite because it is fast, native in Python 3, and doesn't require any extra steps during provisioning (assuming permissions are ok).
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
# Tested with Python 3.6
|
|
# See DSpace Solr docs for tips about parameters
|
|
# https://wiki.duraspace.org/display/DSPACE/Solr
|
|
|
|
from config import SOLR_CORE
|
|
from database import database_connection
|
|
import falcon
|
|
from solr import solr_connection
|
|
|
|
db = database_connection()
|
|
solr = solr_connection()
|
|
|
|
class ItemResource:
|
|
def on_get(self, req, resp, item_id):
|
|
"""Handles GET requests"""
|
|
|
|
cursor = db.cursor()
|
|
# get item views (and catch the TypeError if item doesn't have any views)
|
|
cursor.execute('SELECT views FROM itemviews WHERE id={0}'.format(item_id))
|
|
try:
|
|
views = cursor.fetchone()['views']
|
|
except:
|
|
views = 0
|
|
|
|
# get item downloads (and catch the TypeError if item doesn't have any downloads)
|
|
cursor.execute('SELECT downloads FROM itemdownloads WHERE id={0}'.format(item_id))
|
|
try:
|
|
downloads = cursor.fetchone()['downloads']
|
|
except:
|
|
downloads = 0
|
|
|
|
cursor.close()
|
|
|
|
statistics = {
|
|
'id': item_id,
|
|
'views': views,
|
|
'downloads': downloads
|
|
}
|
|
|
|
resp.media = statistics
|
|
|
|
api = falcon.API()
|
|
api.add_route('/item/{item_id:int}', ItemResource())
|
|
|
|
# vim: set sw=4 ts=4 expandtab:
|