51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
import re
|
|
from pathlib import Path
|
|
|
|
|
|
def fix_file(filepath):
|
|
with open(filepath, "r", encoding="utf-8") as f:
|
|
content = f.read()
|
|
|
|
original = content
|
|
|
|
# 修复 conn.execute(\n """\n -> conn.execute("""
|
|
content = re.sub(r'(\.execute|\.executemany|cursor\.execute)\(\s*\n\s*(""")', r"\1(\2", content)
|
|
|
|
# 修复 """\n) -> """)
|
|
content = re.sub(r'(""")\s*\n\s*\)', r"\1)", content)
|
|
|
|
# 修复不必要的括号: var = (\n expr\n)
|
|
content = re.sub(r"(\w+)\s*=\s*\(\s*\n\s*([^\n]+)\s*\n\s*\)", r"\1 = \2", content)
|
|
|
|
if content != original:
|
|
with open(filepath, "w", encoding="utf-8") as f:
|
|
f.write(content)
|
|
return True
|
|
return False
|
|
|
|
|
|
files = [
|
|
"outlook_web/controllers/scheduler.py",
|
|
"outlook_web/controllers/system.py",
|
|
"outlook_web/repositories/groups.py",
|
|
"outlook_web/db.py",
|
|
"outlook_web/repositories/accounts.py",
|
|
"start.py",
|
|
"outlook_web/services/scheduler.py",
|
|
"outlook_web/controllers/accounts.py",
|
|
"outlook_web/services/refresh.py",
|
|
]
|
|
|
|
modified = 0
|
|
for f in files:
|
|
if Path(f).exists():
|
|
if fix_file(f):
|
|
print(f"✓ {f}")
|
|
modified += 1
|
|
else:
|
|
print(f"- {f}")
|
|
else:
|
|
print(f"✗ {f} 不存在")
|
|
|
|
print(f"\n修改了 {modified} 个文件")
|