0x2620/python-ox
2
1
Fork 0
Code Issues Pull requests Releases Wiki Activity

include title in connections

Browse source
This commit is contained in:
j 2011-09-30 19:52:13 +02:00
parent 0ff79d432f
commit 40b3dabb99
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
ox/web/imdb.py
View file

@ -300,7 +300,18 @@ class Imdb(SiteParser):
if len(self['connections']) == 2 and isinstance(self['connections'][0], basestring): if len(self['connections']) == 2 and isinstance(self['connections'][0], basestring):
self['connections'] = [self['connections']] self['connections'] = [self['connections']]
for rel, data in self['connections']: for rel, data in self['connections']:
cc[unicode(rel)] = re.compile('<a href="/title/tt(\d{7})/">').findall(data) #cc[unicode(rel)] = re.compile('<a href="/title/tt(\d{7})/">(.*?)</a>').findall(data)
def get_conn(c):
title = c[1]
if title.startswith('"') and title.endswith('"'):
title = title[1:-1]
return {
'id': c[0],
'title': title
}
cc[unicode(rel)] = map(get_conn, re.compile('<a href="/title/tt(\d{7})/">(.*?)</a>').findall(data))
self['connections'] = cc self['connections'] = cc
for key in ('countries', 'genres'): for key in ('countries', 'genres'):