-
Notifications
You must be signed in to change notification settings - Fork 96
/
uprev.py
88 lines (66 loc) · 2.46 KB
/
uprev.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
"""Update the pydantic-ai version number everywhere.
Because we have multiple packages which depend on one-another,
we have to update the version number in:
* pyproject.toml
* pydantic_ai_examples/pyproject.toml
* pydantic_ai_slim/pyproject.toml
Usage:
uv run uprev.py <new_version_number>
"""
import re
import sys
from pathlib import Path
ROOT_DIR = Path(__file__).parent
try:
version = sys.argv[1]
except IndexError:
print('Usage: uv run uprev.py <new_version_number>', file=sys.stderr)
sys.exit(1)
old_version: str | None = None
def sub_version(m: re.Match[str]) -> str:
global old_version
prefix = m.group(1)
quote = m.group(2)
old_version = m.group(3)
return f'{prefix}{quote}{version}{quote}'
root_pp = ROOT_DIR / 'pyproject.toml'
root_pp_text = root_pp.read_text()
root_pp_text, _ = re.subn(r'^(version ?= ?)(["\'])(.+)\2$', sub_version, root_pp_text, 1, flags=re.M)
if old_version is None:
print('ERROR: Could not find version in root pyproject.toml', file=sys.stderr)
sys.exit(1)
print(f'Updating version from {old_version!r} to {version!r}')
def replace_deps_version(text: str) -> tuple[str, int]:
ovr = re.escape(old_version)
text, c = re.subn(f'(pydantic-ai-.+?==){ovr}', fr'\g<1>{version}', text, count=5)
text, c2 = re.subn(
fr'^(version ?= ?)(["\']){ovr}\2$', fr'\g<1>\g<2>{version}\g<2>', text, count=5, flags=re.M
)
return text, c + c2
root_pp_text, count_root = replace_deps_version(root_pp_text)
examples_pp = ROOT_DIR / 'pydantic_ai_examples' / 'pyproject.toml'
examples_pp_text = examples_pp.read_text()
examples_pp_text, count_ex = replace_deps_version(examples_pp_text)
slim_pp = ROOT_DIR / 'pydantic_ai_slim' / 'pyproject.toml'
slim_pp_text = slim_pp.read_text()
slim_pp_text, count_slim = replace_deps_version(slim_pp_text)
if count_root == 2 and count_ex == 2 and count_slim == 1:
root_pp.write_text(root_pp_text)
examples_pp.write_text(examples_pp_text)
slim_pp.write_text(slim_pp_text)
print(
f'SUCCESS: replaced version in\n'
f'* {root_pp}\n'
f'* {examples_pp}\n'
f'* {slim_pp}\n\n'
'Now run `make sync` to update the lock files.',
)
else:
print(
f'ERROR:\n'
f' {count_root} version references in {root_pp} (expected 2)\n'
f' {count_ex} version references in {examples_pp} (expected 2)\n'
f' {count_slim} version references in {slim_pp} (expected 1)',
file=sys.stderr,
)
sys.exit(1)