Просмотр исходного кода

Featured images could be used among all kinds of pages.

murchik лет назад: 11
Родитель
Сommit
7817b26ceb
1 измененных файлов с 2 добавлено и 2 удалено
  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 import signals
-from pelican.contents import Content, Article
+from pelican.contents import Article, Draft, Page
 from bs4 import BeautifulSoup
 from bs4 import BeautifulSoup
 
 
 
 
 def images_extraction(instance):
 def images_extraction(instance):
     representativeImage = None
     representativeImage = None
-    if type(instance) == Article:
+    if type(instance) in (Article, Draft, Page):
         if 'image' in instance.metadata:
         if 'image' in instance.metadata:
             representativeImage = instance.metadata['image']
             representativeImage = instance.metadata['image']