summaryrefslogtreecommitdiff
path: root/bin/scrobble.py
blob: 428e6688b1792b6d3ae2701133e3385c10682e87 (plain)
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#modified version of old gobble.py
try:
    import hashlib
    md5hash = hashlib.md5
except ImportError:
    import md5
    md5hash = md5.new
from optparse import OptionParser
import time
from urllib import urlencode
from urllib2 import urlopen


class ScrobbleException(Exception):

    pass


class ScrobbleServer(object):

    def __init__(self, server_name, username, password, client_code='imp'):
        if server_name[:7] != "http://":
            server_name = "http://%s" % (server_name,)
        self.client_code = client_code
        self.name = server_name
        self.password = password
        self.post_data = []
        self.session_id = None
        self.submit_url = None
        self.username = username
        self._handshake()


    def _handshake(self):
        timestamp = int(time.time())
        token = (md5hash(md5hash(self.password).hexdigest()
                    + str(timestamp)).hexdigest())
        auth_url = "%s/?hs=true&p=1.2&u=%s&t=%d&a=%s&c=%s" % (self.name,
                                                              self.username,
                                                              timestamp,
                                                              token,
                                                              self.client_code)
        response = urlopen(auth_url).read()
        lines = response.split("\n")
        if lines[0] != "OK":
            raise ScrobbleException("Server returned: %s" % (response,))
        self.session_id = lines[1]
        self.submit_url = lines[3]

    def submit(self):
        if len(self.post_data) == 0:
            return
        i = 0
        data = []
        for track in self.post_data:
            data += track.get_tuples(i)
            i += 1
        data += [('s', self.session_id)]
        response = urlopen(self.submit_url, urlencode(data)).read()
        if response != "OK\n":
            raise ScrobbleException("Server returned: %s" % (response,))
        self.post_data = []
        time.sleep(1)

    def add_track(self, scrobble_track):
        i = len(self.post_data)
        if i > 49:
            self.submit()
            i = 0
        self.post_data.append(scrobble_track)


class ScrobbleTrack(object):

    def __init__(self, timestamp, trackname, artistname, albumname=None, \
                 trackmbid=None, tracklength=None, tracknumber=None):
        self.timestamp = timestamp
        self.trackname = trackname
        self.artistname = artistname
        self.albumname = albumname
        self.trackmbid = trackmbid
        self.tracklength = tracklength
        self.tracknumber = tracknumber

    def get_tuples(self, i):
        #timestamp = str(int(time.mktime(self.timestamp.utctimetuple())))
        data = []
        data += [('i[%d]' % i, self.timestamp), ('t[%d]' % i, self.trackname),
                 ('a[%d]' % i, self.artistname)]
        if self.albumname is not None:
            data.append(('b[%d]' % i, self.albumname))
        if self.trackmbid is not None:
            data.append(('m[%d]' % i, self.trackmbid))
        if self.tracklength is not None:
            data.append(('l[%d]' % i, self.tracklength))
        if self.tracknumber is not None:
            data.append(('n[%d]' % i, self.tracknumber))
        return data


def get_parser(usage):
    parser = OptionParser(usage=usage)
    parser.add_option('-s', '--server',
                      help="Server to submit to.  Defaults to"
                           " 'turtle.libre.fm'.")
    parser.set_defaults(server='turtle.libre.fm')
    return parser