maxinxin/zhitou_trade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
zhitou_trade/tools.py

1067 lines
33 KiB
Python
Raw Normal View History

first
2025-05-22 16:47:45 +08:00
import socket
import warnings
from hashlib import md5
import requests
from sqlalchemy import create_engine
import pandas as pd
import logging
import time
import sys
import smtplib
from email.header import Header
from email.mime.text import MIMEText
from email.mime.image import MIMEImage
from email.mime.multipart import MIMEMultipart
from email.mime.application import MIMEApplication
import getpass
import os
import uuid
import redis
import ctypes
import platform
import re
import subprocess
import winreg
import random
project_name = 'auto_trade_20230130'
from hashlib import md5
import sys
import datetime
import pandas as pd
import os
from openpyxl import load_workbook
import subprocess
import time
import requests
import requests
import subprocess
import time
# from alibabacloud_tea_openapi import models as open_api_models
# from alibabacloud_credentials.client import Client as CredentialClient
# from alibabacloud_credentials.models import Config as CredentialConfig
# from alibabacloud_ocr_api20210707.client import Client as OcrClient
# from alibabacloud_ocr_api20210707 import models as ocr_models
# from alibabacloud_tea_util import models as util_models
# from alibabacloud_tea_util.client import Client as StreamClient
def connect_to_wifi(ssid='MaxEntropy', password='cskj12345678'):
# 先检测当前网络是否正常
try:
response = requests.get("https://www.baidu.com", timeout=5)
if response.status_code == 200:
print("当前网络连接正常，无需重连WiFi")
return True
except:
print("当前网络不可用，将尝试连接WiFi...")
# 生成 WiFi 配置文件（XML格式）
wifi_config = f"""<?xml version="1.0"?>
<WLANProfile xmlns="http://www.microsoft.com/networking/WLAN/profile/v1">
<name>{ssid}</name>
<SSIDConfig>
<SSID>
<name>{ssid}</name>
</SSID>
</SSIDConfig>
<connectionType>ESS</connectionType>
<connectionMode>auto</connectionMode>
<MSM>
<security>
<authEncryption>
<authentication>WPA2PSK</authentication>
<encryption>AES</encryption>
<useOneX>false</useOneX>
</authEncryption>
<sharedKey>
<keyType>passPhrase</keyType>
<protected>false</protected>
<keyMaterial>{password}</keyMaterial>
</sharedKey>
</security>
</MSM>
</WLANProfile>"""
try:
# 保存配置文件
with open("wifi_profile.xml", "w") as f:
f.write(wifi_config.strip())
# 添加 WiFi 配置文件
subprocess.run(
["netsh", "wlan", "add", "profile", "filename=wifi_profile.xml"],
check=True,
shell=True,
)
# 连接 WiFi
subprocess.run(
["netsh", "wlan", "connect", "name=" + ssid],
check=True,
shell=True,
)
# 等待连接成功
max_retries = 5
for i in range(max_retries):
try:
response = requests.get("https://www.baidu.com", timeout=5)
if response.status_code == 200:
print(f"成功连接到 WiFi: {ssid}")
return True
except:
if i < max_retries - 1:
print(f"等待网络连接...({i + 1}/{max_retries})")
time.sleep(2)
print("WiFi连接失败: 超过最大重试次数")
return False
except subprocess.CalledProcessError as e:
print(f"WiFi配置失败: {e}")
return False
except Exception as e:
print(f"发生未知错误: {e}")
if os.path.exists("wifi_profile.xml"):
os.remove("wifi_profile.xml")
return False
class ExcelDataWriter:
"""
Excel数据写入工具类（定制列名版）
列名顺序：as_of_date, id, account, proxy_ip, ip_location, exit_ip,
ip_check_result, usage_result, ip_switch_time,
transaction_duration, ip_survival_time, remarks,
platform_risk_response
"""
def __init__(self, filename=r'C:\auto_trade_20230130\mysql_table\ip_tracking.xlsx',sheet_name = 'Sheet1'):
"""
初始化Excel写入器
:param filename: Excel文件名（默认ip_tracking.xlsx）
"""
self.filename = filename
self.sheet_name = sheet_name
# 严格按照要求的列名和顺序
self.columns = [
'as_of_date',
'broker',
'id',
'account',
'proxy_ip',
'proxy_port',
'ip_location',
'exit_ip',
'ip_cross_check_result',
'usage_result',
'ip_switch_time',
'transaction_duration',
'ip_survival_time',
'remarks',
'platform_risk_response'
]
# 如果文件不存在，创建带指定列名的空Excel文件
if not os.path.exists(self.filename):
pd.DataFrame(columns=self.columns).to_excel(self.filename, sheet_name=self.sheet_name,index=False)
def write_data(self, data_dict):
"""追加单条数据到 Excel（自动补全缺失列）"""
try:
# 1. 补全缺失列
full_data = {col: data_dict.get(col) for col in self.columns}
# 2. 读取现有数据
if os.path.exists(self.filename):
existing_df = pd.read_excel(self.filename)
# 验证现有列是否匹配
if list(existing_df.columns) != self.columns:
raise ValueError("现有文件的列名不匹配，请使用新文件")
else:
existing_df = pd.DataFrame(columns=self.columns)
# 3. 创建新数据行
new_row = pd.DataFrame([full_data])[self.columns]
# 4. 合并数据
updated_df = pd.concat([existing_df, new_row], ignore_index=True)
# 5. 写入文件
updated_df.to_excel(self.filename, index=False)
print('数据一')
return True
except Exception as e:
print(f"Excel 写入失败: {str(e)}")
return False
def update_latest_record(self, status='成功', remarks='', risk_response=''):
"""
更新最新记录的状态信息
:param status: 使用结果（成功/失败）
:param remarks: 备注信息
:param risk_response: 平台风险响应
:return: 成功返回True，失败返回False
"""
# 读取整个Excel文件
df = pd.read_excel(self.filename, sheet_name=self.sheet_name)
if len(df) == 0:
raise ValueError("Excel文件中没有记录可更新")
# 获取最新记录索引
last_idx = df.index[-1]
# 计算交易持续时间
if pd.notna(df.at[last_idx, 'as_of_date']):
start_time = datetime.datetime.strptime(df.at[last_idx, 'as_of_date'], "%Y-%m-%d %H:%M:%S.%f").timestamp()
transaction_duration = round(time.time() - start_time, 2)
else:
transaction_duration = 0
# 更新字段
df['usage_result'] = df['usage_result'].astype(str)
df['remarks'] = df['remarks'].astype(str)
df['platform_risk_response'] = df['platform_risk_response'].astype(str)
df.at[last_idx, 'usage_result'] = status
df.at[last_idx, 'transaction_duration'] = transaction_duration
df.at[last_idx, 'remarks'] = remarks
df.at[last_idx, 'platform_risk_response'] = risk_response
# 保存回文件
df.to_excel(self.filename, sheet_name=self.sheet_name, index=False)
def get_unavailable_ips(self, broker, account):
"""目标：返回指定券商（brokerhao）下，指定账户（account）不能使用过的代理 IP 列表"""
# 读取整个Excel文件
df = pd.read_excel(self.filename, sheet_name=self.sheet_name)
# 如果数据为空，返回空列表
if len(df) == 0:
return []
# 获取该券商下所有已使用的IP
broker_used_ips = df[df['broker'] == broker]['proxy_ip'].tolist()
# 获取该券商下该账户已使用的IP
account_used_ips = df[(df['broker'] == broker) & (df['account'] == account)]['proxy_ip'].tolist()
# 计算不能使用的IP = 该券商所有已用IP - 该账户已使用的IP
unused_ips = list(set(broker_used_ips) - set(account_used_ips))
return unused_ips
# TempStdoutArea对象为为重定向标准输出准备的空间，以列表形式存在
class TempStdoutArea:
def __init__(self):
self.buffer = []
def write(self, *args, **kwargs):
self.buffer.append(args)
# 实现获取指定控件坐标的对象
class GetControlCoord:
# app为 pywinauto框架实例化的Application对象
def __init__(self, app):
self.app = app
# 获取指定控件坐标的方法，control参数传入指定控件
def get_coord(self, control):
# 重定向标准输出，将原本的标准输出信息写入自定义的空间内，以获取控件信息
stdout = sys.stdout
sys.stdout = TempStdoutArea()
# pywinauto框架提供的print_control_identifiers()方法，可以打印出应用程序的控件信息，其中就包含控件坐标
self.app.print_control_identifiers()
# 将控件信息转移到control_identifiers，并还原标准输出
control_identifiers, sys.stdout = sys.stdout, stdout
# print(control_identifiers.buffer)
all_coord = []
# 遍历控件信息，获取指定控件的坐标
for a_control_identifier in control_identifiers.buffer:
# print(a_control_identifier)
# print(a_control_identifier[0])
a_control_identifier = a_control_identifier[0]
if a_control_identifier.find(control) > -1:
# print(a_control_identifier)
a_coord = []
str_coord = a_control_identifier[
a_control_identifier.find('(') + 1: a_control_identifier.find(')')].split(',')
# print(str_coord)
for i in str_coord:
num = i.strip()[1:]
a_coord.append(num)
all_coord.append(a_coord)
return all_coord
def send_email(subject: str, message: str, receive_users: list = ['1055017575@qq.com', '2293908092@qq.com'],
file_path: str = 'no_path'):
'''
Parameters
----------
subject : str
title.
message : str
message.
receive_users : list, optional
receive. The default is ['1055017575@qq.com'].
file_path : str, optional
path. The default is 'no_path'.
Returns
-------
None.
'''
send_user = 'it_ainvest@zohomail.cn' # zoho
send_pwd = '2022@AIWANTE' # 密码
receive_user = receive_users
msg = MIMEMultipart()
msg['subject'] = subject
msg['from'] = send_user
msg['to'] = ','.join(receive_user)
msg.attach(MIMEText(message, 'html', 'utf-8'))
if file_path != 'no_path':
part = MIMEApplication(open(file_path, 'rb').read())
file_name = file_path.split('/')[-1]
part.add_header('Content-Disposition', 'attachment', filename=file_name)
msg.attach(part)
else:
pass
server = smtplib.SMTP_SSL("smtp.zoho.com.cn", 465)
server.login(send_user, send_pwd)
server.sendmail(send_user, receive_user, msg.as_string())
server.quit()
print('邮件发送成功')
def download_data_from_db(sql, schema_name):
"""
create connection with DB
阿里云数据库连接
:param sql: 查询语句
:param schema_name: 数据库名称
:return: 查询结果数据
"""
cnx = create_engine(
'mysql+pymysql://cn_ainvest_db:cn_ainvest_sd3a1@rm-2zewagytttzk6f24xno.'
'mysql.rds.aliyuncs.com:3306/%s?charset=utf8' %
schema_name, encoding="utf-8", echo=False)
df = pd.read_sql(sql, con=cnx)
return df
# 转换股票代码
def tranTicker(tick):
tick = str(tick)
if len(tick) == 8:
tick = tick.strip('SH').strip('SZ')
if tick.startswith('6') or tick.startswith('11') or tick.startswith('10') or tick.startswith('51'):
tick = 'SH' + tick
elif tick.startswith('0') or tick.startswith('3') or tick.startswith('15') or tick.startswith(
'16') or tick.startswith(
'12'):
tick = 'SZ' + tick
else:
pass
elif len(tick) == 6:
if tick.startswith('6') or tick.startswith('11') or tick.startswith('10') or tick.startswith('51'):
tick = 'SH' + tick
elif tick.startswith('0') or tick.startswith('15') or tick.startswith('12') or tick.startswith(
'16') or tick.startswith('3'):
tick = 'SZ' + tick
else:
pass
else:
num = 6 - len(tick)
tick = 'SZ' + num * '0' + tick
return tick
def ocr_image(image_path):
"""
传入图片路径，返回识别到的文字内容
"""
access_key_id = 'LTAI5tA4dy591hCgEBfsxZei'
access_key_secret = 'edYHXSVYda8HomlEsnsyQzw1QGu1EN'
if not os.path.exists(image_path):
raise FileNotFoundError(f"图片文件不存在: {image_path}")
# 第一步：用新版 credential 初始化
credential_config = CredentialConfig(
type='access_key',
access_key_id=access_key_id,
access_key_secret=access_key_secret
)
credential = CredentialClient(credential_config)
# 第二步：openapi config
config = open_api_models.Config(
credential=credential,
endpoint='ocr-api.cn-hangzhou.aliyuncs.com'
)
# 第三步：初始化 OCR client
client = OcrClient(config)
# 第四步：准备请求参数
body_stream = StreamClient.read_from_file_path(image_path)
request = ocr_models.RecognizeAllTextRequest(
body=body_stream,
type='Advanced'
)
runtime = util_models.RuntimeOptions()
# 第五步：发起请求
response = client.recognize_all_text_with_options(request, runtime)
return response.body.data.content
def get_price(tick, level='1', op="buy"):
level = int(level)
str_ticker = tick.lower()
rep_data = requests.get("http://qt.gtimg.cn/q=" + str_ticker).text
stocks_detail = "".join(rep_data)
stock_detail = stocks_detail.split(";")[0]
if len(stock_detail) < 49:
return
stock = stock_detail.split("~")
buy1 = float(stock[9])
buy2 = float(stock[11])
buy3 = float(stock[13])
buy4 = float(stock[15])
buy5 = float(stock[17])
sell1 = float(stock[19])
sell2 = float(stock[21])
sell3 = float(stock[23])
sell4 = float(stock[25])
sell5 = float(stock[27])
if op == "buy" and level == 1:
return buy1
elif op == "buy" and level == 2:
return buy2
elif op == "buy" and level == 3:
return buy3
elif op == "buy" and level == 4:
return buy4
elif op == "buy" and level == 5:
return buy5
elif op == "sell" and level == 1:
return sell1
elif op == "sell" and level == 2:
return sell2
elif op == "sell" and level == 3:
return sell3
elif op == "sell" and level == 4:
return sell4
elif op == "sell" and level == 5:
return sell5
else:
return float(stock[3])
def get_host_ip():
"""
查询本机ip地址
:return:
"""
s = uuid.UUID(int=uuid.getnode()).hex[-12:]
return s
def tryrun(func):
def wrapper(*args, **kwargs):
while True:
countnum = 0
try:
res = func(*args, **kwargs)
except RuntimeError:
countnum += 1
time.sleep(1)
if countnum > 120:
break
return res
return wrapper
def tryrun2(func):
def wrapper(*args, **kwargs):
while True:
countnum = 0
try:
res = func(*args, **kwargs)
except RuntimeError:
countnum += 1
time.sleep(10)
if countnum > 90:
break
return res
return wrapper
def output(func):
def wrapper(*args, **kwargs):
res = func(*args, **kwargs)
print(args[1])
return res
return wrapper
class Logger():
"""
This is get log class
"""
def __init__(self, path, name, Flevel=logging.DEBUG):
self.time = time.strftime("%Y-%m-%d")
self.fileName = path + '/' + name + ".log"
self.logger = logging.getLogger(self.fileName)
self.logger.setLevel(Flevel)
if not self.logger.handlers:
fmt = logging.Formatter('[%(asctime)s] [%(levelname)s] %(message)s', '%Y-%m-%d %H:%M:%S')
fh = logging.FileHandler(self.fileName, encoding='utf-8')
fh.setFormatter(fmt)
fh.setLevel(Flevel)
self.logger.addHandler(fh)
@output
def debug(self, message):
"""
This is debug
"""
self.logger.debug(message)
@output
def info(self, message):
"""
This is info
"""
self.logger.info(message)
@output
def warn(self, message):
"""
This is warn
"""
self.logger.warn(message)
@output
def error(self, message):
"""
This is error
"""
self.logger.error(message)
@output
def critical(self, message):
"""
This is critical
"""
self.logger.critical(message)
def softInfo(self, message):
"""
This is output
"""
self.logger.info(message)
def init_redis(ip, port1, db1, psw=''):
'''
redis初始化(待修改为密码版本)
Parameters
----------
ip : TYPE
DESCRIPTION.
port1 : TYPE
DESCRIPTION.
db1 : TYPE
DESCRIPTION.
psw : TYPE, optional
DESCRIPTION. The default is ''.
Returns
-------
TYPE
DESCRIPTION.
'''
global r
time.sleep(5)
# r = redis.Redis(host=ip, port=port1, password=psw)
pool = redis.ConnectionPool(host=ip, port=port1, password=psw)
r = redis.Redis(connection_pool=pool)
return r.ping()
def redis_lget(key,num):
'''
redis获取指定key的值
Parameters
----------
key : TYPE
DESCRIPTION.
num : TYPE
DESCRIPTION.
Returns
-------
vec : TYPE
DESCRIPTION.
'''
print('1')
init_redis('82.156.3.152', 6379, 0, '1508181008')
print('2')
vec = r.lrange(key,0,num-1)
print('3')
if len(vec)>0:
print(vec)
return vec
class SetMac(object):
"""
修改 本地连接 mac地址
"""
def __init__(self):
# regex to MAC address like 00-00-00-00-00-00 or 00:00:00:00:00:00 or
# 000000000000
self.MAC_ADDRESS_RE = re.compile(r"""
([0-9A-F]{1,2})[:-]?
([0-9A-F]{1,2})[:-]?
([0-9A-F]{1,2})[:-]?
([0-9A-F]{1,2})[:-]?
([0-9A-F]{1,2})[:-]?
([0-9A-F]{1,2})
""", re.I | re.VERBOSE) # re.I: case-insensitive matching. re.VERBOSE: just look nicer.
self.WIN_REGISTRY_PATH = "SYSTEM\CurrentControlSet\Control\Class\{4D36E972-E325-11CE-BFC1-08002BE10318}"
def is_admin(self):
"""
is user an admin?
:return:
"""
if ctypes.windll.shell32.IsUserAnAdmin() == 0:
print('Sorry! You should run this with administrative privileges if you want to change your MAC address.')
sys.exit()
else:
print('admin')
def get_macinfos(self):
"""
查看所有mac信息
:return:
"""
print('=' * 50)
mac_info = subprocess.check_output('GETMAC /v /FO list', stderr=subprocess.STDOUT)
mac_info = mac_info.decode('gbk')
print('Your MAC address:\n', mac_info)
# 想要匹配的连接名
target_connection_name = "WLAN 2"
# 构建正则表达式模式
pattern = re.compile(
r"连接名:\s+" + re.escape(target_connection_name) + r".*?物理地址:\s+([0-9A-Fa-f-]+)",
re.DOTALL
)
# 执行搜索
match = pattern.search(mac_info).group(1)
return match
def get_target_device(self):
"""
返回 本地连接 网络适配器
:return:
"""
mac_info = subprocess.check_output('GETMAC /v /FO list', stderr=subprocess.STDOUT)
mac_info = mac_info.decode('gbk')
print(mac_info)
search = re.search(r'(WLAN 2)\s+网络适配器: (.+)\s+物理地址:', mac_info)
print(search)
target_name, target_device = (search.group(1), search.group(2).strip()) if search else ('', '')
if not all([target_name, target_device]):
print('Cannot find the target device')
sys.exit()
print(target_name, target_device)
return target_device
def get_network_adapter_info(self):
"""
获取网络适配器信息
re:
{'16': 'Realtek RTL8821CE 802.11ac PCIe Adapter #2'}
"""
try:
# 执行 wmic 命令并获取输出
cmd = "wmic path win32_networkadapter get index, name"
output = subprocess.check_output(cmd, universal_newlines=True)
# 按换行符分割输出内容
lines = output.strip().split('\n')
# 去除每行前后的空白字符
lines = [line.strip() for line in lines]
# 过滤掉空行
non_empty_lines = [line for line in lines if line]
result = {}
# 从第二行开始遍历
for line in non_empty_lines[1:]:
# 查找第一个空白字符的位置
first_space_index = line.find(' ')
# 提取 index
index = line[:first_space_index].strip()
# 提取 name
name = line[first_space_index:].strip()
result[index] = name
return result
except subprocess.CalledProcessError as e:
print(f"命令执行失败，错误代码: {e.returncode}")
print(e.output)
except Exception as e:
print(f"发生异常: {e}")
return []
def set_mac_address(self, target_device, new_mac):
"""
设置新mac地址
:param target_device: 本地连接 网络适配器
:param new_mac: 新mac地址
:return:
"""
if not self.MAC_ADDRESS_RE.match(new_mac):
print('Please input a correct MAC address')
return
# Locate adapter's registry and update network address (mac)
reg_hdl = winreg.ConnectRegistry(None, winreg.HKEY_LOCAL_MACHINE)
key = winreg.OpenKey(reg_hdl, self.WIN_REGISTRY_PATH)
info = winreg.QueryInfoKey(key)
# Find adapter key based on sub keys
adapter_key = None
adapter_path = None
target_index = -1
adapter_dict = self.get_network_adapter_info()
for index in range(info[0]):
subkey = winreg.EnumKey(key, index)
path = self.WIN_REGISTRY_PATH + "\\" + subkey
if subkey == 'Properties':
break
# Check for adapter match for appropriate interface
new_key = winreg.OpenKey(reg_hdl, path)
try:
adapterIndex = str(int(subkey))
adapterName = adapter_dict.get(adapterIndex)
adapterDesc = winreg.QueryValueEx(new_key, "DriverDesc")
if adapterName == target_device:
adapter_path = path
target_index = adapterIndex
break
else:
winreg.CloseKey(new_key)
except (WindowsError) as err:
if err.errno == 2: # register value not found, ok to ignore
pass
else:
raise err
if adapter_path is None:
print('Device not found.')
winreg.CloseKey(key)
winreg.CloseKey(reg_hdl)
return
# Registry path found update mac addr
adapter_key = winreg.OpenKey(reg_hdl, adapter_path, 0, winreg.KEY_WRITE)
# winreg.SetValueEx(adapter_key, "NetworkAddress", 0, winreg.REG_SZ, -----new_mac)
winreg.SetValueEx(adapter_key, "NetworkAddress", 0, winreg.REG_SZ, new_mac)
winreg.CloseKey(adapter_key)
winreg.CloseKey(key)
winreg.CloseKey(reg_hdl)
# Adapter must be restarted in order for change to take affect
# print 'Now you should restart your netsh'
self.restart_adapter(target_index, target_device)
def restart_adapter(self, target_index, target_device):
"""
Disables and then re-enables device interface
"""
if platform.release() == 'XP':
# description, adapter_name, address, current_address = find_interface(device)
cmd = "devcon hwids =net"
try:
result = subprocess.check_output(cmd, stderr=subprocess.STDOUT)
except FileNotFoundError:
raise
query = '(' + target_device + '\r\n\s*.*:\r\n\s*)PCI\\\\(([A-Z]|[0-9]|_|&)*)'
query = query.encode('ascii')
match = re.search(query, result)
cmd = 'devcon restart "PCI\\' + str(match.group(2).decode('ascii')) + '"'
subprocess.check_output(cmd, stderr=subprocess.STDOUT)
else:
cmd = "wmic path win32_networkadapter where index=" + str(target_index) + " call disable"
subprocess.check_output(cmd)
cmd = "wmic path win32_networkadapter where index=" + str(target_index) + " call enable"
subprocess.check_output(cmd)
def run(self):
self.is_admin()
mac = self.get_macinfos()
warnings.filterwarnings('ignore')
scripts_path = os.path.dirname(os.path.realpath(__file__))
root_path = scripts_path[:scripts_path.find(project_name) + len(project_name)]
log = Logger(f'{root_path}/logs', '18242094506')
log.warn('MAC 重置之前:'+mac)
target_device = self.get_target_device()
# self.set_mac_address(target_device, '82E82CF2158F')
# self.set_mac_address(target_device, '28CDC4CBFDCF')
self.set_mac_address(target_device, self.randomMAC())
mac_new = self.get_macinfos()
warnings.filterwarnings('ignore')
scripts_path = os.path.dirname(os.path.realpath(__file__))
root_path = scripts_path[:scripts_path.find(project_name) + len(project_name)]
log = Logger(f'{root_path}/logs', '18242094506')
log.warn('MAC 重置之后:'+mac_new)
# 28-CD-C4-CB-FD-CF
def randomMAC(self):
# mac = [ 0x52, 0x54, 0x00,
# random.randint(0x00, 0x7f),
# random.randint(0x00, 0xff),
# random.randint(0x00, 0xff) ]
#
# while True:
# one_mac = ''.join(map(lambda x: "%02x" % x, mac)).upper()
# used_mac = redis_lget('Mac', -1)
# used_mac = [i.decode() for i in used_mac]
#
# if one_mac not in used_mac:
# r.rpush('Mac',one_mac)
# break
while True:
first_byte = 0x02
# 从 UUIDv4 获取后 5 个字节，随机性极高
random_bytes = uuid.uuid4().bytes
mac_bytes = [first_byte] + list(random_bytes[1:6])
one_mac = ''.join(f'{b:02X}' for b in mac_bytes)
used_mac = redis_lget('Mac', -1)
used_mac = [i.decode() for i in used_mac]
if one_mac not in used_mac:
r.rpush('Mac', one_mac)
break
print('1111111', one_mac)
return one_mac
def get_host_ip():
"""
查询本机ip地址
:return:
"""
s = uuid.UUID(int=uuid.getnode()).hex[-12:]
return s
def get_auth_type(url):
time1 = int(time.time() * 1000)
url = f'http://192.168.100.1/login.cgi?_={time1}'
response = requests.get(url)
if 'WWW-Authenticate' in response.headers:
return response.headers['WWW-Authenticate']
return None
def parse_auth_params(auth_header):
params = {}
for param in auth_header.split(','):
key, value = param.strip().split('=', 1)
params[key] = value.strip('"')
return params
def hex_md5(data):
return md5(data.encode()).hexdigest()
def generate_cnonce():
rand = random.randint(0, 100000)
date = int(time.time() * 1000)
salt = f"{rand}{date}"
tmp = hex_md5(salt)
return tmp[:16]
def compute_digest_response(username, password, realm, nonce, uri, qop, cnonce, nc):
ha1 = hex_md5(f"{username}:{realm}:{password}")
ha2 = hex_md5(f"GET:{uri}")
response = hex_md5(f"{ha1}:{nonce}:{nc}:{cnonce}:{qop}:{ha2}")
return response
def do_login(username, password, base_url):
url = f"{base_url}/login.cgi"
auth_header = get_auth_type(url)
if not auth_header:
print("Failed to get authentication parameters")
return False
auth_params = parse_auth_params(auth_header)
realm = auth_params.get('Digest realm')
nonce = auth_params.get('nonce')
qop = auth_params.get('qop')
if not all([realm, nonce, qop]):
print("Missing required authentication parameters")
return False
cnonce = generate_cnonce()
nc = '00000001'
uri = '/cgi/protected.cgi'
response = compute_digest_response(username, password, realm, nonce, uri, qop, cnonce, nc)
headers = {
'Authorization': f'Digest username="{username}", realm="{realm}", nonce="{nonce}", uri="{uri}", response="{response}", qop={qop}, nc={nc}, cnonce="{cnonce}"'
}
login_url = f"{base_url}/login.cgi?Action=Digest&username={username}&realm={realm}&nonce={nonce}&response={response}&qop={qop}&cnonce={cnonce}&nc={nc}&temp=asr"
response = requests.get(login_url, headers=headers)
uri = "/cgi/xml_action.cgi"
response = compute_digest_response(username, password, realm, nonce, uri, qop, cnonce, nc)
headers = {
'Authorization': f'Digest username="{username}", realm="{realm}", nonce="{nonce}", uri="{uri}", response="{response}", qop={qop}, nc={nc}, cnonce="{cnonce}"'
}
reset_url = 'http://192.168.100.1/xml_action.cgi?method=get&module=duster&file=reset'
response = requests.get(reset_url, headers=headers)
print(response.content)
if response.status_code == 200:
print("Login successful")
return True
else:
print(f"Login failed: {response.status_code} {response.reason}")
return False
def get_ip():
"""
查询本机ip地址
:return:
"""
try:
requests.get('https://checkip.amazonaws.com').text.strip()
except:
return '60.162.69.53'
def reset_proxy_to_default():
"""恢复系统代理为默认（关闭）"""
key = winreg.OpenKey(
winreg.HKEY_CURRENT_USER,
r"Software\\Microsoft\\Windows\\CurrentVersion\\Internet Settings",
0, winreg.KEY_WRITE
)
try:
winreg.SetValueEx(key, "ProxyEnable", 0, winreg.REG_DWORD, 0)
winreg.SetValueEx(key, "ProxyServer", 0, winreg.REG_SZ, "")
winreg.SetValueEx(key, "ProxyOverride", 0, winreg.REG_SZ, "<local>")
finally:
winreg.CloseKey(key)
internet_set_option = ctypes.windll.Wininet.InternetSetOptionW
internet_set_option(0, 37, 0, 0)
internet_set_option(0, 39, 0, 0)
def reset_wlan():
warnings.filterwarnings('ignore')
scripts_path = os.path.dirname(os.path.realpath(__file__))
root_path = scripts_path[:scripts_path.find(project_name) + len(project_name)]
log = Logger(f'{root_path}/logs', '18242094506')
log.warn('===================')
log.warn('ip 重置前：'+get_ip())
count = 0
while True:
base_url = "http://192.168.100.1"
username = "admin"
password = "admin"
do_login(username, password, base_url)
time.sleep(1)
t1 = time.time()
while True:
try:
requests.get('https://www.baidu.com')
print('网络重连成功,共耗时',time.time()-t1,'s')
time.sleep(3)
print('变更后IP',get_ip())
warnings.filterwarnings('ignore')
scripts_path = os.path.dirname(os.path.realpath(__file__))
root_path = scripts_path[:scripts_path.find(project_name) + len(project_name)]
log = Logger(f'{root_path}/logs', '18242094506')
log.warn('ip 重置后：'+ get_ip())
break
except:
pass
one_ip = get_ip()
used_IP = redis_lget('IP', -1)
used_IP = [i.decode() for i in used_IP]
print('!!!!!!one_ip:', one_ip)
print('!!!!!!used_IP:',used_IP)
if one_ip not in used_IP:
r.rpush('IP',one_ip)
print('！！！！入库了 成功rpush')
break
else:
#IP 重复 记录log
warnings.filterwarnings('ignore')
scripts_path = os.path.dirname(os.path.realpath(__file__))
root_path = scripts_path[:scripts_path.find(project_name) + len(project_name)]
log = Logger(f'{root_path}/logs', '18242094506')
log.warn('IP not reset')
if count ==3:
send_email('重启IP出现问题', one_ip)
exit()
if __name__ == '__main__':
print(get_host_ip())
Reference in New Issue Copy Permalink