#!/usr/bin/python3import os
import platform
import argparse
import shutil# 自定义变量
default_system = "ubuntu"
default_redis_version = "6.2.6"
default_install_path = "/usr/local/redis"
default_local_package_dir = os.path.dirname(os.path.abspath(__file__))def stop_firewall():if platform.system() == "Linux":os.system("systemctl stop firewalld")def disable_selinux(system):if system == "centos":os.system("setenforce 0")def install_redis_dependencies(system):if system == "ubuntu":os.system("apt-get -y update")os.system("apt-get install -y build-essential tcl wget")elif system == "centos":os.system("yum -y  update")os.system("yum install -y build-essential tcl wget")else:print(f"Unsupported system: {system}")    def download_redis(version, package_path):if not os.path.exists(os.path.join(package_path, f"redis-{version}.tar.gz")):os.system(f"wget http://download.redis.io/releases/redis-{version}.tar.gz -P {package_path}")def extract_redis(version):os.system(f"tar xvf redis-{version}.tar.gz")def compile_and_install_redis(package_path,version,install_path):os.chdir(f"redis-{version}")os.system(f"make PREFIX={install_path} install")shutil.copy(os.path.join(package_path, f"redis-{version}","redis.conf"),os.path.join(install_path, "redis.conf"))def create_redis_service_file(install_path):redis_service_content = f"""[Unit]
Description=Redis In-Memory Data Store
After=network.target[Service]
Type=forking
ExecStart={install_path}/bin/redis-server {install_path}/redis.conf
ExecStop={install_path}/bin/redis-cli shutdown
Restart=always[Install]
WantedBy=multi-user.target
"""with open("/etc/systemd/system/redis.service", "w") as service_file:service_file.write(redis_service_content)os.system("sudo systemctl daemon-reload")def set_redis_autostart(system):if system == "ubuntu":os.system(f"sudo systemctl enable redis")elif system == "centos":os.system(f"sudo chkconfig redis on")os.system(f"sudo systemctl enable redis")else:print(f"Unsupported system: {system}")
def modify_redis_config(install_path):config_file_path = os.path.join(install_path, "redis.conf")with open(config_file_path, "r") as config_file:config_lines = config_file.readlines()updated_config_lines = []for line in config_lines:if line.startswith("bind"):updated_config_lines.append("bind 0.0.0.0\n")  # Change bind option to bind 0.0.0.0else:updated_config_lines.append(line)with open(config_file_path, "w") as config_file:config_file.writelines(updated_config_lines)def print_installation_info(redis_version, install_path):print("Redis installation completed successfully!")print("Installation details:")print(f"- Redis version: {redis_version}")print(f"- Install path: {install_path}")print(f"- Redis configuration file: {os.path.join(install_path, 'redis.conf')}")print(f"- Redis start command: redis-server")print("Remember to start Redis using the above command.")def main():parser = argparse.ArgumentParser(description="Compile and install Redis with custom options")parser.add_argument("-s", "--system", choices=["ubuntu", "centos"], default=default_system, help="Choose the system (ubuntu or centos)")parser.add_argument("-v", "--version", help="Specify Redis version")parser.add_argument("-p", "--path", help="Specify install path")parser.add_argument("-l", "--local-package", help="Specify local package path")args = parser.parse_args()system = args.system if args.system else default_systemredis_version = args.version if args.version else default_redis_versioninstall_path = args.path if args.path else default_install_pathlocal_package_path = args.local_package if args.local_package else default_local_package_dirstop_firewall()disable_selinux(system)install_redis_dependencies(system)download_redis(redis_version, local_package_path)extract_redis(redis_version)compile_and_install_redis(local_package_path,redis_version,install_path)create_redis_service_file(install_path)set_redis_autostart(system)modify_redis_config(install_path)print_installation_info(redis_version, install_path)if __name__ == "__main__":main()