Change paper size of exported PDF according to what's been chosen

uberwriter/export_dialog.py

@@ -155,6 +155,7 @@ class Export:
         self.dialog = self.builder.get_object("Export")
         self.stack = self.builder.get_object("export_stack")
         self.stack_switcher = self.builder.get_object("format_switcher")
+        self.paper_size = self.builder.get_object("combobox_paper_size")
 
         stack_pdf_disabled = self.builder.get_object("pdf_disabled")
         filename = filename or _("Untitled document.md")
@@ -205,9 +206,7 @@ class Export:
         """
 
         export_type = self.stack.get_visible_child_name()
-        args = [
-            "--variable=papersize:a4"
-        ]
+        args = []
         if export_type == "advanced":
             filename = self.adv_export_name.get_text()
             output_dir = os.path.abspath(self.filechoosers["advanced"].get_current_folder())
@@ -225,6 +224,9 @@ class Export:
             args.extend(self.get_advanced_arguments())
 
         else:
+            args = [
+                "--variable=papersize:a4"
+            ]
             filename = self.filechoosers[export_type].get_filename()
             if filename.endswith("." + export_type):
                 filename = filename[:-len(export_type)-1]
@@ -262,6 +264,11 @@ class Export:
         highlight_style = self.builder.get_object("highlight_style").get_active_text()
 
         conditions = [
+            {
+                "condition": True,
+                "yes": "--variable=papersize:" + self.get_paper_size(),
+                "no": None
+            },
             {
                 "condition": self.builder.get_object("toc").get_active(),
                 "yes": "--toc",
@@ -319,6 +326,16 @@ class Export:
 
         return args
 
+    def get_paper_size(self):
+        paper_size = self.paper_size.get_active_text()
+
+        paper_formats = {
+            "A4": "a4",
+            "US Letter": "letter"
+        }
+
+        return paper_formats[paper_size]
+
     def allow_export(self, widget, data, signal):
         """Disable export button if the visible child is "pdf_disabled"
         """