80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Contains helper functions for accessing the database.
|
|
"""
|
|
import os
|
|
from urllib import parse
|
|
|
|
import config
|
|
|
|
|
|
async def get_random_track(request):
|
|
"""Selects a random track."""
|
|
async with request.app.state.db_pool.acquire() as conn:
|
|
track = await conn.fetchrow(
|
|
"SELECT * FROM track ORDER BY random() LIMIT 1"
|
|
)
|
|
track = dict(track)
|
|
fpath = track.pop('filepath')
|
|
art256 = os.path.join(os.path.dirname(fpath), 'folder-256x256.jpg')
|
|
art64 = os.path.join(os.path.dirname(fpath), 'folder-64x64.jpg')
|
|
track['source'] = convert_filepath(fpath)
|
|
track['artwork'] = [{
|
|
'src': convert_filepath(art256),
|
|
'sizes': '256x256',
|
|
'type': 'image/jpeg'
|
|
}
|
|
]
|
|
return track
|
|
|
|
|
|
async def get_artists(request):
|
|
"""Returns all artists in the database."""
|
|
async with request.app.state.db_pool.acquire() as conn:
|
|
artists = await conn.fetch(
|
|
"SELECT DISTINCT albumartist FROM track ORDER BY albumartist"
|
|
)
|
|
artists = [a['albumartist'] for a in artists]
|
|
return artists
|
|
|
|
|
|
async def get_albums(request, albumartist):
|
|
"""Return all albums associated with a particular albumartist."""
|
|
async with request.app.state.db_pool.acquire() as conn:
|
|
albums = await conn.fetch(
|
|
"SELECT DISTINCT ON (album, date) filepath, album, date "
|
|
"FROM track "
|
|
"WHERE albumartist = $1 ORDER BY date",
|
|
albumartist
|
|
)
|
|
albums = [dict(a) for a in albums]
|
|
for album in albums:
|
|
fpath = album.pop('filepath')
|
|
cpath = os.path.join(os.path.dirname(fpath), 'folder-64x64.jpg')
|
|
album['cover_art_path'] = convert_filepath(cpath)
|
|
return albums
|
|
|
|
|
|
async def get_tracks(request, albumartist, album, date):
|
|
"""Select an album and return all associated tracks."""
|
|
async with request.app.state.db_pool.acquire() as conn:
|
|
tracks = await conn.fetch(
|
|
"SELECT * FROM track "
|
|
"WHERE albumartist = $1 AND "
|
|
"album = $2 AND date = $3 "
|
|
"ORDER BY discnumber ASC, tracknumber ASC",
|
|
albumartist, album, date
|
|
)
|
|
tracks = [dict(t) for t in tracks]
|
|
for track in tracks:
|
|
track['filepath'] = convert_filepath(track['filepath'])
|
|
track.pop('last_modified')
|
|
return tracks
|
|
|
|
|
|
def convert_filepath(path):
|
|
"""Convert a filepath to a URL."""
|
|
path = config.server_homepage + '/library/' + path
|
|
#path = parse.quote(path)
|
|
return path
|