diff --git a/uberwriter/export_dialog.py b/uberwriter/export_dialog.py index 7b9cbd4..6ecd31b 100644 --- a/uberwriter/export_dialog.py +++ b/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" """