Kaynağa Gözat

Merge pull request #614 from trovao/pdf

Make pdf-related settings not required
Justin Mayer 9 yıl önce
ebeveyn
işleme
b565aa4067
1 değiştirilmiş dosya ile 13 ekleme ve 5 silme
  1. 13 5
      pdf/pdf.py

+ 13 - 5
pdf/pdf.py

@@ -23,12 +23,20 @@ class PdfGenerator(Generator):
     rst"""
     def __init__(self, *args, **kwargs):
         super(PdfGenerator, self).__init__(*args, **kwargs)
-        
-        pdf_style_path = os.path.join(self.settings['PDF_STYLE_PATH'])
-        pdf_style = self.settings['PDF_STYLE']
+
+        if 'PDF_STYLE_PATH' in self.settings:
+            pdf_style_path = [self.settings['PDF_STYLE_PATH']]
+        else:
+            pdf_style_path = []
+
+        if 'PDF_STYLE' in self.settings:
+            pdf_style = [self.settings['PDF_STYLE']]
+        else:
+            pdf_style = []
+
         self.pdfcreator = RstToPdf(breakside=0,
-                                   stylesheets=[pdf_style],
-                                   style_path=[pdf_style_path])
+                                   stylesheets=pdf_style,
+                                   style_path=pdf_style_path)
 
     def _create_pdf(self, obj, output_path):
         if obj.source_path.endswith('.rst'):