1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
from zmusic import db
from uuid import uuid4
import time
# This is indeed ugly, but it was suggested by the SQLAlchemy documention
# verbatim as a way of dealing with str-centric apps.
class CoerceUTF8(db.TypeDecorator):
impl = db.Unicode
def process_bind_param(self, value, dialect):
if isinstance(value, str):
value = value.decode('utf-8')
return value
class Song(db.Model):
__tablename__ = 'songs'
filename = db.Column(CoerceUTF8, primary_key=True)
id = db.Column(CoerceUTF8, nullable=False, index=True)
title = db.Column(CoerceUTF8)
album = db.Column(CoerceUTF8)
artist = db.Column(CoerceUTF8)
mimetype = db.Column(CoerceUTF8)
year = db.Column(db.Integer)
track = db.Column(db.Integer)
disc = db.Column(db.Integer)
lastmodified = db.Column(db.Integer)
filesize = db.Column(db.Integer)
length = db.Column(db.Float)
def sync_picard(self, songdict):
self.id = songdict["id"]
self.filename = songdict["filename"]
self.lastmodified = songdict["filestat"].st_mtime
self.filesize = songdict["filestat"].st_size
self.title = songdict["title"]
self.album = songdict["album"]
self.artist = songdict["artist"]
self.mimetype = songdict["mimetype"]
self.length = songdict["length"]
try:
self.year = int(songdict["date"].split("-")[0])
except:
self.year = 0
try:
self.track = int(songdict["tracknumber"])
except:
self.track = 0
try:
self.disc = int(songdict["discnumber"])
except:
self.disc = 0
def to_dict(self):
return { "id": self.id, "track": self.track, "title": self.title, "artist": self.artist,
"album": self.album, "mimetype": self.mimetype, "length": self.length }
class Download(db.Model):
__tablename__ = 'downloads'
id = db.Column(CoerceUTF8, primary_key=True)
leader_id = db.Column(CoerceUTF8)
time = db.Column(db.Integer)
ip = db.Column(CoerceUTF8, index=True)
useragent = db.Column(CoerceUTF8)
song_id = db.Column(CoerceUTF8, db.ForeignKey(Song.id))
artist = db.Column(CoerceUTF8)
album = db.Column(CoerceUTF8)
title = db.Column(CoerceUTF8)
is_zip = db.Column(db.Boolean)
def __init__(self, song, request):
self.artist = song.artist
self.album = song.album
self.title = song.title
self.song_id = song.id
self.time = int(round(time.time() * 100000))
self.useragent = request.headers.get('User-Agent')
ip = request.remote_addr
if (ip.find('::ffff:') == 0 and len(ip) > len('::ffff:')):
ip = ip[len('::ffff:'):]
self.ip = ip
self.id = str(uuid4())
is_zip = False
def to_dict(self):
return { "artist": self.artist, "album": self.album, "title": self.title,
"id": self.song_id, "time": self.time, "useragent": self.useragent,
"ip": self.ip }
|
