Generate QR Codes from URLs in 7 File Formats

import sys
import json
import os
import io
from datetime import datetime

with open(sys.argv[1], encoding="utf-8") as f:
    inputs = json.load(f)

url = inputs["url"]
output_format = inputs["output_format"]
output_dir = inputs["output_dir"]

timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
print(f"Generating QR code for: {url}")
print(f"Output format: {output_format}")

try:
    import qrcode

    def make_qr():
        qr = qrcode.QRCode(
            version=None,
            error_correction=qrcode.constants.ERROR_CORRECT_M,
            box_size=10,
            border=4,
        )
        qr.add_data(url)
        qr.make(fit=True)
        return qr

    def make_png_bytes(qr):
        """Return PNG bytes of the QR code"""
        img = qr.make_image(fill_color="black", back_color="white")
        buf = io.BytesIO()
        img.save(buf, format="PNG")
        buf.seek(0)
        return buf

    if output_format == "jpg":
        filename = f"qrcode_{timestamp}.jpg"
        filepath = os.path.join(output_dir, filename)
        qr = make_qr()
        img = qr.make_image(fill_color="black", back_color="white")
        # JPEG supports only RGB, so convert the image first
        img.convert("RGB").save(filepath, "JPEG", quality=95)

    elif output_format == "png":
        filename = f"qrcode_{timestamp}.png"
        filepath = os.path.join(output_dir, filename)
        qr = make_qr()
        img = qr.make_image(fill_color="black", back_color="white")
        img.save(filepath, "PNG")

    elif output_format == "svg":
        filename = f"qrcode_{timestamp}.svg"
        filepath = os.path.join(output_dir, filename)
        try:
            import qrcode.image.svg
            qr = make_qr()
            img = qr.make_image(image_factory=qrcode.image.svg.SvgPathImage)
            img.save(filepath)
        except ImportError:
            # Fallback: generate PNG when the SVG module is unavailable
            print("Warning: SVG output unavailable. Falling back to PNG.", file=sys.stderr)
            filename = f"qrcode_{timestamp}.png"
            filepath = os.path.join(output_dir, filename)
            qr = make_qr()
            img = qr.make_image(fill_color="black", back_color="white")
            img.save(filepath, "PNG")

    elif output_format == "pdf":
        from reportlab.pdfgen import canvas
        from reportlab.lib.units import mm
        from reportlab.lib.utils import ImageReader

        filename = f"qrcode_{timestamp}.pdf"
        filepath = os.path.join(output_dir, filename)

        qr = make_qr()
        buf = make_png_bytes(qr)

        img_reader = ImageReader(buf)
        qr_size = 80 * mm
        page_w = qr_size + 20 * mm
        page_h = qr_size + 20 * mm

        c = canvas.Canvas(filepath, pagesize=(page_w, page_h))
        c.drawImage(img_reader, 10 * mm, 10 * mm, width=qr_size, height=qr_size)
        c.save()

    elif output_format == "xlsx":
        from openpyxl import Workbook
        from openpyxl.drawing.image import Image as XLImage

        filename = f"qrcode_{timestamp}.xlsx"
        filepath = os.path.join(output_dir, filename)

        qr = make_qr()
        buf = make_png_bytes(qr)

        wb = Workbook()
        ws = wb.active
        ws.title = "QR code"
        ws["A1"] = url

        xl_img = XLImage(buf)
        xl_img.anchor = "A2"
        ws.add_image(xl_img)
        wb.save(filepath)

    elif output_format == "docx":
        from docx import Document
        from docx.shared import Mm

        filename = f"qrcode_{timestamp}.docx"
        filepath = os.path.join(output_dir, filename)

        qr = make_qr()
        buf = make_png_bytes(qr)

        doc = Document()
        doc.add_heading("QR code", level=1)
        doc.add_paragraph(url)
        doc.add_picture(buf, width=Mm(60))
        doc.save(filepath)

    elif output_format == "pptx":
        from pptx import Presentation
        from pptx.util import Inches, Pt
        from pptx.enum.text import PP_ALIGN

        filename = f"qrcode_{timestamp}.pptx"
        filepath = os.path.join(output_dir, filename)

        qr = make_qr()
        buf = make_png_bytes(qr)

        prs = Presentation()
        slide_layout = prs.slide_layouts[6]  # Use the blank layout
        slide = prs.slides.add_slide(slide_layout)

        # Center the QR image on the slide
        qr_size = Inches(3)
        left = (prs.slide_width - qr_size) / 2
        top = (prs.slide_height - qr_size) / 2
        slide.shapes.add_picture(buf, left, top, width=qr_size, height=qr_size)

        # Add the URL as a caption below the QR code
        txBox = slide.shapes.add_textbox(left, top + qr_size, qr_size, Inches(0.4))
        tf = txBox.text_frame
        tf.text = url
        tf.paragraphs[0].alignment = PP_ALIGN.CENTER
        tf.paragraphs[0].runs[0].font.size = Pt(10)

        prs.save(filepath)

    else:
        print(f"Error: unsupported output format: {output_format}", file=sys.stderr)
        sys.exit(1)

    print(f"Saved: {filepath}")

except Exception as e:
    print(f"Error: {e}", file=sys.stderr)
    sys.exit(1)

print("Done")