需求来源:
由于云服务器的安全组限制了只能在公司IP下远程连接,每次更新安全组时只能通过手动更新,办公室IP是动态IP,每隔三天变一次IP,时间久了就觉得麻烦,所以在查阅了腾讯云文档后发现有安全组的SDK,故而此脚本应运而生~~🤣🤣🤣
脚本说明
text
1.通过接口获取办公室公网IP
2.通过获取的IP与旧文件IP比对是否一致,一致则不执行后续操作
3.不一致则发送邮件及企业微信通知,
4.调用腾讯云SDK更新不同账户下办公室IP安全组规则
完整代码示例
python3
# -*- coding: utf-8 -*-
import json
import smtplib
import requests
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
from tencentcloud.common import credential
from tencentcloud.common.profile.client_profile import ClientProfile
from tencentcloud.common.profile.http_profile import HttpProfile
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
from tencentcloud.vpc.v20170312 import vpc_client, models
# 查询安全组规则的版本【Version】及入站规则【Ingress】
def update_ip2security_group(SecretId, SecretKey, security_group_id, account_subject, office_ip):
cred = credential.Credential(SecretId, SecretKey)
# 实例化一个http选项,可选的,没有特殊需求可以跳过
httpProfile = HttpProfile()
httpProfile.endpoint = "vpc.tencentcloudapi.com"
# 实例化一个client选项,可选的,没有特殊需求可以跳过
clientProfile = ClientProfile()
clientProfile.httpProfile = httpProfile
client = vpc_client.VpcClient(cred, "ap-guangzhou", clientProfile)
try:
# 实例化一个请求对象,每个接口都会对应一个request对象
req = models.DescribeSecurityGroupPoliciesRequest()
params = {
"SecurityGroupId": security_group_id
}
req.from_json_string(json.dumps(params))
# 返回的resp是一个DescribeSecurityGroupPoliciesResponse的实例,与请求对象对应
resp = client.DescribeSecurityGroupPolicies(req)
# 输出json格式的字符串回包
json_req = resp.to_json_string()
# 获取当前安全组规则的修改版本
security_group_policy_version = json.loads(json_req)['SecurityGroupPolicySet']['Version']
# policy_index = json.loads(json_req)['SecurityGroupPolicySet']['Ingress'][0]['PolicyIndex']
# 日志打印
handle_info(account_subject, security_group_policy_version, office_ip)
print("当前规则如下:\n", json.loads(json_req)['SecurityGroupPolicySet'])
# 根据安全的顺序的索引来替换
replace_security_policy(SecretId, SecretKey, security_group_id, security_group_policy_version, office_ip)
except TencentCloudSDKException as err:
print(err)
def replace_security_policy(SecretId, SecretKey, security_group_id, version, office_ip):
"""
注意:
此方法是基于安全组规则索引来修改规则,如果增加了新的规则,则之前的添加的规则索引都会+1,
考虑到以上SDK规则要求:
1.建议在使用此方法时先記錄上一次的IP信息,根據上一次IP記錄,
2.使要修改的安全组下只有一条规则即可用该方法。
以上办法任选其一!
"""
cred = credential.Credential(SecretId, SecretKey)
# 实例化一个http选项,可选的,没有特殊需求可以跳过
httpProfile = HttpProfile()
httpProfile.endpoint = "vpc.tencentcloudapi.com"
# 实例化一个client选项,可选的,没有特殊需求可以跳过
clientProfile = ClientProfile()
clientProfile.httpProfile = httpProfile
client = vpc_client.VpcClient(cred, "ap-guangzhou", clientProfile)
try:
# 实例化一个请求对象,每个接口都会对应一个request对象
req = models.ReplaceSecurityGroupPolicyRequest()
params = {
"SecurityGroupId": security_group_id,
"SecurityGroupPolicySet": {
"Version": version,
"Ingress": [
{
"PolicyIndex": 0,
"Protocol": "ALL",
"Port": "ALL",
"CidrBlock": office_ip,
"Action": "ACCEPT",
"PolicyDescription": "办公室IP",
"ModifyTime": ""
}
]
}
}
req.from_json_string(json.dumps(params))
# 返回的resp是一个ReplaceSecurityGroupPolicyResponse的实例,与请求对象对应
resp = client.ReplaceSecurityGroupPolicy(req)
# 输出json格式的字符串回包
print(resp.to_json_string())
print("规则修改执行完毕!")
except TencentCloudSDKException as err:
print(err)
def handle_info(account_subject, version, ip_address):
print(f"当前执行修改的账户为:{account_subject},安全组修改版本为:{version},修改后的IP地址为:{ip_address}")
def send_mail(old_ip, new_ip):
"""
smtp_server : 邮件SMTP服务器地址【这里使用的腾讯企业邮件服务器】
sender_email:发件人地址
sender_password:发件人邮箱授权码【可登录网页端安全设置里获取smtp授权码】
receiver_email_list: 收件人邮箱列表
mail_subject:邮件主题
message:邮件内容正文
"""
smtp_server = "smtp.exmail.qq.com"
sender_email = 'mail@qq.com'
sender_password = "xxxxxxxx"
receiver_email_list = ['zhangsan@qq.com', 'lisi@qq.com']
receiver_email = receiver_email_list
mail_subject = " Office IP Changed!"
# 设置邮件内容
msg = MIMEMultipart()
msg['From'] = sender_email
msg['To'] = ','.join(receiver_email)
msg['Subject'] = mail_subject
message = f"The old ip is {old_ip},The new ip is {new_ip} !"
# 邮件正文
msg.attach(MIMEText(message, 'plain'))
# 连接到QQ邮箱的SMTP服务器
try:
server = smtplib.SMTP_SSL(smtp_server, 465)
server.login(sender_email, sender_password)
server.sendmail(sender_email, receiver_email_list, msg.as_string())
print("邮件发送成功")
server.quit()
except Exception as e:
print("邮件发送失败:", e)
def send_wxwork(old_ip, new_ip):
wx_work_hook = "https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee"
notice1 = {
"msgtype": "markdown",
"markdown": {
"content":
"# IP变动通知\n"
f">旧IP:<font color=\"comment\">{old_ip}</font>\n"
f">新IP:<font color=\"info\">{new_ip}</font>\n"
}
}
response = requests.post(wx_work_hook, json=notice1)
print(response.text)
if __name__ == '__main__':
"""
1.通过接口获取办公室公网IP
2.通过获取的IP与旧文件IP比对是否一致,一致则不执行后续操作
3.不一致则发送邮件及企业微信通知,
4.调用腾讯云SDK更新不同账户下办公室IP安全组规则
"""
Secret_dict = {
"aa": {
"account_subject": "阿里巴巴",
"SecretId": "SecretId",
"SecretKey": "SecretKey",
"security_group_id": "security_group_id"
},
"bb": {
"account_subject": "腾讯",
"SecretId": "SecretId",
"SecretKey": "SecretKey",
"security_group_id": "security_group_id"
}
}
aa_secretId = Secret_dict['aa']['SecretId']
aa_secretKey = Secret_dict['aa']['SecretKey']
aa_security_group_id = Secret_dict['aa']['security_group_id']
aa_account_subject = Secret_dict['aa']['account_subject']
bb_secretId = Secret_dict['bb']['SecretId']
bb_secretKey = Secret_dict['bb']['SecretKey']
bb_security_group_id = Secret_dict['bb']['security_group_id']
bb_account_subject = Secret_dict['bb']['account_subject']
# 1.获取IP
res = requests.get('http://icanhazip.com/)')
net_ip = res.text.strip('\n')
# 2.获取旧文件IP
with open('office_ip', 'r') as f:
file_ip = f.read()
# 3.比对IP
if net_ip == file_ip:
print(f"IP一致!当前IP为:{net_ip}")
exit(1)
print("IP不一致")
# 更新IP文件
with open('office_ip', 'w') as f:
f.seek(0) # 调整句柄到开头
f.write(net_ip)
# 发送邮件通知及企业微信通知
send_mail(file_ip, net_ip)
send_wxwork(file_ip, net_ip)
# 修改aa账户办公室IP安全组规则
update_ip2security_group(aa_secretId, aa_secretKey, aa_security_group_id, aa_account_subject, net_ip)
# 修改bb账户办公室IP安全组规则
update_ip2security_group(bb_secretId, bb_secretKey, bb_security_group_id, bb_account_subject, net_ip)
最后
text
以上完整脚本,有需要可自身场景需求修改。