Viewing file:  main.py (2 KB) -rwxr-xr-x
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
#!/opt/imunify360/venv/bin/python
import argparse
import pathlib
import subprocess
import sys
import logging
from country import migrate_country_codes
from smtp import migrate_smtp
from ips import migrate_ips
from logger_config import setup_logger, capture_exception
def is_csf_running() -> bool:
csf_app = "/usr/sbin/csf"
if not pathlib.Path(csf_app).exists():
return False
return (
subprocess.run(
[csf_app, "--status"], capture_output=True, text=True
).returncode
== 0
)
def csf_conf_exists() -> bool:
return pathlib.Path("/etc/csf/csf.conf").exists()
def should_migrate(logger: logging.Logger) -> bool:
if not csf_conf_exists():
logger.error(
"CSF configuration file does not exist, skipping migration"
)
return False
return True
def parse_args():
"""Parse command line arguments."""
parser = argparse.ArgumentParser(
description="Migrate CSF configuration to Imunify360"
)
parser.add_argument(
"--debug", action="store_true", help="Enable debug logging output"
)
return parser.parse_args()
if __name__ == "__main__":
args = parse_args()
# Setup logger with debug flag (also initializes Sentry)
logger = setup_logger(debug_enabled=args.debug)
if not should_migrate(logger):
sys.exit(1)
try:
logger.info("Starting CSF to Imunify migration")
migrate_country_codes()
migrate_smtp()
migrate_ips()
logger.info("CSF to Imunify migration done")
if is_csf_running():
logger.warning("""
###################################################
CSF is running, please stop it by running:
csf -x
systemctl disable csf
systemctl disable lfd
###################################################""")
except Exception as e:
logger.error(f"Fatal error during migration: {e}")
capture_exception(
e, {"migration_phase": "main", "script": "migrate_csf"}
)
sys.exit(1)
|