Python3获取办公室IP并更新至腾讯云安全组

2024-02-24 08:12

本文主要是介绍Python3获取办公室IP并更新至腾讯云安全组,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

需求来源:
由于云服务器的安全组限制了只能在公司IP下远程连接,每次更新安全组时只能通过手动更新,办公室IP是动态IP,每隔三天变一次IP,时间久了就觉得麻烦,所以在查阅了腾讯云文档后发现有安全组的SDK,故而此脚本应运而生~~🤣🤣🤣

脚本说明

1.通过接口获取办公室公网IP
2.通过获取的IP与旧文件IP比对是否一致,一致则不执行后续操作
3.不一致则发送邮件及企业微信通知,
4.调用腾讯云SDK更新不同账户下办公室IP安全组规则

完整代码示例

# -*- 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 = httpProfileclient = 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 = httpProfileclient = 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_listmail_subject = " Office IP Changed!"# 设置邮件内容msg = MIMEMultipart()msg['From'] = sender_emailmsg['To'] = ','.join(receiver_email)msg['Subject'] = mail_subjectmessage = 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.通过接口获取办公室公网IP2.通过获取的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.获取IPres = requests.get('http://icanhazip.com/)')net_ip = res.text.strip('\n')# 2.获取旧文件IPwith open('office_ip', 'r') as f:file_ip = f.read()# 3.比对IPif 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)

最后

以上完整脚本,有需要可自身场景需求修改。

这篇关于Python3获取办公室IP并更新至腾讯云安全组的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/741495

相关文章

Go语言开发实现查询IP信息的MCP服务器

《Go语言开发实现查询IP信息的MCP服务器》随着MCP的快速普及和广泛应用,MCP服务器也层出不穷,本文将详细介绍如何在Go语言中使用go-mcp库来开发一个查询IP信息的MCP... 目录前言mcp-ip-geo 服务器目录结构说明查询 IP 信息功能实现工具实现工具管理查询单个 IP 信息工具的实现服

JAVA保证HashMap线程安全的几种方式

《JAVA保证HashMap线程安全的几种方式》HashMap是线程不安全的,这意味着如果多个线程并发地访问和修改同一个HashMap实例,可能会导致数据不一致和其他线程安全问题,本文主要介绍了JAV... 目录1. 使用 Collections.synchronizedMap2. 使用 Concurren

C# foreach 循环中获取索引的实现方式

《C#foreach循环中获取索引的实现方式》:本文主要介绍C#foreach循环中获取索引的实现方式,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧... 目录一、手动维护索引变量二、LINQ Select + 元组解构三、扩展方法封装索引四、使用 for 循环替代

Linux下如何使用C++获取硬件信息

《Linux下如何使用C++获取硬件信息》这篇文章主要为大家详细介绍了如何使用C++实现获取CPU,主板,磁盘,BIOS信息等硬件信息,文中的示例代码讲解详细,感兴趣的小伙伴可以了解下... 目录方法获取CPU信息:读取"/proc/cpuinfo"文件获取磁盘信息:读取"/proc/diskstats"文

Vue3组件中getCurrentInstance()获取App实例,但是返回null的解决方案

《Vue3组件中getCurrentInstance()获取App实例,但是返回null的解决方案》:本文主要介绍Vue3组件中getCurrentInstance()获取App实例,但是返回nu... 目录vue3组件中getCurrentInstajavascriptnce()获取App实例,但是返回n

python3 gunicorn配置文件的用法解读

《python3gunicorn配置文件的用法解读》:本文主要介绍python3gunicorn配置文件的使用,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录python3 gunicorn配置文件配置文件服务启动、重启、关闭启动重启关闭总结python3 gun

SpringMVC获取请求参数的方法

《SpringMVC获取请求参数的方法》:本文主要介绍SpringMVC获取请求参数的方法,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下... 目录1、通过ServletAPI获取2、通过控制器方法的形参获取请求参数3、@RequestParam4、@

Python中使用正则表达式精准匹配IP地址的案例

《Python中使用正则表达式精准匹配IP地址的案例》Python的正则表达式(re模块)是完成这个任务的利器,但你知道怎么写才能准确匹配各种合法的IP地址吗,今天我们就来详细探讨这个问题,感兴趣的朋... 目录为什么需要IP正则表达式?IP地址的基本结构基础正则表达式写法精确匹配0-255的数字验证IP地

MySQL更新某个字段拼接固定字符串的实现

《MySQL更新某个字段拼接固定字符串的实现》在MySQL中,我们经常需要对数据库中的某个字段进行更新操作,本文就来介绍一下MySQL更新某个字段拼接固定字符串的实现,感兴趣的可以了解一下... 目录1. 查看字段当前值2. 更新字段拼接固定字符串3. 验证更新结果mysql更新某个字段拼接固定字符串 -

Python获取C++中返回的char*字段的两种思路

《Python获取C++中返回的char*字段的两种思路》有时候需要获取C++函数中返回来的不定长的char*字符串,本文小编为大家找到了两种解决问题的思路,感兴趣的小伙伴可以跟随小编一起学习一下... 有时候需要获取C++函数中返回来的不定长的char*字符串,目前我找到两种解决问题的思路,具体实现如下: