Explorar o código

Featured images could be used among all kinds of pages.

murchik %!s(int64=9) %!d(string=hai) anos
pai
achega
7817b26ceb
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  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']