# -*- coding: utf-8 -*-
import io
path = r'C:\Users\jcxiong\Documents\Codex\MyProject\trafficAudit\traffic-audit-server\src\main\java\com\trafficaudit\reportexport\service\ReportExportService.java'
with io.open(path, encoding='utf-8') as f:
    content = f.read()
has_crlf = '\r\n' in content
lines = content.split('\n')

def patch_func(func, rowidx_from, rowidx_to, unit_from, unit_to):
    start = next(i for i, l in enumerate(lines) if l.strip().startswith('public byte[] ' + func + '('))
    end = len(lines)
    for i in range(start+1, len(lines)):
        if lines[i].strip().startswith('public byte[] '):
            end = i
            break
    cnt_r = cnt_u = 0
    for i in range(start, end):
        if lines[i].strip() == 'int rowIdx = ' + rowidx_from + ';':
            lines[i] = lines[i].replace('int rowIdx = ' + rowidx_from + ';', 'int rowIdx = ' + rowidx_to + ';')
            cnt_r += 1
        if '"' + unit_from + '"' in lines[i]:
            lines[i] = lines[i].replace('"' + unit_from + '"', '"' + unit_to + '"')
            cnt_u += 1
    print(func, '-> rowIdx:', cnt_r, '| unit:', cnt_u)

patch_func('exportFreightRank', '3', '1', '万吨', '吨')
patch_func('exportTurnoverRank', '3', '1', '万吨公里', '吨公里')

new_content = '\n'.join(lines)
if has_crlf:
    new_content = new_content.replace('\n', '\r\n')
with io.open(path, 'w', encoding='utf-8', newline='') as f:
    f.write(new_content)
print('saved')