1
0
mirror of https://github.com/spl0k/supysonic.git synced 2024-11-10 04:02:17 +00:00

Some work on encodings

This commit is contained in:
spl0k 2013-12-09 19:06:14 +01:00
parent 64e686584a
commit 815944ca6f
2 changed files with 41 additions and 5 deletions

View File

@ -129,15 +129,17 @@ class ResponseHelper:
""" """
if not isinstance(dictionary, dict): if not isinstance(dictionary, dict):
raise TypeError('Expecting a dict') raise TypeError('Expecting a dict')
if not all(map(lambda x: isinstance(x, basestring), dictionary.keys())):
raise TypeError('Dictionary keys must be strings')
subelems = { k: v for k, v in dictionary.iteritems() if isinstance(v, dict) } subelems = { k: v for k, v in dictionary.iteritems() if isinstance(v, dict) }
sequences = { k: v for k, v in dictionary.iteritems() if isinstance(v, list) } sequences = { k: v for k, v in dictionary.iteritems() if isinstance(v, list) }
attributes = { k: v for k, v in dictionary.iteritems() if k != '_value_' and k not in subelems and k not in sequences } attributes = { k: v for k, v in dictionary.iteritems() if k != '_value_' and k not in subelems and k not in sequences }
if '_value_' in dictionary: if '_value_' in dictionary:
elem.text = dictionary['_value_'] elem.text = ResponseHelper.value_tostring(dictionary['_value_'])
for attr, value in attributes.iteritems(): for attr, value in attributes.iteritems():
elem.set(attr, str(value).lower() if isinstance(value, bool) else str(value)) elem.set(attr, ResponseHelper.value_tostring(value))
for sub, subdict in subelems.iteritems(): for sub, subdict in subelems.iteritems():
subelem = ElementTree.SubElement(elem, sub) subelem = ElementTree.SubElement(elem, sub)
ResponseHelper.dict2xml(subelem, subdict) ResponseHelper.dict2xml(subelem, subdict)
@ -146,6 +148,14 @@ class ResponseHelper:
subelem = ElementTree.SubElement(elem, seq) subelem = ElementTree.SubElement(elem, seq)
ResponseHelper.dict2xml(subelem, subdict) ResponseHelper.dict2xml(subelem, subdict)
@staticmethod
def value_tostring(value):
if isinstance(value, basestring):
return value
if isinstance(value, bool):
return str(value).lower()
return str(value)
def get_entity(req, ent, param = 'id'): def get_entity(req, ent, param = 'id'):
eid = req.args.get(param) eid = req.args.get(param)
if not eid: if not eid:

View File

@ -4,6 +4,7 @@ from flask import request, send_file, Response
import os.path import os.path
from PIL import Image from PIL import Image
import subprocess import subprocess
import codecs
import config, scanner import config, scanner
from web import app from web import app
@ -145,9 +146,16 @@ def lyrics():
for track in query: for track in query:
lyrics_path = os.path.splitext(track.path)[0] + '.txt' lyrics_path = os.path.splitext(track.path)[0] + '.txt'
if os.path.exists(lyrics_path): if os.path.exists(lyrics_path):
print lyrics_path app.logger.debug('Found lyrics file: ' + lyrics_path)
with open(lyrics_path, 'r') as lyrics_file:
lyrics = lyrics_file.read() try:
lyrics = read_file_as_unicode(lyrics_path)
except UnicodeError:
# Lyrics file couldn't be decoded. Rather than displaying an error, try with the potential next files or
# return no lyrics. Log it anyway.
app.logger.warn('Unsupported encoding for lyrics file ' + lyrics_path)
continue
return request.formatter({ 'lyrics': { return request.formatter({ 'lyrics': {
'artist': track.album.artist.name, 'artist': track.album.artist.name,
'title': track.title, 'title': track.title,
@ -156,3 +164,21 @@ def lyrics():
return request.formatter({ 'lyrics': {} }) return request.formatter({ 'lyrics': {} })
def read_file_as_unicode(path):
""" Opens a file trying with different encodings and returns the contents as a unicode string """
encodings = [ 'utf-8', 'latin1' ] # Should be extended to support more encodings
for enc in encodings:
try:
contents = codecs.open(path, 'r', encoding = enc).read()
app.logger.debug('Read file {} with {} encoding'.format(path, enc))
# Maybe save the encoding somewhere to prevent going through this loop each time for the same file
return contents
except UnicodeError:
pass
# Fallback to ASCII
app.logger.debug('Reading file {} with ascii encoding'.format(path))
return unicode(open(path, 'r').read())