55
66import ruamel .yaml
77
8- SUPPORTED = frozenset (
9- (
10- "black" ,
11- "flake8" ,
12- )
13- )
8+ SUPPORTED = frozenset (('black' , 'flake8' ))
149
1510
1611def main (argv : Sequence [str ] | None = None ) -> int :
1712 parser = argparse .ArgumentParser ()
18- parser .add_argument (" filename" , default = " .pre-commit-config.yaml" )
13+ parser .add_argument (' filename' , default = ' .pre-commit-config.yaml' )
1914
2015 args = parser .parse_args (argv )
2116 filename : str = args .filename
@@ -31,32 +26,33 @@ def main(argv: Sequence[str] | None = None) -> int:
3126
3227 # TODO - validate schema?
3328 versions = {}
34- for repo in loaded [" repos" ]:
35- for hook in repo [" hooks" ]:
36- if (hid := hook ["id" ]) in SUPPORTED :
37- versions [hid ] = repo [" rev" ]
29+ for repo in loaded [' repos' ]:
30+ for hook in repo [' hooks' ]:
31+ if (hid := hook ['id' ]) in SUPPORTED :
32+ versions [hid ] = repo [' rev' ]
3833
3934 updated = []
40- for repo in loaded [" repos" ]:
41- for hook in repo [" hooks" ]:
42- for i , dep in enumerate (hook .get (" additional_dependencies" , ())):
43- name , _ , cur_version = dep .partition ("==" )
35+ for repo in loaded [' repos' ]:
36+ for hook in repo [' hooks' ]:
37+ for i , dep in enumerate (hook .get (' additional_dependencies' , ())):
38+ name , _ , cur_version = dep .partition ('==' )
4439 target_version = versions .get (name , cur_version )
4540 if target_version != cur_version :
46- hook ["additional_dependencies" ][i ] = f"{ name } =={ target_version } "
47- updated .append ((hook ["id" ], name ))
41+ name_and_version = f'{ name } =={ target_version } '
42+ hook ['additional_dependencies' ][i ] = name_and_version
43+ updated .append ((hook ['id' ], name ))
4844
4945 if updated :
50- print (f" Writing updates to { filename } :" )
46+ print (f' Writing updates to { filename } :' )
5147 for hid , name in updated :
52- print (f" \t Setting { hid !r} dependency { name !r} to { versions [name ]} " )
48+ print (f' \t Setting { hid !r} dependency { name !r} to { versions [name ]} ' )
5349
54- with open (filename , "w+" ) as f :
50+ with open (filename , 'w+' ) as f :
5551 yaml .dump (loaded , f )
5652 return 1
5753
5854 return 0
5955
6056
61- if __name__ == " __main__" :
57+ if __name__ == ' __main__' :
6258 raise SystemExit (main ())
0 commit comments