91 lines
3.2 KiB
Python
91 lines
3.2 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
Script to complete audit router registration in all remaining services.
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import re
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
BASE_DIR = Path(__file__).parent.parent / "services"
|
|||
|
|
|
|||
|
|
# Services that still need updates (suppliers, pos, training, notification, external, forecasting)
|
|||
|
|
SERVICES = ['suppliers', 'pos', 'training', 'notification', 'external', 'forecasting']
|
|||
|
|
|
|||
|
|
def update_service(service_name):
|
|||
|
|
main_file = BASE_DIR / service_name / "app" / "main.py"
|
|||
|
|
|
|||
|
|
if not main_file.exists():
|
|||
|
|
print(f"⚠️ {service_name}: main.py not found")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
content = main_file.read_text()
|
|||
|
|
modified = False
|
|||
|
|
|
|||
|
|
# Check if audit is already imported
|
|||
|
|
if 'import.*audit' in content or ', audit' in content:
|
|||
|
|
print(f"✓ {service_name}: audit already imported")
|
|||
|
|
else:
|
|||
|
|
# Add audit import - find the from .api or from app.api import line
|
|||
|
|
patterns = [
|
|||
|
|
(r'(from \.api import [^)]+)(\))', r'\1, audit\2'), # Multi-line with parentheses
|
|||
|
|
(r'(from \.api import .+)', r'\1, audit'), # Single line with .api
|
|||
|
|
(r'(from app\.api import [^)]+)(\))', r'\1, audit\2'), # Multi-line with app.api
|
|||
|
|
(r'(from app\.api import .+)', r'\1, audit'), # Single line with app.api
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for pattern, replacement in patterns:
|
|||
|
|
new_content = re.sub(pattern, replacement, content)
|
|||
|
|
if new_content != content:
|
|||
|
|
content = new_content
|
|||
|
|
modified = True
|
|||
|
|
print(f"✓ {service_name}: added audit import")
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
if not modified:
|
|||
|
|
print(f"⚠️ {service_name}: could not find import pattern, needs manual update")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# Check if audit router is already registered
|
|||
|
|
if 'service.add_router(audit.router)' in content:
|
|||
|
|
print(f"✓ {service_name}: audit router already registered")
|
|||
|
|
else:
|
|||
|
|
# Find the last service.add_router line and add audit router after it
|
|||
|
|
lines = content.split('\n')
|
|||
|
|
last_router_index = -1
|
|||
|
|
|
|||
|
|
for i, line in enumerate(lines):
|
|||
|
|
if 'service.add_router(' in line and 'audit' not in line:
|
|||
|
|
last_router_index = i
|
|||
|
|
|
|||
|
|
if last_router_index != -1:
|
|||
|
|
# Insert audit router after the last router registration
|
|||
|
|
lines.insert(last_router_index + 1, 'service.add_router(audit.router)')
|
|||
|
|
content = '\n'.join(lines)
|
|||
|
|
modified = True
|
|||
|
|
print(f"✓ {service_name}: added audit router registration")
|
|||
|
|
else:
|
|||
|
|
print(f"⚠️ {service_name}: could not find router registration pattern, needs manual update")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
if modified:
|
|||
|
|
main_file.write_text(content)
|
|||
|
|
print(f"✅ {service_name}: updated successfully")
|
|||
|
|
else:
|
|||
|
|
print(f"ℹ️ {service_name}: no changes needed")
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
print("Completing audit router registration in remaining services...\n")
|
|||
|
|
|
|||
|
|
success_count = 0
|
|||
|
|
for service in SERVICES:
|
|||
|
|
if update_service(service):
|
|||
|
|
success_count += 1
|
|||
|
|
print()
|
|||
|
|
|
|||
|
|
print(f"\nCompleted: {success_count}/{len(SERVICES)} services updated successfully")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|