[Feature] make number of prints configurable

photobooth/defaults.cfg

@@ -13,7 +13,7 @@ hide_cursor = False
 style = default
 
 [Camera]
-# Camera module to use (python-gphoto2, gphoto2-cffi, gphoto2-commandline, 
+# Camera module to use (python-gphoto2, gphoto2-cffi, gphoto2-commandline,
 # opencv, picamera, dummy)
 module = python-gphoto2
 # Specify rotation of camera in degree (possible values: 0, 90, 180, 270)
@@ -48,6 +48,8 @@ module = PyQt5
 width = 148
 # Paper height in mm
 height = 100
+# Number of prints from one image
+num_prints = 1
 
 [Photobooth]
 # Show preview while posing time (True/False)

photobooth/gui/GuiPostprocessor.py

@@ -35,12 +35,13 @@ def __init__(self, config):
             paper_size = (config.getInt('Printer', 'width'),
                           config.getInt('Printer', 'height'))
             pdf = config.getBool('Printer', 'pdf')
+            num_prints = config.getInt('Printer', 'num_prints')
             if config.getBool('Printer', 'confirmation'):
                 self._get_task_list.append(
-                    PrintPostprocess(module, paper_size, pdf))
+                    PrintPostprocess(module, paper_size, num_prints, pdf))
             else:
                 self._do_task_list.append(
-                    PrintPostprocess(module, paper_size, pdf))
+                    PrintPostprocess(module, paper_size, num_prints, pdf))
 
     def get(self, picture):
 
@@ -100,7 +101,7 @@ def action(self, action):
 
 class PrintPostprocess(PostprocessTask):
 
-    def __init__(self, printer_module, paper_size, is_pdf, **kwargs):
+    def __init__(self, printer_module, paper_size, is_pdf, num_prints, **kwargs):
 
         super().__init__(**kwargs)
 

photobooth/gui/Qt5Gui/Frames.py

@@ -856,12 +856,18 @@ def createPrinterSettings(self):
         lay_size.addWidget(QtWidgets.QLabel('x'))
         lay_size.addWidget(height)
 
+        num_prints = QtWidgets.QSpinBox()
+        num_prints.setRange(1, 99)
+        num_prints.setValue(self._cfg.getInt('Printer', 'num_prints'))
+        self.add('Printer', 'num_prints', num_prints)
+
         layout = QtWidgets.QFormLayout()
         layout.addRow(_('Enable printing:'), enable)
         layout.addRow(_('Module:'), module)
         layout.addRow(_('Print to PDF (for debugging):'), pdf)
         layout.addRow(_('Ask for confirmation before printing:'), confirmation)
         layout.addRow(_('Paper size [mm]:'), lay_size)
+        layout.addRow(_('Number of prints from one image:'), num_prints)
 
         widget = QtWidgets.QWidget()
         widget.setLayout(layout)
@@ -1048,6 +1054,8 @@ def storeConfigAndRestart(self):
         self._cfg.set('Printer', 'width', self.get('Printer', 'width').text())
         self._cfg.set('Printer', 'height',
                       self.get('Printer', 'height').text())
+        self._cfg.set('Printer', 'num_prints',
+                      self.get('Printer', 'num_prints').text())
 
         self._cfg.set('Mailer', 'enable',
                       str(self.get('Mailer', 'enable').isChecked()))

photobooth/printer/PrinterPyCups.py

@@ -37,7 +37,9 @@
 
 class PrinterPyCups(Printer):
 
-    def __init__(self, page_size, print_pdf=False):
+    def __init__(self, page_size, num_prints, print_pdf=False):
+
+        self.num_prints = num_prints
 
         self._conn = cups.Connection() if cups else None
 
@@ -57,10 +59,12 @@ def __init__(self, page_size, print_pdf=False):
 
     def print(self, picture):
 
-        if self._conn is not None:
-            if isinstance(picture, ImageQt.ImageQt):
-                picture.save(self._tmp_filename)
-            else:
-                picture.save(self._tmp_filename, format='JPEG')
-            self._conn.printFile(self._printer, self._tmp_filename,
-                                 "photobooth", {})
+        for n in range(num_prints):
+            logging.info('Printing picture, {} of {}'.format(n + 1, self.num_prints))
+            if self._conn is not None:
+                if isinstance(picture, ImageQt.ImageQt):
+                    picture.save(self._tmp_filename)
+                else:
+                    picture.save(self._tmp_filename, format='JPEG')
+                self._conn.printFile(self._printer, self._tmp_filename,
+                                    "photobooth", {})

photobooth/printer/PrinterPyQt5.py

@@ -27,9 +27,11 @@
 
 class PrinterPyQt5(Printer):
 
-    def __init__(self, page_size, print_pdf=False):
+    def __init__(self, page_size, num_prints, print_pdf=False):
 
-        super().__init__(page_size)
+        super().__init__(page_size, num_prints)
+
+        self.num_prints = num_prints
 
         self._printer = QPrinter(QPrinter.HighResolution)
         self._printer.setFullPage(True)
@@ -47,15 +49,16 @@ def __init__(self, page_size, print_pdf=False):
 
     def print(self, picture):
 
-        if self._print_pdf:
-            self._printer.setOutputFileName('print_%d.pdf' % self._counter)
-            self._counter += 1
+        for n in range(self.num_prints):
+            if self._print_pdf:
+                self._printer.setOutputFileName('print_%d.pdf' % self._counter)
+                self._counter += 1
 
-        logging.info('Printing picture')
-        logging.debug('Page Size: {}, Print Size: {}, PictureSize: {} '.format(
-            self._printer.paperRect(), self._printer.pageRect(),
-            picture.rect()))
+            logging.info('Printing picture, {} of {}'.format(n + 1, self.num_prints))
+            logging.debug('Page Size: {}, Print Size: {}, PictureSize: {} '.format(
+                self._printer.paperRect(), self._printer.pageRect(),
+                picture.rect()))
 
-        painter = QtGui.QPainter(self._printer)
-        painter.drawImage(self._printer.pageRect(), picture, picture.rect())
-        painter.end()
+            painter = QtGui.QPainter(self._printer)
+            painter.drawImage(self._printer.pageRect(), picture, picture.rect())
+            painter.end()

photobooth/printer/__init__.py

@@ -25,9 +25,10 @@
 
 class Printer:
 
-    def __init__(self, page_size):
+    def __init__(self, page_size, num_prints):
 
         self.pageSize = page_size
+        self.num_prints = num_prints
 
     @property
     def pageSize(self):