Forráskód Böngészése

[interlinks] support img src attr too

Andy Li 9 éve
szülő
commit
8fc2e0ebf9
1 módosított fájl, 8 hozzáadás és 0 törlés
  1. 8 0
      interlinks/interlinks.py

+ 8 - 0
interlinks/interlinks.py

@@ -38,6 +38,14 @@ def content_object_init(instance):
 				if name in interlinks:
 					hi = url.replace(name+">",interlinks[name])
 					link['href'] = hi
+		if 'img' in content:
+			for img in text.find_all('img', src=re.compile("(.+?)>")):
+				url = img.get('src')
+				m = re.search(r"(.+?)>", url).groups()
+				name = m[0]
+				if name in interlinks:
+					hi = url.replace(name+">",interlinks[name])
+					img['src'] = hi
 
 		instance._content = text.decode()