Test 1

Author: scaramallion

.github/workflows/update-package.yml

@@ -0,0 +1,37 @@
+name: update-package
+
+on:
+  # workflow_dispatch:
+  pull_request:
+    branches: [ main ]
+
+jobs:
+  update:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout package
+        uses: actions/checkout@v4
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: "3.10"
+      - name: Install dependencies
+        run: |
+          pip install -U pip
+          pip install mypy
+          mkdir ../pydicom
+          git clone https://github.com/pydicom/pydicom ../pydicom
+          pip install -e ../pydicom
+      - name: Update package
+        run: |
+          ls -l
+          ls -l ../pydicom
+          git status
+          python scripts/update_package.py
+          git status
+      - name: Create Pull Request
+        # Only creates a new PR if there are changes
+        uses: peter-evans/create-pull-request@v6
+        with:
+          token: ${{ secrets.AUTO_UPDATE_PACKAGE }}
+          title: "Update package"

scripts/generate_stubs.py

@@ -6,7 +6,7 @@
 
 BASE_DIRECTORY = Path(__file__).parent.parent
 DS_DST = BASE_DIRECTORY / "custom" / "dataset.pyi"
-DS_SRC = BASE_DIRECTORY / "pydicom-stubs" / "dataset.pyi"
+DS_SRC = BASE_DIRECTORY / "src" / "pydicom-stubs" / "dataset.pyi"
 
 
 ElementDictType = dict[int, tuple[str, str, str, str]]

scripts/update_package.py

@@ -8,7 +8,7 @@
 
 BASE_DIRECTORY = Path(__file__).parent.parent
 SRC_DIRECTORY = BASE_DIRECTORY / "custom"
-DST_DIRECTORY = BASE_DIRECTORY / "pydicom-stubs"
+DST_DIRECTORY = BASE_DIRECTORY / "src" / "pydicom-stubs"
 PYDICOM_DIRECTORY = BASE_DIRECTORY.parent / "pydicom" / "src" / "pydicom"
 
 
@@ -36,29 +36,33 @@
     if DST_DIRECTORY.exists():
         shutil.rmtree(DST_DIRECTORY)
 
+    if (BASE_DIRECTORY / "pydicom").exists():
+        shutil.rmtree(BASE_DIRECTORY / "pydicom")
+
     # Generate basic stub files using mypy's `stubgen`
     print("Generating basic stub files with stubgen")
     subprocess.run(["which", "stubgen"], shell=True)
-    returncode = subprocess.run(
+    p = subprocess.run(
         [
-            f". {os.fspath(BASE_DIRECTORY / 'env' / 'env310' / 'bin' / 'activate')};"
             f"stubgen {os.fspath(PYDICOM_DIRECTORY)} -o .",
         ],
         shell=True,
     )
 
-    if not list(DST_DIRECTORY.glob("*.pyi")):
+    if p.returncode != 0:
         print("  Failed to generate the basic stub files")
         sys.exit(1)
 
+    # Move the basic stubs to the correct location
+    shutil.move(BASE_DIRECTORY / "pydicom", DST_DIRECTORY)
+
     # Generate the custom stub files
     print("Generating custom stub files")
     if not SRC_DIRECTORY.exists():
         SRC_DIRECTORY.mkdir(parents=True, exist_ok=True)
 
     subprocess.run(
         [
-            f". {os.fspath(BASE_DIRECTORY / 'env' / 'env310' / 'bin' / 'activate')};"
             "python scripts/generate_stubs.py",
         ],
         shell=True,