Browse Source

Featured images could be used among all kinds of pages.

murchik 10 years ago
parent
commit
7817b26ceb
1 changed files with 2 additions and 2 deletions
  1. 2 2
      representative_image/representative_image.py

+ 2 - 2
representative_image/representative_image.py

@@ -1,11 +1,11 @@
 from pelican import signals
-from pelican.contents import Content, Article
+from pelican.contents import Article, Draft, Page
 from bs4 import BeautifulSoup
 
 
 def images_extraction(instance):
     representativeImage = None
-    if type(instance) == Article:
+    if type(instance) in (Article, Draft, Page):
         if 'image' in instance.metadata:
             representativeImage = instance.metadata['image']