mirror of
https://github.com/TECHNOFAB11/bumpver.git
synced 2025-12-12 06:20:08 +01:00
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
# This file is part of the pycalver project
|
|
# https://github.com/mbarkhau/pycalver
|
|
#
|
|
# (C) 2018 Manuel Barkhau (@mbarkhau)
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
import io
|
|
import difflib
|
|
import logging
|
|
import typing as typ
|
|
|
|
from . import parse
|
|
|
|
log = logging.getLogger("pycalver.rewrite")
|
|
|
|
|
|
def rewrite_lines(
|
|
old_lines: typ.List[str], patterns: typ.List[str], new_version: str
|
|
) -> typ.List[str]:
|
|
new_version_nfo = parse.parse_version_info(new_version)
|
|
new_version_fmt_kwargs = new_version_nfo._asdict()
|
|
|
|
new_lines = old_lines.copy()
|
|
|
|
matches: typ.List[parse.PatternMatch] = parse.parse_patterns(old_lines, patterns)
|
|
|
|
for m in matches:
|
|
replacement = m.pattern.format(**new_version_fmt_kwargs)
|
|
span_l, span_r = m.span
|
|
new_line = m.line[:span_l] + replacement + m.line[span_r:]
|
|
new_lines[m.lineno] = new_line
|
|
|
|
return new_lines
|
|
|
|
|
|
def rewrite(
|
|
new_version: str, file_patterns: typ.Dict[str, typ.List[str]], dry=False, verbose=False
|
|
) -> None:
|
|
for filepath, patterns in file_patterns.items():
|
|
with io.open(filepath, mode="rt", encoding="utf-8") as fh:
|
|
content = fh.read()
|
|
|
|
# TODO (mb 2018-09-18): Detect line sep
|
|
line_sep = "\n"
|
|
|
|
old_lines = content.splitlines()
|
|
new_lines = rewrite_lines(old_lines, patterns, new_version)
|
|
|
|
if dry or verbose:
|
|
diff_lines = difflib.unified_diff(
|
|
old_lines, new_lines, lineterm="", fromfile="a/" + filepath, tofile="b/" + filepath
|
|
)
|
|
print("\n".join(diff_lines))
|
|
|
|
if dry:
|
|
continue
|
|
|
|
new_content = line_sep.join(new_lines)
|
|
with io.open(filepath, mode="wt", encoding="utf-8") as fh:
|
|
fh.write(new_content)
|