使用 Python 和 OpenCV 優化 QR 碼掃描以實現無縫重定向

2024-03-13 23:26:44

优化 QR 码扫描儀以實現成功掃描後的重定向

問題陳述

在開發使用 OpenCV、Flask 和 Python 的 QR 碼掃描儀時，我遇到了一個問題。成功掃描和解碼 QR 碼後，影片會停止或暫停，但不會進行重定向。我希望避免使用 JavaScript，除非它是一個小腳本，有助於重定向。我曾嘗試在 HTML 中嵌入 JavaScript 函式，但沒有成功。它會回調到另一個端點，該端點會在成功讀取 QR 碼時使用 JSON 回應。當这种情况發生時，並没有錯誤，瀏覽器只是暫停。

解決方法

要解決這個問題，我們需要：

在 QR 碼成功解碼後，手動關閉視窗和相機。
將 DECODED_QR_CODE 全域變數更新為解碼的資料。
使用 redirect() 方法重定向到首頁。

改進後的程式碼

import cv2
from flask import Flask, render_template, redirect, Response
from pyzbar.pyzbar import decode

app = Flask(__name__, template_folder='templates')

DECODED_QR_CODE = "PLACEHOLDER"

@app.route('/')
def index():
    global DECODED_QR_CODE
    return f"""<!DOCTYPE html>
    <html lang="en">
    <head>
        <meta charset="UTF-8">
        <meta name="viewport" content="width=device-width, initial-scale=1.0">
        
    </head>
    <body>
        <h1>{ DECODED_QR_CODE }</h1>
        <a href="/scanner">
        <button type="button">Go to QR Code Scanner</button>
        </a>
    </body>
    </html>"""

@app.route('/scanner')
def second():
    return render_template('scan.html')

@app.route('/video_feed')
def video_feed():
    return Response(gen_frames(), mimetype='multipart/x-mixed-replace; boundary=frame')

def gen_frames():
    camera = cv2.VideoCapture(0)  # Use 0 for the default camera
    global DECODED_QR_CODE
    while True:
        success, frame = camera.read()  # Read the camera frame
        if not success:
            break
        else:
            # Detect and decode QR codes in the frame
            decoded_objects = decode(frame)
            for obj in decoded_objects:
                # Print decoded data
                print(obj.data.decode('utf-8'))
                # if valid QR code detected, redirect to the index page
                if obj.data.decode('utf-8'):
                    DECODED_QR_CODE = obj.data.decode('utf-8')
                    cv2.destroyAllWindows()
                    camera.release()
                    return redirect("/")
                
            ret, buffer = cv2.imencode('.jpg', frame)
            frame = buffer.tobytes()
            yield (b'--frame\r\n'
                   b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')  # Concat frame one by one and show result

if __name__ == "__main__":
    app.run(host='0.0.0.0', debug=True)