This is an automated email from the ASF dual-hosted git repository.

colinlee pushed a commit to branch fix_release_for_python_pypi_for_read_tree
in repository https://gitbox.apache.org/repos/asf/tsfile.git

commit 0ee0fa0458e327de6af51af710387a28eb20018f
Author: ColinLee <[email protected]>
AuthorDate: Mon Oct 20 15:12:51 2025 +0800

    Add workflow for python release.
---
 .github/workflows/unit-test-cpp.yml          |   8 +-
 .github/workflows/unit-test-python.yml       |  11 ++
 .github/workflows/wheels.yml                 | 169 ++++++++++++++++++++
 cpp/pom.xml                                  |  33 ----
 cpp/src/CMakeLists.txt                       |   1 +
 cpp/test/CMakeLists.txt                      |   1 +
 pom.xml                                      |   2 +
 python/how_to_build_and_release_py_tsfile.md |  28 ++++
 python/pom.xml                               |  18 +++
 python/pyproject.toml                        |  69 +++++++++
 python/requirements.txt                      |   4 +-
 python/setup.py                              | 221 ++++++++++++---------------
 python/tsfile/__init__.py                    |   4 -
 python/tsfile/tsfile_cpp.pxd                 |   6 +-
 14 files changed, 406 insertions(+), 169 deletions(-)

diff --git a/.github/workflows/unit-test-cpp.yml 
b/.github/workflows/unit-test-cpp.yml
index 08c13801..429e0b95 100644
--- a/.github/workflows/unit-test-cpp.yml
+++ b/.github/workflows/unit-test-cpp.yml
@@ -104,10 +104,16 @@ jobs:
               core.setOutput('platform_suffix', ``)
             }
 
-      - name: Install clang-format
+      - name: Install dependencies
         shell: bash
         run: |
           if [[ "$RUNNER_OS" == "Linux" ]]; then
+            if command -v apt-get >/dev/null 2>&1; then
+              sudo apt-get update
+              sudo apt-get install -y uuid-dev
+            elif command -v yum >/dev/null 2>&1; then
+              sudo yum install -y libuuid-devel
+            fi
             sudo update-alternatives --install /usr/bin/clang-format 
clang-format /usr/bin/clang-format-17 100
             sudo update-alternatives --set clang-format 
/usr/bin/clang-format-17
           elif [[ "$RUNNER_OS" == "Windows" ]]; then
diff --git a/.github/workflows/unit-test-python.yml 
b/.github/workflows/unit-test-python.yml
index c9993597..dd16ae3e 100644
--- a/.github/workflows/unit-test-python.yml
+++ b/.github/workflows/unit-test-python.yml
@@ -60,6 +60,17 @@ jobs:
           key: ${{ runner.os }}-m2-${{ hashFiles('**/pom.xml') }}
           restore-keys: ${{ runner.os }}-m2-
 
+      - name: Install dependencies
+        shell: bash
+        run: |
+          if [[ "$RUNNER_OS" == "Linux" ]]; then
+            if command -v apt-get >/dev/null 2>&1; then
+              sudo apt-get update
+              sudo apt-get install -y uuid-dev
+            elif command -v yum >/dev/null 2>&1; then
+              sudo yum install -y libuuid-devel
+            fi
+          fi
       # On Windows systems the 'mvnw' script needs an additional ".cmd" 
appended.
       - name: Calculate platform suffix
         id: platform_suffix
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
new file mode 100644
index 00000000..6e9dacec
--- /dev/null
+++ b/.github/workflows/wheels.yml
@@ -0,0 +1,169 @@
+name: Build TsFile wheels(multi-platform)
+
+on:
+  push:
+    branches:
+      - "release_v*.*.*"
+  pull_request:
+    paths:
+      - "cpp/**"
+      - "python/**"
+      - ".github/**"
+  workflow_dispatch:
+
+jobs:
+  build:
+    name: Build wheels on ${{ matrix.name }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        include:
+          - name: linux-x86_64
+            os: ubuntu-22.04
+            platform: linux
+            cibw_archs_linux: "x86_64"
+
+          - name: linux-aarch64
+            os: ubuntu-22.04-arm
+            platform: linux
+            cibw_archs_linux: "aarch64"
+
+          - name: macos-x86_64
+            os: macos-13
+            platform: macos
+            cibw_archs_macos: "x86_64"
+
+          - name: macos-arm64
+            os: macos-14
+            platform: macos
+            cibw_archs_macos: "arm64"
+#          currently, compile on windows is not supported for cibuildwheel
+#          - name: windows-amd64
+#            os: windows-2022
+#            platform: windows
+#            cibw_archs_windows: "AMD64"
+    
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v4
+        with:
+          submodules: false
+          fetch-depth: 0
+
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.11"
+      
+      - name: Set up Java 17
+        uses: actions/setup-java@v4
+        with:
+          distribution: temurin
+          java-version: "17"
+
+      - name: Install system deps (macOS)
+        if: matrix.platform == 'macos'
+        run: |
+          set -eux
+          brew update
+          brew install pkg-config || true
+
+      - name: Install build tools
+        run: |
+          python -m pip install -U pip wheel
+          python -m pip install cibuildwheel==2.21.3
+
+#      - name: Build C++ core via Maven(win)
+#        if: matrix.platform == 'windows'
+#        shell: bash
+#        run: |
+#          set -euxo pipefail
+#          chmod +x mvnw || true
+#          ./mvnw -Pwith-cpp clean verify package \
+#            -DskipTests -Dspotless.check.skip=true -Dspotless.apply.skip=true
+#          test -d cpp/target/build/lib
+#          test -d cpp/target/build/include
+      
+      - name: Build C++ core via Maven (macOS)
+        if: matrix.platform == 'macos'
+        shell: bash
+        env:
+          MACOSX_DEPLOYMENT_TARGET: "12.0"
+          CFLAGS: "-mmacosx-version-min=12.0"
+          CXXFLAGS: "-mmacosx-version-min=12.0"
+          LDFLAGS: "-mmacosx-version-min=12.0"
+        run: |
+          set -euxo pipefail
+          chmod +x mvnw || true
+          ./mvnw -Pwith-cpp clean verify package \
+            -DskipTests -Dspotless.check.skip=true -Dspotless.apply.skip=true \
+            -Dcmake.args="-DCMAKE_OSX_DEPLOYMENT_TARGET=12.0"
+          otool -l cpp/target/build/lib/libtsfile*.dylib | grep -A2 
LC_VERSION_MIN_MACOSX || true
+            
+      - name: Build wheels via cibuildwheel
+        if: matrix.platform != 'macos'
+        env: 
+          CIBW_ARCHS_LINUX: ${{ matrix.cibw_archs_linux }}
+#          CIBW_ARCHS_WINDOWS: ${{ matrix.cibw_archs_windows }}
+
+          CIBW_BUILD: "cp39-* cp310-* cp311-* cp312-* cp313-* cp314-*"
+          CIBW_SKIP: "pp* *-musllinux*"
+
+          CIBW_MANYLINUX_X86_64_IMAGE: "manylinux2014"
+          CIBW_MANYLINUX_AARCH64_IMAGE: "manylinux2014"
+
+          MACOSX_DEPLOYMENT_TARGET: "12.0"
+
+          CIBW_BEFORE_ALL_LINUX: |
+            set -euxo pipefail
+            if command -v yum >/dev/null 2>&1; then
+              yum install -y wget tar gzip pkgconfig libuuid-devel 
libblkid-devel
+            else
+              echo "Not a yum-based image?" ; exit 1
+            fi
+            ARCH="$(uname -m)"
+            mkdir -p /opt/java
+            if [ "$ARCH" = "x86_64" ]; then
+              
JDK_URL="https://download.oracle.com/java/17/archive/jdk-17.0.12_linux-x64_bin.tar.gz";
+            else
+              # aarch64
+              
JDK_URL="https://download.oracle.com/java/17/archive/jdk-17.0.12_linux-aarch64_bin.tar.gz";
+            fi
+            curl -L -o /tmp/jdk17.tar.gz "$JDK_URL"
+            tar -xzf /tmp/jdk17.tar.gz -C /opt/java
+            export JAVA_HOME=$(echo /opt/java/jdk-17.0.12*)
+            export PATH="$JAVA_HOME/bin:$PATH"
+            java -version
+
+            chmod +x mvnw || true
+            ./mvnw -Pwith-cpp clean verify package \
+              -DskipTests -Dspotless.check.skip=true -Dspotless.apply.skip=true
+            test -d cpp/target/build/lib && test -d cpp/target/build/include
+                
+          CIBW_TEST_COMMAND: >
+            python -c "import tsfile, tsfile.tsfile_reader as r; 
print('import-ok:')"
+          CIBW_BUILD_VERBOSITY: "1"
+        run: cibuildwheel --output-dir wheelhouse python
+      
+      - name: Build wheels via cibuildwheel (macOS)
+        if: matrix.platform == 'macos'
+        env:
+          CIBW_ARCHS_MACOS: ${{ matrix.cibw_archs_macos }}
+          CIBW_BUILD: "cp39-* cp310-* cp311-* cp312-* cp313-* cp314-*"
+#          CIBW_BUILD: "cp313-*"
+          CIBW_SKIP: "pp*"
+          CIBW_ENVIRONMENT_MACOS: "MACOSX_DEPLOYMENT_TARGET=12.0"
+          MACOSX_DEPLOYMENT_TARGET: "12.0"
+          CIBW_TEST_COMMAND: >
+            python -c "import tsfile, tsfile.tsfile_reader as r; 
print('import-ok:')"
+          CIBW_BUILD_VERBOSITY: "1"
+        run: cibuildwheel --output-dir wheelhouse python
+
+      - name: Upload wheels as artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: tsfile-wheels-${{ matrix.name }}
+          path: wheelhouse/*.whl
+    
+          
diff --git a/cpp/pom.xml b/cpp/pom.xml
index fa6b0070..572b8cd2 100644
--- a/cpp/pom.xml
+++ b/cpp/pom.xml
@@ -151,39 +151,6 @@
         </plugins>
     </build>
     <profiles>
-        <profile>
-            <id>linux-install-uuid-dev</id>
-            <activation>
-                <os>
-                    <family>unix</family>
-                    <name>Linux</name>
-                </os>
-            </activation>
-            <build>
-                <plugins>
-                    <plugin>
-                        <groupId>org.codehaus.mojo</groupId>
-                        <artifactId>exec-maven-plugin</artifactId>
-                        <executions>
-                            <execution>
-                                <id>install-uuid-dev</id>
-                                <phase>validate</phase>
-                                <goals>
-                                    <goal>exec</goal>
-                                </goals>
-                            </execution>
-                        </executions>
-                        <configuration>
-                            <executable>bash</executable>
-                            <arguments>
-                                <argument>-c</argument>
-                                <argument>sudo apt-get update &amp;&amp; sudo 
apt-get install -y uuid-dev</argument>
-                            </arguments>
-                        </configuration>
-                    </plugin>
-                </plugins>
-            </build>
-        </profile>
         <!-- When running on jenkins, download the sonar build-wrapper, so we 
can do a code analysis -->
         <profile>
             <id>jenkins-build</id>
diff --git a/cpp/src/CMakeLists.txt b/cpp/src/CMakeLists.txt
index 86c67f2b..3d2d292f 100644
--- a/cpp/src/CMakeLists.txt
+++ b/cpp/src/CMakeLists.txt
@@ -36,6 +36,7 @@ set(ANTLR4_WITH_STATIC_CRT OFF)
 set(PROJECT_INCLUDE_DIR
         ${CMAKE_SOURCE_DIR}/src
         ${THIRD_PARTY_INCLUDE}/google_snappy
+        ${THIRD_PARTY_INCLUDE}/zlib-1.2.13
         ${CMAKE_SOURCE_DIR}/third_party/lz4
         ${CMAKE_SOURCE_DIR}/third_party/lzokay
         ${CMAKE_SOURCE_DIR}/third_party/zlib-1.2.13
diff --git a/cpp/test/CMakeLists.txt b/cpp/test/CMakeLists.txt
index 1beb310d..c4886641 100644
--- a/cpp/test/CMakeLists.txt
+++ b/cpp/test/CMakeLists.txt
@@ -80,6 +80,7 @@ include_directories(
         ${LIBRARY_INCLUDE_DIR}
         ${THIRD_PARTY_INCLUDE}
         ${THIRD_PARTY_INCLUDE}/google_snappy
+        ${THIRD_PARTY_INCLUDE}/zlib-1.2.13
         ${CMAKE_SOURCE_DIR}/third_party/lz4
         ${CMAKE_SOURCE_DIR}/third_party/google_snappy
         ${CMAKE_SOURCE_DIR}/third_party/lzokay
diff --git a/pom.xml b/pom.xml
index 51fbe314..dbd4143b 100644
--- a/pom.xml
+++ b/pom.xml
@@ -135,6 +135,8 @@
                             <exclude>**/tsfile.egg-info/**</exclude>
                             <!-- Exclude third_party-->
                             <exclude>**/third_party/**</exclude>
+                            <exclude>**/.python-version</exclude>
+                            <exclude>**/**venv-py**/**</exclude>
                         </excludes>
                     </configuration>
                 </plugin>
diff --git a/python/how_to_build_and_release_py_tsfile.md 
b/python/how_to_build_and_release_py_tsfile.md
new file mode 100644
index 00000000..ea7cea42
--- /dev/null
+++ b/python/how_to_build_and_release_py_tsfile.md
@@ -0,0 +1,28 @@
+<!--
+
+    Licensed to the Apache Software Foundation (ASF) under one
+    or more contributor license agreements.  See the NOTICE file
+    distributed with this work for additional information
+    regarding copyright ownership.  The ASF licenses this file
+    to you under the Apache License, Version 2.0 (the
+    "License"); you may not use this file except in compliance
+    with the License.  You may obtain a copy of the License at
+
+        http://www.apache.org/licenses/LICENSE-2.0
+
+    Unless required by applicable law or agreed to in writing,
+    software distributed under the License is distributed on an
+    "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+    KIND, either express or implied.  See the License for the
+    specific language governing permissions and limitations
+    under the License.
+
+-->
+
+## Build on local 
+
+## Build on docker
+
+## Build for release
+
+## Release TsFile to Pypi and Conda
\ No newline at end of file
diff --git a/python/pom.xml b/python/pom.xml
index 765084f7..8474470b 100644
--- a/python/pom.xml
+++ b/python/pom.xml
@@ -90,6 +90,24 @@
                             </arguments>
                         </configuration>
                     </execution>
+                    <execution>
+                        <id>python-install-wheel</id>
+                        <phase>validate</phase>
+                        <goals>
+                            <goal>exec</goal>
+                        </goals>
+                        <configuration>
+                            
<executable>${python.venv.bin}${python.exe.bin}</executable>
+                            <arguments>
+                                <argument>-m</argument>
+                                <argument>pip</argument>
+                                <argument>install</argument>
+                                <argument>--only-binary=:all:</argument>
+                                <argument>--no-deps</argument>
+                                
<argument>${project.basedir}/dist/tsfile-*.whl</argument>
+                            </arguments>
+                        </configuration>
+                    </execution>
                     <execution>
                         <id>compile-python-code</id>
                         <phase>compile</phase>
diff --git a/python/pyproject.toml b/python/pyproject.toml
new file mode 100644
index 00000000..cb8d098d
--- /dev/null
+++ b/python/pyproject.toml
@@ -0,0 +1,69 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+#
+
+[build-system]
+requires = [
+    "setuptools>=69",
+    "wheel", 
+    "Cython>=3", 
+    "numpy>=1.20"
+    ]
+
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "tsfile"
+version = "2.1.2"
+requires-python = ">=3.9"
+description = "TsFile Python"
+readme = {file = "README.md", content-type = "text/markdown"}
+maintainers = [
+    {name = "Apache TsFile Developers", email = "[email protected]"}
+]
+dependencies = [
+    "numpy>=1.20",
+    "pandas>=2.0"
+]
+
+[project.urls]
+Homepage = "https://tsfile.apache.org/";
+Documentation = 
"https://tsfile.apache.org/zh/UserGuide/latest/QuickStart/Navigating_Time_Series_Data.html";
+Repository = "https://github.com/apache/tsfile";
+Issues = "https://github.com/apache/tsfile/issues";
+
+[tool.setuptools]
+package-dir = {"" = "."}
+
+[tool.setuptools.packages.find]
+where = ["."]
+include = ["tsfile*"]
+
+[tool.setuptools.package-data]
+tsfile = [
+    "*.pxd",
+    "*.pxi",
+    "*.so",
+    "*.so.*",
+    "*.dylib",
+    "*.dylib.*",
+    "*.dll",
+    "*.dll.a",
+    "*.lib",
+    "*.lib.a",
+    "include/**/*"
+]
diff --git a/python/requirements.txt b/python/requirements.txt
index 06f90808..08baaac7 100644
--- a/python/requirements.txt
+++ b/python/requirements.txt
@@ -18,8 +18,8 @@
 #
 
 cython==3.0.10
-numpy==1.26.4
-pandas==2.2.2
+numpy
+pandas
 setuptools==78.1.1
 wheel==0.45.1
 
diff --git a/python/setup.py b/python/setup.py
index 329cc2aa..a80ef7eb 100644
--- a/python/setup.py
+++ b/python/setup.py
@@ -19,151 +19,120 @@
 import os
 import platform
 import shutil
-
+import sys
 import numpy as np
+
+from pathlib import Path
 from Cython.Build import cythonize
 from setuptools import setup, Extension
 from setuptools.command.build_ext import build_ext
 
-version = "2.1.0.dev0"
-system = platform.system()
-
-
-def copy_tsfile_lib(source_dir, target_dir, suffix):
-    lib_file_name = f"libtsfile.{suffix}"
-    source = os.path.join(source_dir, lib_file_name)
-    target = os.path.join(target_dir, lib_file_name)
-
-    if os.path.exists(source):
-        shutil.copyfile(source, target)
-
-    if system == "Linux":
-        link_name = os.path.join(target_dir, "libtsfile.so")
-        if os.path.exists(link_name):
-            os.remove(link_name)
-        os.symlink(lib_file_name, link_name)
-    elif system == "Darwin":
-        link_name = os.path.join(target_dir, "libtsfile.dylib")
-        if os.path.exists(link_name):
-            os.remove(link_name)
-        os.symlink(lib_file_name, link_name)
-
-
-project_dir = os.path.dirname(os.path.abspath(__file__))
-tsfile_py_include = os.path.join(project_dir, "tsfile", "include")
-
-if os.path.exists(tsfile_py_include):
-    shutil.rmtree(tsfile_py_include)
-
-shutil.copytree(
-    os.path.join(project_dir, "..", "cpp", "target", "build", "include"),
-    os.path.join(tsfile_py_include, ""),
-)
-
-
-def copy_tsfile_header(source):
-    for file in source:
-        if os.path.exists(file):
-            target = os.path.join(tsfile_py_include, os.path.basename(file))
-            shutil.copyfile(file, target)
+ROOT = Path(__file__).parent.resolve()
+PKG = ROOT / "tsfile"
+CPP_OUT = ROOT / ".." / "cpp" / "target" / "build"
+CPP_LIB = CPP_OUT / "lib"
+CPP_INC = CPP_OUT / "include"
 
+version = "2.1.2"
+system = platform.system()
 
-## Copy C wrapper header.
-# tsfile/cpp/src/cwrapper/tsfile_cwrapper.h
-source_headers = [
-    os.path.join(project_dir, "..", "cpp", "src", "cwrapper", 
"tsfile_cwrapper.h"),
-]
-
-copy_tsfile_header(source_headers)
-
-## Copy shared library
-tsfile_shared_source_dir = os.path.join(project_dir, "..", "cpp", "target", 
"build", "lib")
-tsfile_shared_dir = os.path.join(project_dir, "tsfile")
-
-if system == "Darwin":
-    copy_tsfile_lib(tsfile_shared_source_dir, tsfile_shared_dir, version + 
".dylib")
-elif system == "Linux":
-    copy_tsfile_lib(tsfile_shared_source_dir, tsfile_shared_dir, "so." + 
version)
+(PKG / "include").mkdir(exist_ok=True)
+if (PKG / "include").exists() and CPP_INC.exists():
+    shutil.rmtree(PKG / "include")
+    shutil.copytree(CPP_INC, PKG / "include")
+if sys.platform.startswith("linux"):
+    candidates = sorted(CPP_LIB.glob("libtsfile.so*"), key=lambda p: 
len(p.name), reverse=True)
+    if not candidates:
+        raise FileNotFoundError("missing libtsfile.so* in build output")
+    src = candidates[0]
+    dst = PKG / src.name
+    shutil.copy2(src, dst)
+    link_name = PKG / "libtsfile.so"
+    shutil.copy2(src, link_name)
+
+elif sys.platform == "darwin":
+    candidates = sorted(CPP_LIB.glob("libtsfile.*.dylib")) or 
list(CPP_LIB.glob("libtsfile.dylib"))
+    if not candidates:
+        raise FileNotFoundError("missing libtsfile*.dylib in build output")
+    src = candidates[0]
+    dst = PKG / src.name
+    shutil.copy2(src, dst)
+    link_name = PKG / "libtsfile.dylib"
+    shutil.copy2(src, link_name)
+elif sys.platform == "win32":
+    for base_name in ("libtsfile",):
+        dll_candidates = sorted(CPP_LIB.glob(f"{base_name}*.dll"), key=lambda 
p: len(p.name), reverse=True)
+        dll_a_candidates = sorted(CPP_LIB.glob(f"{base_name}*.dll.a"), 
key=lambda p: len(p.name), reverse=True)
+
+        if not dll_candidates:
+            raise FileNotFoundError(f"missing {base_name}*.dll in build 
output")
+        if not dll_a_candidates:
+            raise FileNotFoundError(f"missing {base_name}*.dll.a in build 
output")
+
+        dll_src = dll_candidates[0]
+        dll_a_src = dll_a_candidates[0]
+
+        shutil.copy2(dll_src, PKG / f"{base_name}.dll")
+        shutil.copy2(dll_a_src, PKG / f"{base_name}.dll.a")
 else:
-    copy_tsfile_lib(tsfile_shared_source_dir, tsfile_shared_dir, "dll")
-
-tsfile_include_dir = os.path.join(project_dir, "tsfile", "include")
-
-ext_modules_tsfile = [
-    # utils: from python to c or c to python.
-    Extension(
-        "tsfile.tsfile_py_cpp",
-        sources=[os.path.join("tsfile", "tsfile_py_cpp.pyx")],
-        libraries=["tsfile"],
-        library_dirs=[tsfile_shared_dir],
-        include_dirs=[tsfile_include_dir, np.get_include()],
-        runtime_library_dirs=[tsfile_shared_dir] if system != "Windows" else 
None,
-        extra_compile_args=(
-            ["-std=c++11"] if system != "Windows" else ["-std=c++11", 
"-DMS_WIN64"]
-        ),
-        language="c++",
-    ),
-    # query data and describe schema: tsfile reader module
-    Extension(
-        "tsfile.tsfile_reader",
-        sources=[os.path.join("tsfile", "tsfile_reader.pyx")],
-        libraries=["tsfile"],
-        library_dirs=[tsfile_shared_dir],
-        depends=[os.path.join("tsfile", "tsfile_py_cpp.pxd")],
-        include_dirs=[tsfile_include_dir, np.get_include()],
-        runtime_library_dirs=[tsfile_shared_dir] if system != "Windows" else 
None,
-        extra_compile_args=(
-            ["-std=c++11"] if system != "Windows" else ["-std=c++11", 
"-DMS_WIN64"]
-        ),
-        language="c++",
-    ),
-    # write data and register schema: tsfile writer module
-    Extension(
-        "tsfile.tsfile_writer",
-        sources=[os.path.join("tsfile", "tsfile_writer.pyx")],
-        libraries=["tsfile"],
-        library_dirs=[tsfile_shared_dir],
-        depends=[os.path.join("tsfile", "tsfile_py_cpp.pxd")],
-        include_dirs=[tsfile_include_dir, np.get_include()],
-        runtime_library_dirs=[tsfile_shared_dir] if system != "Windows" else 
None,
-        extra_compile_args=(
-            ["-std=c++11"] if system != "Windows" else ["-std=c++11", 
"-DMS_WIN64"]
-        ),
-        language="c++",
-    )
-]
+    raise RuntimeError(f"Unsupported platform: {sys.platform}")
 
 
 class BuildExt(build_ext):
-    def build_extensions(self):
-        numpy_include = np.get_include()
-        for ext in self.extensions:
-            ext.include_dirs.append(numpy_include)
-        super().build_extensions()
+    def run(self):
+        super().run()
 
     def finalize_options(self):
-        if system == "Windows":
+        if sys.platform == "win32":
             self.compiler = "mingw32"
         super().finalize_options()
 
 
+extra_compile_args = []
+extra_link_args = []
+runtime_library_dirs = []
+libraries = []
+library_dirs = [str(PKG)]
+include_dirs = [str(PKG), np.get_include(), str(PKG / "include")]
+
+if sys.platform.startswith("linux"):
+    libraries = ["tsfile"]
+    extra_compile_args += ["-O3", "-std=c++11", "-fvisibility=hidden", "-fPIC"]
+    runtime_library_dirs = ["$ORIGIN"]
+    extra_link_args += ["-Wl,-rpath,$ORIGIN"]
+elif sys.platform == "darwin":
+    libraries = ["tsfile"]
+    extra_compile_args += ["-O3", "-std=c++11", "-fvisibility=hidden", "-fPIC"]
+    extra_link_args += ["-Wl,-rpath,@loader_path", "-stdlib=libc++"]
+elif sys.platform == "win32":
+    libraries = ["Tsfile"]
+    extra_compile_args += ["-O2", "-std=c++11", "-DSIZEOF_VOID_P=8", 
"-D__USE_MINGW_ANSI_STDIO=1", "-DMS_WIN64",
+                           "-D_WIN64"]
+else:
+    raise RuntimeError(f"Unsupported platform: {sys.platform}")
+
+common = dict(
+    language="c++",
+    include_dirs=include_dirs,
+    library_dirs=library_dirs,
+    libraries=libraries,
+    extra_compile_args=extra_compile_args,
+    extra_link_args=extra_link_args,
+    runtime_library_dirs=runtime_library_dirs,
+)
+
+exts = [
+    Extension("tsfile.tsfile_py_cpp", ["tsfile/tsfile_py_cpp.pyx"], **common),
+    Extension("tsfile.tsfile_reader", ["tsfile/tsfile_reader.pyx"], **common),
+    Extension("tsfile.tsfile_writer", ["tsfile/tsfile_writer.pyx"], **common),
+]
+
 setup(
     name="tsfile",
-    version=version,
-    description="Tsfile reader and writer for python",
-    url="https://tsfile.apache.org";,
-    author='"Apache TsFile"',
+    version="2.1.0",
     packages=["tsfile"],
-    license="Apache 2.0",
-    ext_modules=cythonize(ext_modules_tsfile),
-    cmdclass={"build_ext": BuildExt},
-    include_dirs=[np.get_include()],
-    package_dir={"tsfile": "./tsfile"},
-    package_data={
-        "tsfile": [
-            "libtsfile.*",
-            "*.pxd"
-        ]
-    },
+    package_dir={"": "."},
     include_package_data=True,
+    ext_modules=cythonize(exts, compiler_directives={"language_level": 3}),
+    cmdclass={"build_ext": BuildExt},
 )
diff --git a/python/tsfile/__init__.py b/python/tsfile/__init__.py
index bf755fce..5b120c0d 100644
--- a/python/tsfile/__init__.py
+++ b/python/tsfile/__init__.py
@@ -19,10 +19,6 @@
 import ctypes
 import os
 import platform
-system = platform.system()
-if system == "Windows":
-    ctypes.WinDLL(os.path.join(os.path.dirname(__file__), "libtsfile.dll"), 
winmode=0)
-
 from .constants import *
 from .schema import *
 from .row_record import *
diff --git a/python/tsfile/tsfile_cpp.pxd b/python/tsfile/tsfile_cpp.pxd
index e9a67364..e7f45fea 100644
--- a/python/tsfile/tsfile_cpp.pxd
+++ b/python/tsfile/tsfile_cpp.pxd
@@ -22,7 +22,7 @@ from libc.stdint cimport uint32_t, int32_t, int64_t, 
uint64_t, uint8_t
 ctypedef int32_t ErrorCode
 
 # import symbols from tsfile_cwrapper.h
-cdef extern from "./tsfile_cwrapper.h":
+cdef extern from "cwrapper/tsfile_cwrapper.h":
     # common
     ctypedef int64_t timestamp
 
@@ -202,7 +202,7 @@ cdef extern from "./tsfile_cwrapper.h":
 
 
 
-cdef extern from "./common/config/config.h" namespace "common":
+cdef extern from "common/config/config.h" namespace "common":
     cdef cppclass ConfigValue:
         uint32_t tsblock_mem_inc_step_size_
         uint32_t tsblock_max_memory_
@@ -224,7 +224,7 @@ cdef extern from "./common/config/config.h" namespace 
"common":
         uint8_t string_encoding_type_;
         uint8_t default_compression_type_;
 
-cdef extern from "./common/global.h" namespace "common":
+cdef extern from "common/global.h" namespace "common":
     ConfigValue g_config_value_
     int set_datatype_encoding(uint8_t data_type, uint8_t encoding)
     int set_global_compression(uint8_t compression)

Reply via email to