bpmcdevitt
a6fb367ed4
- Extract database models from monolithic main.py (2,373 lines) into organized modules - Implement service layer pattern with dedicated business logic classes - Split API endpoints into modular FastAPI routers by functionality - Add centralized configuration management with environment variable handling - Create proper separation of concerns across data, service, and presentation layers **Architecture Changes:** - models/: SQLAlchemy database models (CVE, SigmaRule, RuleTemplate, BulkProcessingJob) - config/: Centralized settings and database configuration - services/: Business logic (CVEService, SigmaRuleService, GitHubExploitAnalyzer) - routers/: Modular API endpoints (cves, sigma_rules, bulk_operations, llm_operations) - schemas/: Pydantic request/response models **Key Improvements:** - 95% reduction in main.py size (2,373 → 120 lines) - Updated 15+ backend files with proper import structure - Eliminated circular dependencies and tight coupling - Enhanced testability with isolated service components - Better code organization for team collaboration **Backward Compatibility:** - All API endpoints maintain same URLs and behavior - Zero breaking changes to existing functionality - Database schema unchanged - Environment variables preserved 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
58 lines
No EOL
1.7 KiB
Python
58 lines
No EOL
1.7 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Script to delete all SIGMA rules from the database
|
|
This will clear existing rules so they can be regenerated with the improved LLM client
|
|
"""
|
|
|
|
from models import SigmaRule, SessionLocal
|
|
import logging
|
|
|
|
# Setup logging
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
def delete_all_sigma_rules():
|
|
"""Delete all SIGMA rules from the database"""
|
|
|
|
db = SessionLocal()
|
|
|
|
try:
|
|
# Count existing rules
|
|
total_rules = db.query(SigmaRule).count()
|
|
logger.info(f"Found {total_rules} SIGMA rules in database")
|
|
|
|
if total_rules == 0:
|
|
logger.info("No SIGMA rules to delete")
|
|
return 0
|
|
|
|
# Delete all SIGMA rules
|
|
logger.info("Deleting all SIGMA rules...")
|
|
deleted_count = db.query(SigmaRule).delete()
|
|
db.commit()
|
|
|
|
logger.info(f"✅ Successfully deleted {deleted_count} SIGMA rules")
|
|
|
|
# Verify deletion
|
|
remaining_rules = db.query(SigmaRule).count()
|
|
logger.info(f"Remaining rules in database: {remaining_rules}")
|
|
|
|
return deleted_count
|
|
|
|
except Exception as e:
|
|
logger.error(f"Error deleting SIGMA rules: {e}")
|
|
db.rollback()
|
|
raise
|
|
finally:
|
|
db.close()
|
|
|
|
if __name__ == "__main__":
|
|
print("🗑️ Deleting all SIGMA rules from database...")
|
|
print("This will allow regeneration with the improved LLM client.")
|
|
|
|
deleted_count = delete_all_sigma_rules()
|
|
|
|
if deleted_count > 0:
|
|
print(f"\n🎉 Successfully deleted {deleted_count} SIGMA rules!")
|
|
print("You can now regenerate them with the fixed LLM prompts.")
|
|
else:
|
|
print("\n✅ No SIGMA rules were found to delete.") |