mirror of
https://github.com/spl0k/supysonic.git
synced 2024-11-14 22:22:18 +00:00
401 lines
11 KiB
Python
Executable File
401 lines
11 KiB
Python
Executable File
# coding: utf-8
|
|
|
|
import config
|
|
|
|
from flask.ext.sqlalchemy import SQLAlchemy
|
|
from sqlalchemy.types import TypeDecorator, BINARY
|
|
from sqlalchemy.ext.hybrid import *
|
|
from sqlalchemy.dialects.postgresql import UUID as pgUUID
|
|
|
|
import uuid, datetime, time
|
|
import mimetypes
|
|
import os.path
|
|
|
|
database = SQLAlchemy()
|
|
|
|
session = database.session
|
|
|
|
Column = database.Column
|
|
Table = database.Table
|
|
String = database.String
|
|
ForeignKey = database.ForeignKey
|
|
func = database.func
|
|
Integer = database.Integer
|
|
Boolean = database.Boolean
|
|
DateTime = database.DateTime
|
|
relationship = database.relationship
|
|
backref = database.backref
|
|
|
|
|
|
|
|
class UUID(TypeDecorator):
|
|
"""Platform-somewhat-independent UUID type
|
|
|
|
Uses Postgresql's UUID type, otherwise uses BINARY(16),
|
|
should be more efficient than a CHAR(32).
|
|
|
|
Mix of http://stackoverflow.com/a/812363
|
|
and http://www.sqlalchemy.org/docs/core/types.html#backend-agnostic-guid-type
|
|
"""
|
|
|
|
impl = BINARY
|
|
|
|
def load_dialect_impl(self, dialect):
|
|
if dialect.name == 'postgresql':
|
|
return dialect.type_descriptor(pgUUID())
|
|
else:
|
|
return dialect.type_descriptor(BINARY(16))
|
|
|
|
def process_bind_param(self, value, dialect):
|
|
if value and isinstance(value, uuid.UUID):
|
|
if dialect.name == 'postgresql':
|
|
return str(value)
|
|
return value.bytes
|
|
if value and not isinstance(value, uuid.UUID):
|
|
raise ValueError, 'value %s is not a valid uuid.UUID' % value
|
|
return None
|
|
|
|
def process_result_value(self, value, dialect):
|
|
if value:
|
|
if dialect.name == 'postgresql':
|
|
return uuid.UUID(value)
|
|
return uuid.UUID(bytes = value)
|
|
return None
|
|
|
|
def is_mutable(self):
|
|
return False
|
|
|
|
@staticmethod
|
|
def gen_id_column():
|
|
return Column(UUID, primary_key = True, default = uuid.uuid4)
|
|
|
|
def now():
|
|
return datetime.datetime.now().replace(microsecond = 0)
|
|
|
|
|
|
class User(database.Model):
|
|
|
|
id = UUID.gen_id_column()
|
|
name = Column(String(64), unique = True)
|
|
mail = Column(String(256))
|
|
password = Column(String(40))
|
|
salt = Column(String(6))
|
|
admin = Column(Boolean, default = False)
|
|
lastfm_session = Column(String(32), nullable = True)
|
|
lastfm_status = Column(Boolean, default = True) # True: ok/unlinked, False: invalid session
|
|
|
|
last_play_id = Column(UUID, ForeignKey('track.id', ondelete = 'SET NULL'), nullable = True)
|
|
last_play = relationship('Track')
|
|
last_play_date = Column(DateTime, nullable = True)
|
|
|
|
def as_subsonic_user(self):
|
|
return {
|
|
'username': self.name,
|
|
'email': self.mail,
|
|
'scrobblingEnabled': self.lastfm_session is not None and self.lastfm_status,
|
|
'adminRole': self.admin,
|
|
'settingsRole': True,
|
|
'downloadRole': True,
|
|
'uploadRole': False,
|
|
'playlistRole': True,
|
|
'coverArtRole': False,
|
|
'commentRole': False,
|
|
'podcastRole': False,
|
|
'streamRole': True,
|
|
'jukeboxRole': False,
|
|
'shareRole': False
|
|
}
|
|
|
|
class ClientPrefs(database.Model):
|
|
|
|
user_id = Column(UUID, ForeignKey('user.id'), primary_key = True)
|
|
client_name = Column(String(32), nullable = False, primary_key = True)
|
|
format = Column(String(8), nullable = True)
|
|
bitrate = Column(Integer, nullable = True)
|
|
|
|
class Folder(database.Model):
|
|
|
|
id = UUID.gen_id_column()
|
|
root = Column(Boolean, default = False)
|
|
path = Column(String(4096)) # should be unique, but mysql don't like such large columns
|
|
created = Column(DateTime, default = now)
|
|
has_cover_art = Column(Boolean, default = False)
|
|
last_scan = Column(Integer, default = 0)
|
|
|
|
@hybrid_property
|
|
def name(self):
|
|
return self.path[self.path.rfind(os.sep) + 1:]
|
|
|
|
def get_children(self):
|
|
return Folder.query.filter(Folder.path.like(self.path + '/%%')).filter(~Folder.path.like(self.path + '/%%/%%'))
|
|
|
|
def as_subsonic_child(self, user):
|
|
info = {
|
|
'id': str(self.id),
|
|
'isDir': True,
|
|
'title': self.name,
|
|
'album': self.name,
|
|
'created': self.created.isoformat()
|
|
}
|
|
if not self.root:
|
|
parent = session.query(Folder) \
|
|
.filter(Folder.path.like(self.path[:len(self.path)-len(self.name)-1])) \
|
|
.order_by(func.length(Folder.path).desc()).first()
|
|
if(parent):
|
|
info['parent'] = str(parent.id)
|
|
info['artist'] = parent.name
|
|
if self.has_cover_art:
|
|
info['coverArt'] = str(self.id)
|
|
|
|
starred = StarredFolder.query.get((user.id, self.id))
|
|
if starred:
|
|
info['starred'] = starred.date.isoformat()
|
|
|
|
rating = RatingFolder.query.get((user.id, self.id))
|
|
if rating:
|
|
info['userRating'] = rating.rating
|
|
avgRating = RatingFolder.query.filter(RatingFolder.rated_id == self.id).value(func.avg(RatingFolder.rating))
|
|
if avgRating:
|
|
info['averageRating'] = avgRating
|
|
|
|
return info
|
|
|
|
class Artist(database.Model):
|
|
|
|
id = UUID.gen_id_column()
|
|
name = Column(String(255), nullable=False)
|
|
albums = relationship('Album', backref = 'artist')
|
|
|
|
def as_subsonic_artist(self, user):
|
|
info = {
|
|
'id': str(self.id),
|
|
'name': self.name,
|
|
# coverArt
|
|
'albumCount': len(self.albums)
|
|
}
|
|
|
|
starred = StarredArtist.query.get((user.id, self.id))
|
|
if starred:
|
|
info['starred'] = starred.date.isoformat()
|
|
|
|
return info
|
|
|
|
class Album(database.Model):
|
|
|
|
id = UUID.gen_id_column()
|
|
name = Column(String(255))
|
|
artist_id = Column(UUID, ForeignKey('artist.id'))
|
|
tracks = relationship('Track', backref = 'album', cascade="delete")
|
|
|
|
def as_subsonic_album(self, user):
|
|
info = {
|
|
'id': str(self.id),
|
|
'name': self.name,
|
|
'artist': self.artist.name,
|
|
'artistId': str(self.artist_id),
|
|
'songCount': len(self.tracks),
|
|
'duration': sum(map(lambda t: t.duration, self.tracks)),
|
|
'created': min(map(lambda t: t.created, self.tracks)).isoformat()
|
|
}
|
|
if self.tracks[0].folder.has_cover_art:
|
|
info['coverArt'] = str(self.tracks[0].folder_id)
|
|
|
|
starred = StarredAlbum.query.get((user.id, self.id))
|
|
if starred:
|
|
info['starred'] = starred.date.isoformat()
|
|
|
|
return info
|
|
|
|
def sort_key(self):
|
|
year = min(map(lambda t: t.year if t.year else 9999, self.tracks))
|
|
return '%i%s' % (year, self.name.lower())
|
|
|
|
class Track(database.Model):
|
|
|
|
id = UUID.gen_id_column()
|
|
disc = Column(Integer)
|
|
number = Column(Integer)
|
|
title = Column(String(256))
|
|
year = Column(Integer, nullable = True)
|
|
genre = Column(String(256), nullable = True)
|
|
duration = Column(Integer)
|
|
album_id = Column(UUID, ForeignKey('album.id'))
|
|
bitrate = Column(Integer)
|
|
|
|
path = Column(String(4096)) # should be unique, but mysql don't like such large columns
|
|
created = Column(DateTime, default = now)
|
|
last_modification = Column(Integer)
|
|
|
|
play_count = Column(Integer, default = 0)
|
|
last_play = Column(DateTime, nullable = True)
|
|
|
|
folder_id = Column(UUID, ForeignKey('folder.id', ondelete="CASCADE"))
|
|
folder = relationship('Folder', backref = 'tracks')
|
|
|
|
def as_subsonic_child(self, user):
|
|
info = {
|
|
'id': str(self.id),
|
|
'parent': str(self.folder.id),
|
|
'isDir': False,
|
|
'title': self.title,
|
|
'album': self.album.name,
|
|
'artist': self.album.artist.name,
|
|
'track': self.number,
|
|
'size': os.path.getsize(self.path),
|
|
'contentType': mimetypes.guess_type(self.path),
|
|
'suffix': self.suffix(),
|
|
'duration': self.duration,
|
|
'bitRate': self.bitrate,
|
|
'path': self.path,
|
|
'isVideo': False,
|
|
'discNumber': self.disc,
|
|
'created': self.created.isoformat(),
|
|
'albumId': str(self.album.id),
|
|
'artistId': str(self.album.artist.id),
|
|
'type': 'music'
|
|
}
|
|
|
|
if self.year:
|
|
info['year'] = self.year
|
|
if self.genre:
|
|
info['genre'] = self.genre
|
|
if self.folder.has_cover_art:
|
|
info['coverArt'] = str(self.folder_id)
|
|
|
|
starred = StarredTrack.query.get((user.id, self.id))
|
|
if starred:
|
|
info['starred'] = starred.date.isoformat()
|
|
|
|
rating = RatingTrack.query.get((user.id, self.id))
|
|
if rating:
|
|
info['userRating'] = rating.rating
|
|
avgRating = RatingTrack.query.filter(RatingTrack.rated_id == self.id).value(func.avg(RatingTrack.rating))
|
|
if avgRating:
|
|
info['averageRating'] = avgRating
|
|
|
|
if self.suffix() == 'flac':
|
|
info['transcodedContentType'] = 'audio/ogg'
|
|
info['transcodedSuffix'] = 'ogg'
|
|
|
|
return info
|
|
|
|
def duration_str(self):
|
|
ret = '%02i:%02i' % ((self.duration % 3600) / 60, self.duration % 60)
|
|
if self.duration >= 3600:
|
|
ret = '%02i:%s' % (self.duration / 3600, ret)
|
|
return ret
|
|
|
|
def suffix(self):
|
|
return os.path.splitext(self.path)[1][1:].lower()
|
|
|
|
def sort_key(self):
|
|
return (self.album.artist.name + self.album.name + ("%02i" % self.disc) + ("%02i" % self.number) + self.title).lower()
|
|
|
|
class StarredFolder(database.Model):
|
|
|
|
user_id = Column(UUID, ForeignKey('user.id'), primary_key = True)
|
|
starred_id = Column(UUID, ForeignKey('folder.id'), primary_key = True)
|
|
date = Column(DateTime, default = now)
|
|
|
|
user = relationship('User')
|
|
starred = relationship('Folder')
|
|
|
|
class StarredArtist(database.Model):
|
|
|
|
user_id = Column(UUID, ForeignKey('user.id'), primary_key = True)
|
|
starred_id = Column(UUID, ForeignKey('artist.id'), primary_key = True)
|
|
date = Column(DateTime, default = now)
|
|
|
|
user = relationship('User')
|
|
starred = relationship('Artist')
|
|
|
|
class StarredAlbum(database.Model):
|
|
|
|
user_id = Column(UUID, ForeignKey('user.id'), primary_key = True)
|
|
starred_id = Column(UUID, ForeignKey('album.id'), primary_key = True)
|
|
date = Column(DateTime, default = now)
|
|
|
|
user = relationship('User')
|
|
starred = relationship('Album')
|
|
|
|
class StarredTrack(database.Model):
|
|
|
|
user_id = Column(UUID, ForeignKey('user.id'), primary_key = True)
|
|
starred_id = Column(UUID, ForeignKey('track.id'), primary_key = True)
|
|
date = Column(DateTime, default = now)
|
|
|
|
user = relationship('User')
|
|
starred = relationship('Track')
|
|
|
|
class RatingFolder(database.Model):
|
|
|
|
user_id = Column(UUID, ForeignKey('user.id'), primary_key = True)
|
|
rated_id = Column(UUID, ForeignKey('folder.id'), primary_key = True)
|
|
rating = Column(Integer)
|
|
|
|
user = relationship('User')
|
|
rated = relationship('Folder')
|
|
|
|
class RatingTrack(database.Model):
|
|
|
|
user_id = Column(UUID, ForeignKey('user.id'), primary_key = True)
|
|
rated_id = Column(UUID, ForeignKey('track.id'), primary_key = True)
|
|
rating = Column(Integer)
|
|
|
|
user = relationship('User')
|
|
rated = relationship('Track')
|
|
|
|
class ChatMessage(database.Model):
|
|
|
|
id = UUID.gen_id_column()
|
|
user_id = Column(UUID, ForeignKey('user.id'))
|
|
time = Column(Integer, default = lambda: int(time.time()))
|
|
message = Column(String(512))
|
|
|
|
user = relationship('User')
|
|
|
|
def responsize(self):
|
|
return {
|
|
'username': self.user.name,
|
|
'time': self.time * 1000,
|
|
'message': self.message
|
|
}
|
|
|
|
playlist_track_assoc = Table('playlist_track', database.Model.metadata,
|
|
Column('playlist_id', UUID, ForeignKey('playlist.id')),
|
|
Column('track_id', UUID, ForeignKey('track.id'))
|
|
)
|
|
|
|
class Playlist(database.Model):
|
|
|
|
id = UUID.gen_id_column()
|
|
user_id = Column(UUID, ForeignKey('user.id'))
|
|
name = Column(String(256))
|
|
comment = Column(String(256), nullable = True)
|
|
public = Column(Boolean, default = False)
|
|
created = Column(DateTime, default = now)
|
|
|
|
user = relationship('User')
|
|
tracks = relationship('Track', secondary = playlist_track_assoc)
|
|
|
|
def as_subsonic_playlist(self, user):
|
|
info = {
|
|
'id': str(self.id),
|
|
'name': self.name if self.user_id == user.id else '[%s] %s' % (self.user.name, self.name),
|
|
'owner': self.user.name,
|
|
'public': self.public,
|
|
'songCount': len(self.tracks),
|
|
'duration': sum(map(lambda t: t.duration, self.tracks)),
|
|
'created': self.created.isoformat()
|
|
}
|
|
if self.comment:
|
|
info['comment'] = self.comment
|
|
return info
|
|
|
|
def init_db():
|
|
database.create_all()
|
|
|
|
def recreate_db():
|
|
database.drop_all()
|
|
database.create_all()
|