285 lines
12 KiB
Python
285 lines
12 KiB
Python
#coding:utf-8
|
||
|
||
import requests
|
||
import re
|
||
import pytesseract
|
||
from PIL import Image
|
||
from tempfile import NamedTemporaryFile
|
||
|
||
import redis
|
||
|
||
def vercode(cookie=None):
|
||
urls = 'https://cp.baycheer.com:4433/index.php?main=login&act=vercode'#图片链接（每秒更新，距离当前时间最近的时间戳最近为可使用图片）
|
||
|
||
|
||
|
||
|
||
if cookie is not None:
|
||
haders = {
|
||
'Cookie': cookie #使用之前的图片链接Cooike
|
||
}
|
||
res = requests.post(url=urls,headers=haders)
|
||
ress = re.findall('Cookie (.*?) for',str(res.cookies))#获取图片链接Cooike，图片Cooike必须和登录请求的Cooike保持一致
|
||
f1 = NamedTemporaryFile(mode='wb+',suffix='.png')
|
||
f1.write(res.content)#导入文件路径（待处理图片）
|
||
f1.seek(0)
|
||
img = Image.open(f1)#未处理图片
|
||
lim = img.convert('L')
|
||
threshold = 165 # 灰度阈值设为165(可调整)，低于这个值的点全部填白色
|
||
table = []
|
||
for j in range(256):
|
||
if j < threshold:
|
||
table.append(1)
|
||
else:
|
||
table.append(0)
|
||
bim = lim.point(table, '1')#连接各个且分片儿
|
||
f2 = NamedTemporaryFile(mode='wb+',suffix='.png')
|
||
bim.save(f2)#处理完成导入
|
||
f2.seek(0)
|
||
text = pytesseract.image_to_string(Image.open(f2))#开始识别
|
||
return text.replace('\n',''),
|
||
|
||
def Vc(user='honghuayuan',pswd= 'a12345'):
|
||
resetcookie = 0
|
||
try :
|
||
r = redis.StrictRedis(host='192.168.100.44', port=7379, db=11,password="123456")
|
||
viewcookie = r.get('cpmasosessid'+user)
|
||
url="https://cp.baycheer.com/index.php?main=panel"#主页
|
||
header={'Cookie':viewcookie.decode('utf8')}
|
||
resp = requests.get(url=url,headers=header)
|
||
exists = re.findall(r'欢迎进入本公司后台管理系统',resp.text)
|
||
if exists != []:
|
||
return r.get('cpmasosessid'+user).decode('utf8')
|
||
|
||
else :
|
||
resetcookie = 1
|
||
except:
|
||
resetcookie = 1
|
||
|
||
|
||
while resetcookie ==1:
|
||
urls = 'https://cp.baycheer.com:4433/index.php?main=login&act=vercode'#图片链接（每秒更新，距离当前时间最近的时间戳最近为可使用图片）
|
||
res = requests.post(url=urls)
|
||
ress = re.findall('Cookie (.*?) for',str(res.cookies))#获取图片链接Cooike，图片Cooike必须和登录请求的Cooike保持一致
|
||
f1 = NamedTemporaryFile(mode='wb+',suffix='.png')
|
||
f1.write(res.content)#导入文件路径（待处理图片）
|
||
f1.seek(0)
|
||
img = Image.open(f1)#未处理图片
|
||
lim = img.convert('L')
|
||
threshold = 165 # 灰度阈值设为165(可调整)，低于这个值的点全部填白色
|
||
table = []
|
||
for j in range(256):
|
||
if j < threshold:
|
||
table.append(1)
|
||
else:
|
||
table.append(0)
|
||
bim = lim.point(table, '1')#连接各个且分片儿
|
||
f2 = NamedTemporaryFile(mode='wb+',suffix='.png')
|
||
bim.save(f2)#处理完成导入
|
||
f2.seek(0)
|
||
text = pytesseract.image_to_string(Image.open(f2))#开始识别
|
||
url = 'https://cp.baycheer.com:4433/index.php?main=login&act=check'
|
||
haders = {
|
||
'Cookie': ress[0] #使用之前的图片链接Cooike
|
||
}
|
||
data = {
|
||
'user': user,
|
||
'pswd': pswd,
|
||
'vercode': text.replace('\n',''), #识别后的验证码结果
|
||
'issub': '0',
|
||
'subuser': ''
|
||
}
|
||
trylogin = requests.post(url=url,headers=haders,data=data).text# 尝试使用验证码登录
|
||
if trylogin != 'success':
|
||
print('验证码错误正在尝试重新识别验证码')
|
||
continue
|
||
else:
|
||
try:
|
||
r.set("cpmasosessid"+user,ress[0])
|
||
except:
|
||
pass
|
||
resetcookie = 0
|
||
return ress[0]
|
||
break
|
||
|
||
def erp_cookie(type ='erp'):
|
||
r = redis.StrictRedis(host='192.168.100.44', port=7379, db=11,password="123456")
|
||
if type == 'erp':
|
||
cookie=r.get('erpcookie').decode('utf-8')
|
||
url = "https://erp.baycheer.com/"
|
||
header = {
|
||
'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.0.0 Safari/537.36'
|
||
}
|
||
header['cookie']=cookie
|
||
f = requests.get(url=url,headers=header)
|
||
ned = re.findall('企业微信扫码登录',f.text)
|
||
if ned == ['企业微信扫码登录']:
|
||
mainUrl = "https://erp.baycheer.com/login/"#地址
|
||
service=Service( r'E:\chromedriver.exe')#谷歌驱动路劲
|
||
service.command_line_args()
|
||
service.start()
|
||
options = webdriver.ChromeOptions()
|
||
options.add_argument('--headless')
|
||
options.add_argument('ignore-certificate-errors')
|
||
options.add_argument('--start-maximized')
|
||
options.add_argument('--no-sandbox')
|
||
options.add_argument('--disable-gpu')
|
||
options.add_argument(' -port=9222')
|
||
driver=webdriver.Remote(service.service_url,options=options)
|
||
driver.file_detector = UselessFileDetector()##################
|
||
driver.get(mainUrl)
|
||
input_box=driver.find_element(By.ID,'uid')
|
||
input_pswd=driver.find_element(By.ID,'password')
|
||
login_box=driver.find_element(By.CLASS_NAME,'btn')
|
||
input_box.send_keys('10818')
|
||
input_pswd.send_keys('taaw0NCl')
|
||
login_box.click()
|
||
cookie ="sess=" + driver.get_cookie('sess')['value']
|
||
r.set('erpcookie',cookie)
|
||
driver.close()
|
||
elif type == 'bayshop':
|
||
cookie=r.get('bayshopcookie').decode('utf-8')
|
||
url = "https://shop.baycheer.com/"
|
||
header = {
|
||
'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.0.0 Safari/537.36'
|
||
}
|
||
header['cookie']=cookie
|
||
f = requests.get(url=url,headers=header)
|
||
ned = re.findall('请通过公司系统漫游登录',f.text)
|
||
if ned == ['请通过公司系统漫游登录']:
|
||
mainUrl = "https://erp.baycheer.com/login/"#地址
|
||
ssoUrl ='https://erp.baycheer.com/sso'
|
||
sceUrl = 'https://erp.baycheer.com/sso/remoteLogin?sso_site_id=3642'
|
||
thrUrl = 'https://shop.baycheer.com/'
|
||
service=Service('/usr/bin/chromedriver')#谷歌驱动路劲
|
||
service.command_line_args()
|
||
service.start()
|
||
options = webdriver.ChromeOptions()
|
||
options.add_argument('--headless')
|
||
options.add_argument('user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/79.0.3945.79 Safari/537.36')
|
||
options.add_argument('ignore-certificate-errors')
|
||
options.add_argument('--start-maximized')
|
||
options.add_argument('--no-sandbox')
|
||
options.add_argument('--disable-gpu')
|
||
options.add_argument(' -port=9222')
|
||
driver=webdriver.Remote(service.service_url,options=options)
|
||
driver.file_detector = UselessFileDetector()##################
|
||
driver.get(mainUrl)
|
||
input_box=driver.find_element(By.ID,'uid')
|
||
input_pswd=driver.find_element(By.ID,'password')
|
||
login_box=driver.find_element(By.CLASS_NAME,'btn')
|
||
input_box.send_keys('10818')
|
||
input_pswd.send_keys('taaw0NCl')
|
||
login_box.click()
|
||
cookie ="sess=" + driver.get_cookie('sess')['value']
|
||
r.set('erpcookie',cookie)
|
||
driver.get(ssoUrl)
|
||
driver.get(sceUrl)
|
||
import time
|
||
time.sleep(5)
|
||
driver.get(thrUrl)
|
||
cookie ="sess=" + driver.get_cookie('sess')['value']
|
||
r.set('bayshopcookie',cookie)
|
||
driver.quit()
|
||
r.close()
|
||
return cookie
|
||
def zenid():
|
||
r = redis.StrictRedis(host='192.168.100.44', port=7379, db=11,password="123456")
|
||
zenAdminID=r.get('zenAdminID').decode('utf-8')
|
||
url = 'https://www.beautifulhalo.com/masodress/index.php'
|
||
header = {'Cookie': "zenAdminID={};".format(zenAdminID)}
|
||
res = requests.get(url=url, headers=header).text
|
||
retry =0
|
||
if res == 'please use OA remote login this system. if you have any question, please contact with KING.' :
|
||
retry =1
|
||
if 'Checking if the site connection is secure' in res:
|
||
retry =1
|
||
if retry ==1:
|
||
t=Vc()[7:]
|
||
service=Service('/usr/bin/chromedriver')
|
||
service.command_line_args()
|
||
service.start()
|
||
options = webdriver.ChromeOptions()
|
||
options.add_argument('--headless')
|
||
options.add_argument('--no-sandbox')
|
||
options.add_argument('--disable-gpu')
|
||
options.add_argument(' -port=9222')
|
||
driver=webdriver.Remote(service.service_url,options=options)
|
||
mainurl='http://cp.baycheer.com/index.php?main=main'
|
||
panelUrl = "http://cp.baycheer.com/index.php?main=panel"#地址
|
||
remoteurl='http://cp.baycheer.com/index.php?main=sys_remote_login&act=remotelogin&id=303'
|
||
driver.get(mainurl)
|
||
driver.delete_all_cookies()
|
||
driver.add_cookie({'name':'sessid','value':f'{t}'})
|
||
driver.get(panelUrl)
|
||
driver.get(remoteurl)
|
||
zenAdminID=re.findall(r'zenAdminID=(.*)$',driver.current_url)[0]
|
||
driver.quit()
|
||
r.set('zenAdminID',zenAdminID)
|
||
return zenAdminID
|
||
|
||
|
||
def twtoken():
|
||
"""
|
||
先获取ACCESS token
|
||
如果ACCESS TOKEN存在且过期时间没到，则不管他，如果到了则重新获取一次，考虑刷新机制
|
||
"""
|
||
import sys
|
||
import pendulum
|
||
sys.path.append('E:/python/workspace/taotian')
|
||
import iop
|
||
from urllib.parse import quote
|
||
import webbrowser
|
||
appkey = '501176'
|
||
appSecret = '9evPt00JW5Z63WXUsjOYlKjPqhkGkBjX'
|
||
def escape_special_characters(url):
|
||
return quote(url, safe=':/?&=') # 允许保留 :/?&= 这些符号
|
||
|
||
r = redis.StrictRedis(host='192.168.100.44', port=7379, db=11,password="123456")
|
||
tokenbody = r.hgetall('twtoken')
|
||
data = {k.decode('utf-8'): v.decode('utf-8') for k, v in tokenbody.items()}
|
||
data['expire_date'] = "2024-05-25"
|
||
if pendulum.parse(data['expire_date']) >= pendulum.now():
|
||
return data["access_token"]
|
||
|
||
elif pendulum.parse(data['refresh_expire_date']) >= pendulum.now():
|
||
url = 'https://api.taobao.global/rest'
|
||
client = iop.IopClient(url, appkey ,appSecret)
|
||
request = iop.IopRequest('/auth/token/refresh')
|
||
request.add_api_param('refresh_token', data['refresh_token'])
|
||
response = client.execute(request)
|
||
for key in response.body:
|
||
if key in data:
|
||
data[key] = response.body[key]
|
||
return data['access_token']
|
||
|
||
else:
|
||
"""
|
||
重新授权
|
||
"""
|
||
redirect_uri = 'http://region-42.seetacloud.com:40555/post'
|
||
original_url = f'https://api.taobao.global/oauth/authorize?response_type=code&redirect_uri={redirect_uri}&force_auth=true&client_id={appkey}'
|
||
formatted_url = escape_special_characters(original_url)
|
||
webbrowser.open_new(formatted_url)
|
||
webbrowser.open_new_tab(formatted_url)
|
||
print("账号: 贝勤国际有限公司:chanpin01\r\n密码: cp123456")
|
||
code = input("输入获得的code:")
|
||
|
||
request = iop.IopRequest('/auth/token/create')
|
||
request.add_api_param('refresh_token', code)
|
||
response = client.execute(request)
|
||
data = response.body
|
||
data['expire_date']=pendulum.now().add(seconds=int(data['expires_in'])).strftime("%Y-%m-%d")
|
||
data['refresh_expire_date']=pendulum.now().add(seconds=int(data['refresh_expires_in'])).strftime("%Y-%m-%d")
|
||
r.hmset('twtoken',data)
|
||
return data['access_token']
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
if __name__=='__main__':
|
||
s = vercode()
|
||
print(s) |