mirror of
https://gitlab.science.ru.nl/technicie/MarietjeDjango.git
synced 2025-12-09 22:52:23 +01:00
Add basic metrics export
This commit is contained in:
@ -12,11 +12,16 @@ from django.shortcuts import get_object_or_404
|
||||
from django.views.decorators.http import require_http_methods
|
||||
from django.conf import settings
|
||||
from mutagen import File
|
||||
from prometheus_client import Counter
|
||||
|
||||
|
||||
from marietje.utils import song_to_dict, playlist_song_to_dict, send_to_bertha
|
||||
from queues.models import PlaylistSong, QueueCommand
|
||||
from songs.models import Song
|
||||
|
||||
request_counter = Counter('marietje_requests', 'Queue requests on marietje', ['queue'])
|
||||
upload_counter = Counter('marietje_uploads', 'Songs uploaded to marietje')
|
||||
|
||||
|
||||
def api_auth_required(view_func):
|
||||
@wraps(view_func)
|
||||
@ -215,6 +220,7 @@ def request(request):
|
||||
queue = request.user.queue
|
||||
song = get_object_or_404(Song, id=request.POST.get('id'), deleted=False)
|
||||
if queue.request(song, request.user):
|
||||
request_counter.labels(queue=queue.name).inc()
|
||||
return JsonResponse({
|
||||
'success': True
|
||||
})
|
||||
@ -265,6 +271,7 @@ def upload(request):
|
||||
song = Song(user=request.user, artist=artists[i], title=titles[i], hash=hash, duration=duration)
|
||||
song.save()
|
||||
|
||||
upload_counter.inc()
|
||||
return JsonResponse({'success': True})
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user