Allow create_baseline_stubs.py to add packages to existing stub (#8809)

This commit is contained in:
Samuel T
2022-09-30 10:49:58 -04:00
committed by GitHub
parent 380022c650
commit f511f1f4e9

View File

@@ -62,15 +62,16 @@ def run_isort(stub_dir: str) -> None:
def create_metadata(stub_dir: str, version: str) -> None:
"""Create a METADATA.toml file."""
m = re.match(r"[0-9]+.[0-9]+", version)
if m is None:
match = re.match(r"[0-9]+.[0-9]+", version)
if match is None:
sys.exit(f"Error: Cannot parse version number: {version}")
fnam = os.path.join(stub_dir, "METADATA.toml")
version = m.group(0)
assert not os.path.exists(fnam)
print(f"Writing {fnam}")
with open(fnam, "w") as f:
f.write(
filename = os.path.join(stub_dir, "METADATA.toml")
version = match.group(0)
if os.path.exists(filename):
return
print(f"Writing {filename}")
with open(filename, "w") as file:
file.write(
f"""\
version = "{version}.*"
@@ -153,8 +154,9 @@ def main() -> None:
project, version = info
stub_dir = os.path.join("stubs", project)
if os.path.exists(stub_dir):
sys.exit(f"Error: {stub_dir} already exists (delete it first)")
package_dir = os.path.join(stub_dir, package)
if os.path.exists(package_dir):
sys.exit(f"Error: {package_dir} already exists (delete it first)")
run_stubgen(package, stub_dir)