Add disting to pypi to release steps

Author: timkpaine

litex_setup.py

@@ -4,11 +4,9 @@
 import sys
 import time
 import subprocess
-import shutil
 import hashlib
 import argparse
 
-import urllib.request
 
 start_time   = time.time()
 current_path = os.path.abspath(os.curdir)
@@ -207,6 +205,31 @@ def git_tag(tag=None):
     os.system(f"git tag {tag}")
     os.system(f"git push --tags")
 
+# Python Build helpers -----------------------------------------------------------------------------
+def bump_version(version):
+    with open("setup.py", "r") as f:
+        content = f.readlines()
+    with open("setup.py", "w") as f:
+        for line in content:
+            # Find the version line
+            if line.strip().startswith("version"):
+                # Get the current version
+                current_version = line.strip().split('=')[1].strip(" ,\"")
+                # Print the current version for debugging
+                print_status(f"Current version: {current_version}")
+                # Print the new version for debugging
+                print_status(f"New version: {version}")
+
+                line = line.replace(current_version, version)
+            f.write(line)
+
+def build_repo():
+    os.system(f"python -m build --sdist --wheel")
+    os.system(f"python -m twine check dist/*")
+
+def push_to_pypi():
+    os.system(f"python -m twine upload dist/*")
+
 # Git repositories initialization ------------------------------------------------------------------
 
 def litex_setup_init_repos(config="standard", tag=None, dev_mode=False):
@@ -330,11 +353,28 @@ def litex_setup_release_repos(tag):
         for name in install_configs["full"]:
             if name in ["migen"]:
                 continue
-            repo = git_repos[name]
+
             os.chdir(os.path.join(current_path, name))
+
             # Tag Repo.
             print_status(f"Tagging {name} Git repository as {tag}...")
             git_tag(tag=tag)
+
+            # TODO: Not all repos have been uplifted for redistribution.
+            # Eventually, this "if" check should be removed and the bump
+            # versioning part moved above the git tag
+            if name in ("litex",):
+                # Bump the version
+                print_status(f"Bumping the version of {name} python package...")
+                bump_version(version=tag)
+
+                # Make sure the repo builds for distribution
+                print_status(f"Building {name} python package...")
+                build_repo()
+
+                # Push disted artifacts to pypi
+                print_status(f"Publishing {name} to pypi...")
+                push_to_pypi()
     else:
         print_status(f"Not confirmed, exiting.")
 

setup.py

@@ -10,7 +10,7 @@
 
 setup(
     name                          = "litex",
-    version                       = "2024.08",
+    version                       = "2024.12",
     description                   = "Python SoC/Core builder for building FPGA based systems.",
     long_description              = long_description,
     long_description_content_type = "text/markdown",
@@ -32,7 +32,11 @@
           "meson"
           "pexpect"
           "setuptools"
-          "requests"
+          "requests",
+          # Tools for distribution
+          "build",
+          "twine",
+          "wheel",
         ]
     },
     packages                       = find_packages(exclude=("test*", "sim*", "doc*")),