#!/usr/bin/env python3
"""
脚本名称: backup.py
描述: 系统备份脚本
作者: SRE Team
日期: 2026-04-27
版本: 1.0.0
依赖: requests, PyYAML
"""

import os
import sys
import logging
from datetime import datetime

# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    filename='/var/log/backup.log'
)
logger = logging.getLogger(__name__)

def backup_file(source, destination):
    """
    备份文件
    
    Args:
        source: 源文件路径
        destination: 目标文件路径
    
    Returns:
        bool: 备份是否成功
    
    Raises:
        FileNotFoundError: 源文件不存在
        PermissionError: 权限不足
    """
    try:
        logger.info(f"开始备份: {source} -> {destination}")
        
        # 检查源文件
        if not os.path.exists(source):
            raise FileNotFoundError(f"源文件不存在: {source}")
        
        # 确保目标目录存在
        os.makedirs(os.path.dirname(destination), exist_ok=True)
        
        # 复制文件
        with open(source, 'rb') as src, open(destination, 'wb') as dst:
            dst.write(src.read())
        
        logger.info(f"备份成功: {source} -> {destination}")
        return True
        
    except Exception as e:
        logger.error(f"备份失败: {str(e)}")
        raise

import yaml
import os

def load_config(config_file='config.yml'):
    """加载配置文件"""
    try:
        with open(config_file, 'r', encoding='utf-8') as f:
            config = yaml.safe_load(f)
        return config
    except Exception as e:
        logger.error(f"加载配置文件失败: {str(e)}")
        raise

# 全局配置
CONFIG = load_config()

import argparse

def parse_args():
    """解析命令行参数"""
    parser = argparse.ArgumentParser(
        description='系统备份脚本',
        formatter_class=argparse.RawTextHelpFormatter
    )
    
    parser.add_argument(
        '-c', '--config',
        default='config.yml',
        help='配置文件路径 (默认: config.yml)'
    )
    
    parser.add_argument(
        '-d', '--dry-run',
        action='store_true',
        help='模拟运行，不执行实际备份'
    )
    
    parser.add_argument(
        '-v', '--verbose',
        action='store_true',
        help='详细模式'
    )
    
    parser.add_argument(
        '-f', '--force',
        action='store_true',
        help='强制备份，覆盖已有文件'
    )
    
    return parser.parse_args()

# 示例
if __name__ == '__main__':
    args = parse_args()
    logger.info(f"配置文件: {args.config}")
    logger.info(f"模拟运行: {args.dry_run}")
    logger.info(f"详细模式: {args.verbose}")
    logger.info(f"强制备份: {args.force}")

#!/usr/bin/env python3
"""
脚本名称: wechat_alert.py
描述: 企业微信告警通知脚本
"""

import requests
import json
import sys
import logging

# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)

def send_wechat(webhook_url, alarm_title, alarm_content, mentioned_list=None):
    """
    发送企业微信告警
    
    Args:
        webhook_url: 企业微信机器人webhook URL
        alarm_title: 告警标题
        alarm_content: 告警内容
        mentioned_list: 提及的用户列表
    
    Returns:
        dict: 响应结果
    """
    try:
        payload = {
            "msgtype": "markdown",
            "markdown": {
                "content": f"### {alarm_title}\n>{alarm_content}\n>来源：SRE监控系统"
            }
        }
        
        if mentioned_list:
            payload["markdown"]["mentioned_list"] = mentioned_list
        
        headers = {
            "Content-Type": "application/json"
        }
        
        logger.info(f"发送微信告警: {alarm_title}")
        resp = requests.post(
            webhook_url,
            json=payload,
            headers=headers,
            timeout=10
        )
        
        resp.raise_for_status()
        result = resp.json()
        
        if result.get("errcode") == 0:
            logger.info("微信告警发送成功")
        else:
            logger.error(f"微信告警发送失败: {result.get('errmsg')}")
        
        return result
        
    except Exception as e:
        logger.error(f"发送微信告警失败: {str(e)}")
        raise

def send_dingtalk(webhook_url, alarm_title, alarm_content, at_mobiles=None):
    """
    发送钉钉告警
    """
    try:
        payload = {
            "msgtype": "markdown",
            "markdown": {
                "title": alarm_title,
                "text": f"### {alarm_title}\n{alarm_content}\n> 来源：SRE监控系统"
            }
        }
        
        if at_mobiles:
            payload["at"] = {
                "atMobiles": at_mobiles,
                "isAtAll": False
            }
        
        headers = {
            "Content-Type": "application/json"
        }
        
        logger.info(f"发送钉钉告警: {alarm_title}")
        resp = requests.post(
            webhook_url,
            json=payload,
            headers=headers,
            timeout=10
        )
        
        resp.raise_for_status()
        result = resp.json()
        
        if result.get("errcode") == 0:
            logger.info("钉钉告警发送成功")
        else:
            logger.error(f"钉钉告警发送失败: {result.get('errmsg')}")
        
        return result
        
    except Exception as e:
        logger.error(f"发送钉钉告警失败: {str(e)}")
        raise

if __name__ == "__main__":
    if len(sys.argv) < 4:
        print(f"用法: {sys.argv[0]} <webhook_url> <title> <content> [mentions]")
        sys.exit(1)
    
    webhook_url = sys.argv[1]
    title = sys.argv[2]
    content = sys.argv[3]
    mentions = sys.argv[4].split(',') if len(sys.argv) > 4 else None
    
    if 'qyapi.weixin.qq.com' in webhook_url:
        send_wechat(webhook_url, title, content, mentions)
    elif 'oapi.dingtalk.com' in webhook_url:
        send_dingtalk(webhook_url, title, content, mentions)
    else:
        print("不支持的webhook URL")
        sys.exit(1)

#!/usr/bin/env python3
"""
脚本名称: cmdb_sync.py
描述: 服务器信息同步到CMDB
"""

import os
import json
import logging
import requests
from datetime import datetime

# 配置日志
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    filename='/var/log/cmdb_sync.log'
)
logger = logging.getLogger(__name__)

class CMDBConnector:
    """CMDB连接器"""
    
    def __init__(self, api_url, api_key):
        self.api_url = api_url
        self.api_key = api_key
        self.headers = {
            "Content-Type": "application/json",
            "Authorization": f"Bearer {api_key}"
        }
    
    def get_server(self, server_id):
        """获取服务器信息"""
        url = f"{self.api_url}/servers/{server_id}"
        try:
            resp = requests.get(url, headers=self.headers, timeout=10)
            resp.raise_for_status()
            return resp.json()
        except Exception as e:
            logger.error(f"获取服务器信息失败: {str(e)}")
            return None
    
    def create_server(self, server_data):
        """创建服务器"""
        url = f"{self.api_url}/servers"
        try:
            resp = requests.post(url, json=server_data, headers=self.headers, timeout=10)
            resp.raise_for_status()
            return resp.json()
        except Exception as e:
            logger.error(f"创建服务器失败: {str(e)}")
            return None
    
    def update_server(self, server_id, server_data):
        """更新服务器信息"""
        url = f"{self.api_url}/servers/{server_id}"
        try:
            resp = requests.put(url, json=server_data, headers=self.headers, timeout=10)
            resp.raise_for_status()
            return resp.json()
        except Exception as e:
            logger.error(f"更新服务器信息失败: {str(e)}")
            return None

def get_server_info():
    """获取本地服务器信息"""
    server_info = {}
    
    # 获取主机名
    server_info['hostname'] = os.uname().nodename
    
    # 获取IP地址
    import socket
    server_info['ip_address'] = socket.gethostbyname(socket.gethostname())
    
    # 获取系统信息
    server_info['os'] = os.uname().sysname
    server_info['os_version'] = os.uname().release
    
    # 获取CPU信息
    try:
        with open('/proc/cpuinfo', 'r') as f:
            cpu_info = f.read()
        server_info['cpu_model'] = cpu_info.split('model name')[1].split('\n')[0].split(':')[1].strip()
        server_info['cpu_cores'] = str(os.cpu_count())
    except Exception as e:
        logger.error(f"获取CPU信息失败: {str(e)}")
    
    # 获取内存信息
    try:
        with open('/proc/meminfo', 'r') as f:
            mem_info = f.read()
        total_mem = mem_info.split('MemTotal:')[1].split('\n')[0].split(':')[1].strip()
        server_info['memory'] = total_mem
    except Exception as e:
        logger.error(f"获取内存信息失败: {str(e)}")
    
    # 获取磁盘信息
    try:
        import subprocess
        df_output = subprocess.check_output(['df', '-h'], text=True)
        server_info['disk_info'] = df_output
    except Exception as e:
        logger.error(f"获取磁盘信息失败: {str(e)}")
    
    # 获取时间戳
    server_info['last_sync'] = datetime.now().isoformat()
    
    return server_info

def main():
    """主函数"""
    logger.info("开始同步服务器信息到CMDB")
    
    # 配置
    API_URL = "https://cmdb.example.com/api/v1"
    API_KEY = "YOUR_API_KEY"
    
    # 初始化CMDB连接器
    cmdb = CMDBConnector(API_URL, API_KEY)
    
    # 获取服务器信息
    server_info = get_server_info()
    logger.info(f"获取服务器信息: {server_info['hostname']}")
    
    # 检查服务器是否存在
    existing_server = cmdb.get_server(server_info['hostname'])
    
    if existing_server:
        # 更新服务器信息
        logger.info("服务器已存在，更新信息")
        result = cmdb.update_server(server_info['hostname'], server_info)
    else:
        # 创建新服务器
        logger.info("服务器不存在，创建新记录")
        result = cmdb.create_server(server_info)
    
    if result:
        logger.info("CMDB同步成功")
    else:
        logger.error("CMDB同步失败")

if __name__ == "__main__":
    main()