From 0ea2874f887b05ffec4e4dcd110bfadaad39b11f Mon Sep 17 00:00:00 2001
From: liyulingyue <852433440@qq.com>
Date: Thu, 16 Jan 2025 07:05:23 +0800
Subject: [PATCH] auto opencc

---
 setup.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index b15dedbac..2d38426e9 100644
--- a/setup.py
+++ b/setup.py
@@ -31,6 +31,25 @@ HERE = Path(os.path.abspath(os.path.dirname(__file__)))
 VERSION = '0.0.0'
 COMMITID = 'none'
 
+
+def determine_opencc_version():
+    # get gcc version
+    gcc_version = None
+    try:
+        output = sp.check_output(
+            ['gcc', '--version'], stderr=sp.STDOUT, text=True)
+        for line in output.splitlines():
+            if "gcc" in line:
+                gcc_version = line.split()[-1]
+    except Exception as e:
+        gcc_version = None
+
+    # determine opencc version
+    if gcc_version and gcc_version.startswith("9."):
+        return "opencc==1.1.6"  # GCC 9 need opencc==1.1.6
+    return "opencc"  # default
+
+
 base = [
     "braceexpand",
     "editdistance",
@@ -48,7 +67,7 @@ base = [
     "matplotlib<=3.8.4",
     "nara_wpe",
     "onnxruntime>=1.11.0",
-    "opencc==1.1.6",
+    determine_opencc_version(),  # opencc or opencc==1.1.6
     "opencc-python-reimplemented",
     "pandas",
     "paddleaudio>=1.1.0",