laser_cal/api_test.py

import sys
import json
import data_clean as dc
import base64
import io
import locale

# 设置 sys.stdout 和 sys.stdin 的编码为 UTF-8
sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8')
sys.stdin = io.TextIOWrapper(sys.stdin.buffer, encoding='utf-8')
locale.getpreferredencoding = lambda: 'UTF-8'


if __name__ == "__main__":
    # **🔹 解析命令行参数**
    if len(sys.argv) < 2:
        print(json.dumps({"error": "No API specified"}))
        sys.exit(1)

    api_name = sys.argv[1]  # 第一个参数是 API 名称

    if api_name == "getpath":
        return_path = str(dc.result_main())
        print(json.dumps({'obj': return_path}, ensure_ascii=False))

    elif api_name == "loaddata":
        if len(sys.argv) < 3:
            print(json.dumps({"error": "No data"}))
            sys.exit(1)
        data = json.loads(base64.b64decode(sys.argv[2]).decode("utf-8"))
        return_list = dc.data_analyse(data)
        print(json.dumps(return_list, ensure_ascii=False))

    else:
        print(json.dumps({"error": "Invalid API"}))