Increase test coverage

Signed-off-by: DL6ER <[email protected]>

Author: DL6ER

app/labeldesigner/label.py

@@ -392,10 +392,6 @@ def _draw_text(self, img = None, bboxes = [], text_offset = (0, 0)):
         draw = ImageDraw.Draw(img)
         y = 0
 
-        # Fix for completely empty text
-        if len(self.text) == 0 or len(self.text[0]['text']) == 0:
-            self.text[0]['text'] = " "
-
         # Iterate over lines of text
         for i, line in enumerate(self.text):
             color = self._fore_color
@@ -478,6 +474,9 @@ def _draw_text(self, img = None, bboxes = [], text_offset = (0, 0)):
         return bboxes
 
     def _compute_bbox(self, bboxes):
+        # Edge case: No text
+        if not bboxes:
+            return (0, 0, 0, 0)
         # Iterate over right margins of multiple text lines and find the maximum
         # width needed to fit all lines of text
         max_width = max(bbox[0][2] for bbox in bboxes)

app/labeldesigner/routes.py

@@ -67,21 +67,22 @@ def get_barcodes():
     }
 
 
-@bp.route('/api/preview', methods=['POST', 'GET'])
-def get_preview_from_image():
+@bp.route('/api/preview', methods=['POST'])
+def preview_from_image():
     # Set log level if provided
     log_level = request.values.get('log_level')
     if log_level:
         level = getattr(logging, log_level.upper(), None)
         if isinstance(level, int):
             current_app.logger.setLevel(level)
-    label = create_label_from_request(request)
     try:
+        label = create_label_from_request(request)
         im = label.generate(rotate=True)
     except Exception as e:
         current_app.logger.exception(e)
-        # Generate error 400 response
-        return make_response(jsonify({'message': str(e)}), 400)
+        error = 413 if "too long" in str(e) else 400
+        # Generate error response
+        return make_response(jsonify({'message': str(e)}), error)
 
     return_format = request.values.get('return_format', 'png')
 
@@ -198,45 +199,42 @@ def get_label_dimensions(label_size):
             raise LookupError("Unknown label_size")
         return ls['dots_printable']
 
-    def get_font_path(font_family_name, font_style_name):
+    def get_font_path(line: dict):
         try:
-            if font_family_name is None or font_style_name is None:
-                font_family_name = current_app.config['LABEL_DEFAULT_FONT_FAMILY']
-                font_style_name = current_app.config['LABEL_DEFAULT_FONT_STYLE']
+            font_family_name = line.get('font_family')
+            font_style_name = line.get('font_style')
             if font_family_name not in FONTS.fonts:
                 raise LookupError("Unknown font family: %s" % font_family_name)
             if font_style_name not in FONTS.fonts[font_family_name]:
                 font_style_name = current_app.config['LABEL_DEFAULT_FONT_STYLE']
             if font_style_name not in FONTS.fonts[font_family_name]:
-                raise LookupError("Unknown font style: %s for font %s" % (font_style_name, font_family_name))
+                raise LookupError("Unknown font style: %s for font %s" %
+                                  (font_style_name, font_family_name))
             font_path = FONTS.fonts[font_family_name][font_style_name]
         except KeyError:
             raise LookupError("Couln't find the font & style")
         return font_path
 
     def get_uploaded_image(image):
-        try:
-            name, ext = os.path.splitext(image.filename)
-            if ext.lower() in ('.png', '.jpg', '.jpeg'):
-                image = imgfile_to_image(image)
-                if context['image_mode'] == 'grayscale':
-                    return convert_image_to_grayscale(image)
-                elif context['image_mode'] == 'red_and_black':
-                    return convert_image_to_red_and_black(image)
-                elif context['image_mode'] == 'colored':
-                    return image
-                else:
-                    return convert_image_to_bw(image, context['image_bw_threshold'])
-            elif ext.lower() in ('.pdf'):
-                image = pdffile_to_image(image, DEFAULT_DPI)
-                if context['image_mode'] == 'grayscale':
-                    return convert_image_to_grayscale(image)
-                else:
-                    return convert_image_to_bw(image, context['image_bw_threshold'])
+        name, ext = os.path.splitext(image.filename)
+        if ext.lower() in ('.png', '.jpg', '.jpeg'):
+            image = imgfile_to_image(image)
+            if context['image_mode'] == 'grayscale':
+                return convert_image_to_grayscale(image)
+            elif context['image_mode'] == 'red_and_black':
+                return convert_image_to_red_and_black(image)
+            elif context['image_mode'] == 'colored':
+                return image
             else:
-                return None
-        except AttributeError:
-            return None
+                return convert_image_to_bw(image, context['image_bw_threshold'])
+        elif ext.lower() in ('.pdf'):
+            image = pdffile_to_image(image, DEFAULT_DPI)
+            if context['image_mode'] == 'grayscale':
+                return convert_image_to_grayscale(image)
+            else:
+                return convert_image_to_bw(image, context['image_bw_threshold'])
+        else:
+            raise ValueError("Unsupported file type")
 
     if context['print_type'] == 'text':
         label_content = LabelContent.TEXT_ONLY
@@ -271,13 +269,30 @@ def get_uploaded_image(image):
     if label_orientation == LabelOrientation.ROTATED:
         height, width = width, height
 
+    # Fix for completely empty text
+    if len(context['text']) > 0 and len(context['text'][0]['text']) == 0:
+        context['text'][0]['text'] = " "
+
     # For each line in text, we determine and add the font path
     for line in context['text']:
-        line['font_path'] = get_font_path(line['font_family'], line['font_style'])
+        if 'font_family' not in line:
+            line['font_family'] = current_app.config['LABEL_DEFAULT_FONT_FAMILY']
+        if 'font_style' not in line:
+            line['font_style'] = current_app.config['LABEL_DEFAULT_FONT_STYLE']
+        if 'font_size' not in line:
+            raise ValueError("Font size is required")
+        line['font_path'] = get_font_path(line)
+
+        # Reject extraordinary long texts
+        if len(line['text']) > 10_000:
+            raise ValueError("Text is too long")
 
     fore_color = (255, 0, 0) if context['print_color'] == 'red' else (0, 0, 0)
     border_color = (255, 0, 0) if context['border_color'] == 'red' else (0, 0, 0)
 
+    uploaded = request.files.get('image', None)
+    image = get_uploaded_image(uploaded) if uploaded is not None else None
+
     return SimpleLabel(
         width=width,
         height=height,
@@ -295,7 +310,7 @@ def get_uploaded_image(image):
         barcode_type=context['barcode_type'],
         qr_size=context['qrcode_size'],
         qr_correction=context['qrcode_correction'],
-        image=get_uploaded_image(request.files.get('image', None)),
+        image=image,
         image_fit=context['image_fit'],
         border_thickness=context['border_thickness'],
         border_roundness=context['border_roundness'],

app/labeldesigner/templates/labeldesigner.html

@@ -391,8 +391,8 @@
 {% endblock %}
 
 {% block javascript %}
-const url_for_print_label = '{{ url_for('.print_label') }}';
-const url_for_get_preview = '{{ url_for('.get_preview_from_image') }}';
+const url_for_print = '{{ url_for('.print_label') }}';
+const url_for_preview = '{{ url_for('.preview_from_image') }}';
 const url_for_get_font_styles = '{{url_for('.get_font_styles')}}';
 const url_for_get_barcodes = '{{ url_for('.get_barcodes') }}';
 const url_for_get_printer_status = '{{ url_for('.get_printer_status') }}';

app/static/js/main.js

@@ -245,7 +245,7 @@ function gen_label(preview = true, cut_once = false) {
     }
 
     // Send printing request
-    const url = preview ? (url_for_get_preview + '?return_format=base64') : url_for_print_label;
+    const url = preview ? (url_for_preview + '?return_format=base64') : url_for_print;
     $.ajax({
         type: 'POST',
         url: url,
@@ -307,9 +307,9 @@ var imageDropZone;
 Dropzone.options.myAwesomeDropzone = {
     url: function () {
         if (dropZoneMode == 'preview') {
-            return url_for_get_preview + "?return_format=base64";
+            return url_for_preview + "?return_format=base64";
         } else {
-            return url_for_print_label;
+            return url_for_print;
         }
     },
     paramName: "image",