dinislage/CouchPotatoServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Make sure title is valid when adding it to index

Browse Source
This commit is contained in:
Ruud
2014-03-24 21:40:05 +01:00
parent e7420367f1
commit 0861b21532
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
couchpotato/core/media/_base/media/index.py
+5 -5
View File
@@ -99,7 +99,7 @@ from couchpotato.core.helpers.encoding import simplifyString"""
class TitleIndex(TreeBasedIndex): class TitleIndex(TreeBasedIndex):
_version = 1 _version = 2
custom_header = """from CodernityDB.tree_index import TreeBasedIndex custom_header = """from CodernityDB.tree_index import TreeBasedIndex
from string import ascii_letters from string import ascii_letters
@@ -113,14 +113,14 @@ from couchpotato.core.helpers.encoding import toUnicode, simplifyString"""
return self.simplify(key) return self.simplify(key)
def make_key_value(self, data): def make_key_value(self, data):
if data.get('_t') == 'media' and data.get('title') is not None: if data.get('_t') == 'media' and data.get('title') is not None and len(data.get('title')) > 0:
return self.simplify(data['title']), None return self.simplify(data['title']), None
def simplify(self, title): def simplify(self, title):
title = toUnicode(title) title = toUnicode(title)
nr_prefix = '' if title[0] in ascii_letters else '#' nr_prefix = '' if title and len(title) > 0 and title[0] in ascii_letters else '#'
title = simplifyString(title) title = simplifyString(title)
for prefix in ['the ']: for prefix in ['the ']:
@@ -132,7 +132,7 @@ from couchpotato.core.helpers.encoding import toUnicode, simplifyString"""
class StartsWithIndex(TreeBasedIndex): class StartsWithIndex(TreeBasedIndex):
_version = 1 _version = 2
custom_header = """from CodernityDB.tree_index import TreeBasedIndex custom_header = """from CodernityDB.tree_index import TreeBasedIndex
from string import ascii_letters from string import ascii_letters
@@ -158,4 +158,4 @@ from couchpotato.core.helpers.encoding import toUnicode, simplifyString"""
title = title[len(prefix):] title = title[len(prefix):]
break break
return str(title[0] if title[0] in ascii_letters else '#').lower() return str(title[0] if title and len(title) > 0 and title[0] in ascii_letters else '#').lower()