pre-commit/pre-commit-hooks
1
0
Fork 0
mirror of https://github.com/pre-commit/pre-commit-hooks.git synced 2026-03-29 18:16:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: process fstring when python version gte 3.12

Browse source
This commit is contained in:
ifuryst 2024-04-15 18:07:21 +08:00
parent 0d20f18212
commit 8a5e9ade6c
2 changed files with 17 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
pre_commit_hooks/string_fixer.py
View file

@ -48,6 +48,8 @@ def fix_strings(filename: str) -> int:
splitcontents = list(contents)
fstring_depth = 0
fstring_content = ''
f_erow = f_ecol = -1
# Iterate in reverse so the offsets are always correct
tokens_l = list(tokenize.generate_tokens(io.StringIO(contents).readline))
@ -55,8 +57,17 @@ def fix_strings(filename: str) -> int:
for token_type, token_text, (srow, scol), (erow, ecol), _ in tokens:
if token_type == FSTRING_START: # pragma: >=3.12 cover
fstring_depth += 1
splitcontents[
line_offsets[srow] + scol:
line_offsets[f_erow] + f_ecol
] = handle_match(token_text + fstring_content)
fstring_content = ''
elif token_type == FSTRING_END: # pragma: >=3.12 cover
fstring_depth -= 1
fstring_content = token_text + fstring_content
f_erow, f_ecol = erow, ecol
elif fstring_depth != 0: # pragma: >=3.12 cover
fstring_content = token_text + fstring_content
elif fstring_depth == 0 and token_type == tokenize.STRING:
new_text = handle_match(token_text)
splitcontents[