Skip to content

Commit c34f274

Browse files
committed
use resource.display_title when stripping show title
1 parent 8252f9a commit c34f274

File tree

3 files changed

+9
-3
lines changed

3 files changed

+9
-3
lines changed

Contents/Code/generic.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,9 @@ def ListTVShow(endpoint, show_title, refresh = 0):
4343
import re
4444

4545
container, response = render_listings(endpoint + '/episodes', show_title, return_response = True, flags = ['persisted'])
46-
title_regex = re.compile(ur'^(.*)' + re.escape(show_title) + ur':?\s+', re.UNICODE)
46+
title_regex = re.compile(ur'^(.*)'
47+
+ re.escape(response['resource']['display_title'])
48+
+ ur':?\s+', re.UNICODE)
4749

4850
for item in container.objects:
4951
md = title_regex.match(item.title)

Contents/Tests/helpers/listings_elements.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
show = dict(
1616
_type = 'show',
1717
display_title = 'foo',
18-
endpoint = 'bar',
18+
endpoint = '/shows/1',
1919
display_overview = '42',
2020
artwork = 'http://example.com/artwork.jpg'
2121
)

Contents/Tests/test_generic.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,9 +286,13 @@ def download_includes(e):
286286

287287
return False
288288

289+
response = dict(
290+
items = [ mocks['episode'], mocks['episode2'] ],
291+
resource = mocks['show']
292+
)
289293
@mock.patch.object(bridge.favorite, 'includes', return_value = True)
290294
@mock.patch.object(bridge.download, 'includes', side_effect = download_includes)
291-
@mock.patch.object(JSON, 'ObjectFromURL', return_value = dict(items = [ mocks['episode'], mocks['episode2'] ]))
295+
@mock.patch.object(JSON, 'ObjectFromURL', return_value = response)
292296
def test(*a):
293297
container = generic.ListTVShow(endpoint = '/shows/1', show_title = 'foo')
294298
first = container.objects[1]

0 commit comments

Comments
 (0)