store size in the mp3 table

server/djrandom/frontend/frontend.py

@@ -83,9 +83,7 @@ def song_info_json(sha1):
     if not mp3:
         abort(404)
     # Use the mp3.to_dict() data, but add some more things...
-    data = mp3.to_dict()
-    data['size'] = os.path.getsize(mp3.path)
-    return jsonify(data)
+    return jsonify(mp3.to_dict())
 
 
 @app.route('/json/playing', methods=['POST'])

server/djrandom/model/mp3.py

@@ -15,6 +15,7 @@ class MP3(Base):
     ready = Column(Boolean, default=False, index=True)
     error = Column(Boolean, default=False, index=True)
     path = Column(String(1024))
+    size = Column(Integer())
     artist = Column(Unicode(256))
     title = Column(Unicode(256))
     album = Column(Unicode(256))
@@ -32,6 +33,7 @@ class MP3(Base):
                 'album': self.album,
                 'genre': self.genre,
                 'sha1': self.sha1,
+                'size': self.size,
                 'uploaded_at': self.uploaded_at}
 
 

server/djrandom/receiver/receiver.py

@@ -39,6 +39,7 @@ def _upload_mp3(incoming_fd, sha1):
     if utils.sha1_of_file(mp3.path) != sha1:
         log.error('sha1 mismatch')
         raise Error('SHA1 mismatch')
+    mp3.size = os.path.getsize(mp3.path)
     Session.add(mp3)
     Session.commit()
     log.info('successfully stored %s' % (sha1,))