Skip to content

Commit 4d4655f

Browse files
committed
Update opengraph.py
1 parent aff44c8 commit 4d4655f

1 file changed

Lines changed: 10 additions & 4 deletions

File tree

imageresolver/plugins/opengraph.py

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import logging
44
from bs4 import BeautifulSoup
55
from operator import itemgetter
6+
from urlparse import urlparse, urlunparse
67

78
class Plugin:
89
def get_image(self, url, soup):
@@ -32,7 +33,7 @@ def get_image(self, url, soup):
3233
if ogimages_len == 1:
3334
logger = logging.getLogger('ImageResolver')
3435
logger.debug('Resolving using plugin ' + str(os.path.basename(__file__)) + ' ' + str(url))
35-
return ogimages[0]['url']
36+
resolved_image = ogimages[0]['url']
3637
else:
3738
for image in ogimages:
3839
if re.search('(large|big)', image['url'], re.IGNORECASE):
@@ -41,9 +42,14 @@ def get_image(self, url, soup):
4142
image['score'] += 1
4243

4344
ogimages.sort(key=itemgetter('score'), reverse=True)
44-
45-
return ogimages[0]['url']
46-
45+
resolved_image = ogimages[0]['url']
46+
47+
if not re.search('^https?:', resolved_image):
48+
if resolved_image.startswith('//'):
49+
return 'http:' + resolved_image
50+
else:
51+
return resolved_image
52+
4753

4854
return None
4955

0 commit comments

Comments
 (0)