62 lines
2.0 KiB
Python
62 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
"""Fix import statements in all pipeline services"""
|
|
|
|
import os
|
|
import re
|
|
|
|
def fix_imports(filepath):
|
|
"""Fix import statements in a Python file"""
|
|
with open(filepath, 'r') as f:
|
|
content = f.read()
|
|
|
|
# Pattern to match the old import style
|
|
old_pattern = r"# 상위 디렉토리의 shared 모듈 import\nsys\.path\.append\(os\.path\.join\(os\.path\.dirname\(__file__\), '\.\.', 'shared'\)\)\nfrom ([\w, ]+) import ([\w, ]+)"
|
|
|
|
# Replace with new import style
|
|
def replace_imports(match):
|
|
modules = match.group(1)
|
|
items = match.group(2)
|
|
|
|
# Build new import statements
|
|
imports = []
|
|
if 'models' in modules:
|
|
imports.append(f"from shared.models import {items}" if 'models' in modules else "")
|
|
if 'queue_manager' in modules:
|
|
imports.append(f"from shared.queue_manager import QueueManager")
|
|
|
|
return "# Import from shared module\n" + "\n".join(filter(None, imports))
|
|
|
|
# Apply the replacement
|
|
new_content = re.sub(old_pattern, replace_imports, content)
|
|
|
|
# Also handle simpler patterns
|
|
new_content = new_content.replace(
|
|
"sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'shared'))\nfrom models import",
|
|
"from shared.models import"
|
|
)
|
|
new_content = new_content.replace(
|
|
"\nfrom queue_manager import",
|
|
"\nfrom shared.queue_manager import"
|
|
)
|
|
|
|
# Write back if changed
|
|
if new_content != content:
|
|
with open(filepath, 'w') as f:
|
|
f.write(new_content)
|
|
print(f"Fixed imports in {filepath}")
|
|
return True
|
|
return False
|
|
|
|
# Files to fix
|
|
files_to_fix = [
|
|
"monitor/monitor.py",
|
|
"google-search/google_search.py",
|
|
"article-assembly/article_assembly.py",
|
|
"rss-collector/rss_collector.py",
|
|
"ai-summarizer/ai_summarizer.py"
|
|
]
|
|
|
|
for file_path in files_to_fix:
|
|
full_path = os.path.join(os.path.dirname(__file__), file_path)
|
|
if os.path.exists(full_path):
|
|
fix_imports(full_path) |