feat: migrate from poetry to uv

.github/workflows/ci.yml

@@ -17,25 +17,19 @@ jobs:
         options: --health-cmd="mysqladmin ping" --health-interval 10s --health-timeout 5s --health-retries 5
     strategy:
       matrix:
-        python-version: [ "3.9", "3.10", "3.11", "3.12", "3.13" ]
+        python-version: [ "3.9", "3.10", "3.11", "3.12", "3.13", "3.14" ]
     steps:
-      - uses: actions/cache@v4
-        with:
-          path: ~/.cache/pip
-          key: ${{ runner.os }}-pip-${{ hashFiles('**/poetry.lock') }}
-          restore-keys: |
-            ${{ runner.os }}-pip-
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
-      - name: Install and configure Poetry
-        run: |
-          pip install -U pip poetry
-          poetry config virtualenvs.create false
-          pip3 install wheel setuptools pip --upgrade
+          allow-prereleases: true
+      - uses: astral-sh/setup-uv@v6
+        with:
+          enable-cache: true
       - name: Run ci
-        run: make ci
+        run: uv run make ci
+
   ci_mariadb:
     runs-on: ubuntu-latest
     services:
@@ -48,22 +42,15 @@ jobs:
         options: --health-cmd="mariadb-admin ping -uroot -p${MYSQL_ROOT_PASSWORD}" --health-interval 10s --health-timeout 5s --health-retries 5
     strategy:
       matrix:
-        python-version: [ "3.9", "3.10", "3.11", "3.12", "3.13" ]
+        python-version: [ "3.9", "3.10", "3.11", "3.12", "3.13", "3.14" ]
     steps:
-      - uses: actions/cache@v4
-        with:
-          path: ~/.cache/pip
-          key: ${{ runner.os }}-pip-${{ hashFiles('**/poetry.lock') }}
-          restore-keys: |
-            ${{ runner.os }}-pip-
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
           python-version: ${{ matrix.python-version }}
-      - name: Install and configure Poetry
-        run: |
-          pip install -U pip poetry
-          poetry config virtualenvs.create false
-          pip3 install wheel setuptools pip --upgrade
+          allow-prereleases: true
+      - uses: astral-sh/setup-uv@v6
+        with:
+          enable-cache: true
       - name: Run ci
-        run: make ci
+        run: uv run make ci

.github/workflows/pypi.yml

@@ -28,9 +28,9 @@ jobs:
       - uses: actions/checkout@v4
         with:
           fetch-depth: 0
-      - uses: abatilo/actions-poetry@v3
+      - uses: astral-sh/setup-uv@v6
       - name: Build
-        run: poetry build
+        run: uv build
       - uses: actions/upload-artifact@v4
         with:
           name: cibw-sdist

Makefile

@@ -3,10 +3,10 @@ py_warn = PYTHONDEVMODE=1
 MYSQL_PASS ?= "123456"
 
 up:
-	@poetry update
+	@uv lock --upgrade
 
 deps:
-	@poetry install --all-groups
+	uv sync --active --inexact --all-groups --all-extras $(options)
 
 _style:
 	@ruff format $(checkfiles)
@@ -30,7 +30,7 @@ clean:
 	@rm -rf *.so && rm -rf build && rm -rf dist && rm -rf asyncmy/*.c && rm -rf asyncmy/*.so && rm -rf asyncmy/*.html
 
 build: clean
-	@poetry build
+	@uv build
 
 benchmark: deps
 	@python benchmark/main.py

README.md

@@ -118,8 +118,7 @@ async def run():
             await cursor.execute("SELECT 1")
             ret = await cursor.fetchone()
             assert ret == (1,)
-    pool.close()
-    await pool.wait_closed()
+    await pool.aclose()
 
 if __name__ == '__main__':
     asyncio.run(run())

asyncmy/pool.pyx

@@ -103,6 +103,11 @@ class Pool(asyncio.AbstractServer):
 
         self._closed = True
 
+    async def aclose(self):
+        """Close pool and wait for closing all pool's connections."""
+        self.close()
+        await self.wait_closed()
+
     def acquire(self):
         """Acquire free connection from the pool."""
         coro = self._acquire()
@@ -194,8 +199,7 @@ class Pool(asyncio.AbstractServer):
         return self
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
-        self.close()
-        await self.wait_closed()
+        await self.aclosed()
 
 
 def create_pool(

asyncmy/version.py

@@ -1 +1,2 @@
-__VERSION__ = "0.2.10"
+__version__ = "0.2.10"
+__VERSION__ = __version__

build.py

@@ -1,17 +1,8 @@
-from distutils.command.build_ext import build_ext
-
 from Cython.Build import cythonize
 
 
-def build(setup_kwargs):
-    setup_kwargs.update(
-        {
-            "ext_modules": cythonize(
-                [
-                    "asyncmy/*.pyx",
-                ],
-                compiler_directives={"language_level": 3},
-            ),
-            "cmdclass": {"build_ext": build_ext},
-        }
+def pdm_build_update_setup_kwargs(context, setup_kwargs) -> None:
+    setup_kwargs["ext_modules"] = cythonize(
+        ["asyncmy/*.pyx"],
+        language_level="3",
     )