package com.yang.springbootlucene.controller;

import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.core.io.ClassPathResource;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

/**
 * @author:yjx
 * @description:
 * @date:2019/11/28 15:29
 */
@Slf4j
@RestController
@RequestMapping("/user")
public class UserController {

  @Value("${pyzbar.script}")
  private String scriptPath;

  private static final String BASE_64_PREFIX = "data:image/png;base64,";

  /**
   * 這種方式在源碼中成功調(diào)用，但達(dá)成jar包后，找不到腳本路徑，所以腳本必須放在外面
   */
  @Deprecated
  private static String pyZbarScriptPath;
  /* static {
    ClassPathResource pathResource = new ClassPathResource("/script/my_py_zbar.py");
    try {
      pyZbarScriptPath = pathResource.getFile().getAbsolutePath();
    } catch (IOException e) {
      e.printStackTrace();
    }
  } */

  @RequestMapping("/cameraScanPyZbar")
  public Object cameraScanPyZbar(@RequestBody String base64) throws IOException {
    if (this.checkBase64Head(base64)) {
      //1.去掉base64字符的頭部
      String base64Str = this.cutHead(base64);

      //2.創(chuàng)建臨時(shí)文件（由于圖片的base64字符太長(zhǎng)，不支持直接以命令參數(shù)的形式傳遞，故將字符串寫入臨時(shí)文件，而后python程序讀取臨時(shí)文件內(nèi)容）
      String tempPath = "./" + Thread.currentThread().getName();
      File tempFile = new File(tempPath);
      FileWriter fileWriter = new FileWriter(tempFile, false);
      fileWriter.write(base64Str);
      fileWriter.flush();
      fileWriter.close();

      //3.調(diào)用pyzbar解析base64字符串
      String plainText = PyZbarUtil.executePyzbar("python", scriptPath, tempFile.getAbsolutePath());

      //4.刪除臨時(shí)文件
      tempFile.delete();
      System.err.println("--------->" + plainText);
      return plainText;
    } else {
      return "參數(shù)格式錯(cuò)誤";
    }
  }

  /**
   * 校驗(yàn)Base64值是否已規(guī)定的串開始
   *
   * @param base64
   * @return
   */
  private boolean checkBase64Head(String base64) {
    return base64.startsWith(BASE_64_PREFIX);
  }

  /**
   * Base64去頭
   *
   * @param base64
   * @return
   */
  private String cutHead(String base64) {
    return base64.replaceAll(BASE_64_PREFIX, "");
  }
}

package com.yang.springbootlucene.controller;

import java.io.BufferedReader;
import java.io.InputStreamReader;

public final class PyZbarUtil {

  /**
   * 腳本執(zhí)行工具類
   *
   * @param lang    命令語(yǔ)言
   * @param scriptPath 腳本絕對(duì)路勁
   * @param base64Path   base64文件絕對(duì)路徑
   * @return
   */
  public static String executePyzbar(String lang, String scriptPath, String base64Path) {
    String[] arguments = new String[]{lang, scriptPath, base64Path};
    try {
      Process process = Runtime.getRuntime().exec(arguments);
      int re = process.waitFor();
      if (re == 0) {
        try (BufferedReader in = new BufferedReader(new InputStreamReader(process.getInputStream(), "GBK"));) {
          return in.readLine();
        }
      } else {
        System.err.println("腳本調(diào)用失敗");
      }
    } catch (Exception e) {
      e.printStackTrace();
    }
    return null;
  }
}

# -*-coding:UTF-8-*-

import sys
import base64
from io import BytesIO
import pyzbar.pyzbar as pyzbar
from PIL import Image,ImageEnhance


'''將base64轉(zhuǎn)換為字節(jié)流'''
def convert_base64_to_byte_stream(base64_str):
  # 1.解碼Base64字符串
  img_data = base64.b64decode(base64_str)

  # 2.將節(jié)碼結(jié)果轉(zhuǎn)為字節(jié)流
  byte_stream = BytesIO(img_data)
  return byte_stream


'''從將字節(jié)流解析二維碼'''
def parse_byte_stream_qr_code(byte_stream):
  # 3.打開字節(jié)流得到圖片對(duì)象
  img = Image.open(byte_stream)

  img = ImageEnhance.Brightness(img).enhance(2.0) #增加亮度
  img = ImageEnhance.Contrast(img).enhance(4.0) #增加對(duì)比度
  # img = ImageEnhance.Sharpness(img).enhance(17.0) #銳利化
  # img = img.convert('L') #灰度化

  # img.show() # 播放圖片，供測(cè)試用

  # 4.調(diào)用pyzbar解析圖片中的二維碼
  barcodes = pyzbar.decode(img)

  # 5.打印解析結(jié)果
  return barcodes[0].data.decode("utf-8")


def main(argv):
#  print(parse_byte_stream_qr_code(convert_base64_to_byte_stream(argv[1])))
  print(parse_byte_stream_qr_code(convert_base64_to_byte_stream(open(argv[1]).readline())))


if __name__ == "__main__":
  main(sys.argv)