setup.py

import os.path as op
import string

from setuptools import setup
from setuptools_scm import get_version


def local_version(version):
    """Patch in a version that can be uploaded to test PyPI."""
    scm_version = get_version()
    if "dev" in scm_version:
        gh_in_int = []
        for char in version.node:
            if char.isdigit():
                gh_in_int.append(str(char))
            else:
                gh_in_int.append(str(string.ascii_letters.find(char)))
        return "".join(gh_in_int)
    else:
        return ""


opts = {
    "use_scm_version": {
        "write_to": op.join("afqinsight", "_version.py"),
        "local_scheme": local_version,
    }
}


if __name__ == "__main__":
    setup(**opts)