67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
import os
|
|
import glob
|
|
import re
|
|
|
|
def process_file(filepath):
|
|
try:
|
|
with open(filepath, 'r', encoding='utf-8') as f:
|
|
content = f.read()
|
|
except UnicodeDecodeError:
|
|
with open(filepath, 'r', encoding='cp949') as f:
|
|
content = f.read()
|
|
|
|
# Remove @Data
|
|
content = re.sub(r'^\s*@Data\s*\n', '', content, flags=re.MULTILINE)
|
|
|
|
# Skip interfaces and enums
|
|
if 'public interface ' in content or 'public enum ' in content:
|
|
return
|
|
|
|
if 'public abstract class ' in content:
|
|
return
|
|
|
|
# Add imports
|
|
imports = ["lombok.Getter", "lombok.Setter", "lombok.Builder", "lombok.NoArgsConstructor", "lombok.AllArgsConstructor"]
|
|
import_block = ""
|
|
for imp in imports:
|
|
if f"import {imp};" not in content:
|
|
import_block += f"import {imp};\n"
|
|
|
|
if import_block:
|
|
content = re.sub(r'^(package\s+[^;]+;)\s*\n', f"\\1\n\n{import_block}", content, count=1, flags=re.MULTILINE)
|
|
|
|
# Add annotations
|
|
annotations = []
|
|
if '@Getter' not in content: annotations.append('@Getter')
|
|
if '@Setter' not in content: annotations.append('@Setter')
|
|
if '@Builder' not in content: annotations.append('@Builder')
|
|
if '@NoArgsConstructor' not in content: annotations.append('@NoArgsConstructor')
|
|
if '@AllArgsConstructor' not in content: annotations.append('@AllArgsConstructor')
|
|
|
|
if annotations:
|
|
anno_str = '\n'.join(annotations) + '\n'
|
|
content = re.sub(r'^(public\s+class\s+)', f"{anno_str}\\1", content, count=1, flags=re.MULTILINE)
|
|
|
|
# Remove manual ToolMetadata() and ZtUsacInDto() constructors
|
|
content = re.sub(r'^\s*public\s+ToolMetadata\(\)\s*\{\}\s*\n', '', content, flags=re.MULTILINE)
|
|
content = re.sub(r'^\s*public\s+ZtUsacInDto\(\)\s*\{\s*setPuseYn\("Y"\);\s*\}\s*\n', '', content, flags=re.MULTILINE)
|
|
|
|
# Fix puseYn builder default
|
|
if 'private String puseYn;' in content and 'ZtUsacInDto' in filepath:
|
|
content = content.replace('private String puseYn;', '@Builder.Default\n private String puseYn = "Y";')
|
|
|
|
# Remove @Builder from AuditInfo (to prevent inheritance clash)
|
|
if 'AuditInfo.java' in filepath:
|
|
content = content.replace('@Builder\n', '')
|
|
|
|
with open(filepath, 'w', encoding='utf-8') as f:
|
|
f.write(content)
|
|
print(f"Processed: {filepath}")
|
|
|
|
# Find all DTOs
|
|
for root, dirs, files in os.walk('.'):
|
|
if '\\dto\\' in root or '/dto/' in root:
|
|
for file in files:
|
|
if file.endswith('.java') and not file.endswith('Request.java') and not file.endswith('Response.java') and file != 'SampleGlowMessage.java':
|
|
process_file(os.path.join(root, file))
|