@@ -32,23 +32,32 @@ def update_version_in_file(file_path, old_version, new_version):
32
32
file .write (content )
33
33
34
34
35
- def main ():
36
- bump_type = sys .argv [1 ]
35
+ def main (bump_type , packages ):
37
36
38
- for package_dir in os .listdir ("." ):
39
- if os .path .isdir (package_dir ) and package_dir .startswith ("plain" ):
40
- pyproject_path = os .path .join (package_dir , "pyproject.toml" )
37
+ for package_dir in packages :
38
+ pyproject_path = os .path .join (package_dir , "pyproject.toml" )
41
39
42
- with open (pyproject_path , "r" ) as file :
43
- content = file .read ()
40
+ with open (pyproject_path , "r" ) as file :
41
+ content = file .read ()
44
42
45
- version_match = re .search (r'version = "(.*?)"' , content )
46
- if version_match :
47
- old_version = version_match .group (1 )
48
- new_version = bump_version (old_version , bump_type )
49
- update_version_in_file (pyproject_path , old_version , new_version )
50
- print (f"Bumped { package_dir } from { old_version } to { new_version } " )
43
+ version_match = re .search (r'version = "(.*?)"' , content )
44
+ if version_match :
45
+ old_version = version_match .group (1 )
46
+ new_version = bump_version (old_version , bump_type )
47
+ update_version_in_file (pyproject_path , old_version , new_version )
48
+ print (f"Bumped { package_dir } from { old_version } to { new_version } " )
51
49
52
50
53
51
if __name__ == "__main__" :
54
- main ()
52
+ bump_type = sys .argv [1 ]
53
+
54
+ if len (sys .argv ) > 1 :
55
+ packages = sys .argv [2 :]
56
+ else :
57
+ packages = [
58
+ package
59
+ for package in os .listdir ("." )
60
+ if os .path .isdir (package ) and package .startswith ("plain" )
61
+ ]
62
+
63
+ main (bump_type , packages )
0 commit comments