Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
西城十七妹
dr_py
提交
38092847
dr_py
项目概览
西城十七妹
/
dr_py
与 Fork 源项目一致
Fork自
晚风拂柳颜 / dr_py
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
dr_py
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
38092847
编写于
8月 31, 2022
作者:
H
hjdhnx
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
去除ddddocr依赖,增加在线验证码接口配置
上级
21d52490
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
167 addition
and
5 deletion
+167
-5
classes/cms.py
classes/cms.py
+2
-1
config.py
config.py
+1
-0
py/flaskOcrDz.py
py/flaskOcrDz.py
+148
-0
utils/encode.py
utils/encode.py
+16
-4
未找到文件。
classes/cms.py
浏览文件 @
38092847
...
...
@@ -38,6 +38,7 @@ class CMS:
self
.
play_disable
=
new_conf
.
get
(
'PLAY_DISABLE'
,
False
)
self
.
retry_count
=
new_conf
.
get
(
'RETRY_CNT'
,
3
)
self
.
lazy_mode
=
new_conf
.
get
(
'LAZYPARSE_MODE'
)
self
.
ocr_api
=
new_conf
.
get
(
'OCR_API'
)
try
:
self
.
vod
=
redirect
(
url_for
(
'vod'
)).
headers
[
'Location'
]
except
:
...
...
@@ -679,7 +680,7 @@ class CMS:
html
=
r
.
text
# print(html)
if
html
.
find
(
'输入验证码'
)
>
-
1
:
cookie
=
verifyCode
(
url
,
self
.
headers
,
self
.
timeout
,
self
.
retry_count
)
cookie
=
verifyCode
(
url
,
self
.
headers
,
self
.
timeout
,
self
.
retry_count
,
self
.
ocr_api
)
# cookie = ''
if
not
cookie
:
return
{
...
...
config.py
浏览文件 @
38092847
...
...
@@ -25,4 +25,5 @@ WALL_PAPER_ENABLE = True # 启用自定义壁纸
WALL_PAPER
=
"https://picsum.photos/1280/720/?blur=10"
# 自定义壁纸,可注释
SUP_PORT
=
9001
# supervisord 服务端口
RETRY_CNT
=
3
# 验证码重试次数
OCR_API
=
'http://192.168.3.224:9000/api/ocr_img'
# 验证码识别接口,传参数data
# {% if config.WALL_PAPER %}"wallpaper":"{{ config.WALL_PAPER }}",{% endif %}
\ No newline at end of file
py/flaskOcrDz.py
0 → 100644
浏览文件 @
38092847
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# File : flaskOcrDz.py
# Author: DaShenHan&道长-----先苦后甜,任凭晚风拂柳颜------
# Date : 2021/11/1
import
json
from
flask
import
Flask
,
jsonify
,
request
,
Response
import
requests
import
ddddocr
ocr
=
ddddocr
.
DdddOcr
()
app
=
Flask
(
__name__
)
app
.
config
[
"JSON_AS_ASCII"
]
=
False
# jsonify返回的中文正常显示
@
app
.
route
(
"/"
,
methods
=
[
'GET'
])
def
index
():
return
'欢迎使用简单验证码文字识别,海阔视界道长专用'
def
hexStringTobytes
(
str
):
str
=
str
.
replace
(
" "
,
""
)
print
(
str
)
return
bytes
.
fromhex
(
str
)
def
bytesToHexString
(
bs
):
return
''
.
join
([
'%02X '
%
b
for
b
in
bs
])
class
LocalOcr
:
def
__init__
(
self
,
file
=
None
,
url
=
'http://127.0.0.1:10000'
):
self
.
file
=
file
self
.
url
=
url
self
.
yzm
=
self
.
yzm_ocr
(
0
)
def
read
(
self
):
return
self
.
yzm
def
yzm_ocr
(
self
,
count
=
0
):
try
:
r
=
requests
.
post
(
self
.
url
,
data
=
self
.
file
,
timeout
=
(
0.5
,
2
))
yzm
=
r
.
text
return
yzm
except
:
if
count
<
3
:
count
+=
1
return
self
.
yzm_ocr
(
count
)
else
:
return
""
def
ocr2
(
hex
):
if
type
(
hex
)
==
list
:
hex
=
''
.
join
(
hex
)
img_bytes
=
hexStringTobytes
(
hex
)
# print(img_bytes)
# with open('1.png','wb+') as f:
# f.write(img_bytes)
dm_url
=
'http://dm.mudery.com:10000'
im_ocr
=
LocalOcr
(
img_bytes
,
dm_url
)
yzm
=
im_ocr
.
read
()
ret
=
{
'msg'
:
'ok'
,
'ret'
:
yzm
,
'code'
:
0
,
'detail'
:
'验证码识别成功'
}
print
(
ret
)
return
jsonify
(
ret
)
def
docr
(
hex
):
if
type
(
hex
)
==
list
:
hex
=
''
.
join
(
hex
)
img_bytes
=
hexStringTobytes
(
hex
)
try
:
img_str
=
img_bytes
.
decode
(
"latin1"
)
if
img_str
.
find
(
'html'
)
>
-
1
:
ret
=
{
'msg'
:
'拜托,我收到你传过来的数据是个网页而不是图片,麻烦解密后把图片的hex给我'
,
'ret'
:
img_str
[:
500
],
'code'
:
-
1
,
'detail'
:
'图片识别失败'
}
return
jsonify
(
ret
)
else
:
res
=
ocr
.
classification
(
img_bytes
)
ret
=
{
'msg'
:
'ok'
,
'ret'
:
res
,
'code'
:
0
,
'detail'
:
'图片识别成功'
}
# print(ret)
return
jsonify
(
ret
)
except
Exception
as
e
:
# print(f'{e}')
ret
=
{
'msg'
:
'error'
,
'ret'
:
f
'
{
e
}
'
,
'code'
:
-
2
,
'detail'
:
'发生了意外的错误'
}
return
ret
@
app
.
route
(
"/api/ocr"
,
methods
=
[
'GET'
,
'POST'
])
def
ocr_fast
():
args
=
{}
try
:
ctp
=
request
.
content_type
if
request
.
method
==
'POST'
:
if
ctp
.
find
(
'application/json'
)
>
-
1
:
try
:
args
=
request
.
json
except
Exception
as
e
:
# args = request.get_data(as_text=True)
args
=
{}
else
:
args
=
request
.
form
elif
request
.
method
==
'GET'
:
args
=
request
.
args
if
not
args
.
get
(
'hex'
):
return
'缺少必传参数:hex!'
except
Exception
as
e
:
return
jsonify
({
'msg'
:
'非法调用'
,
'code'
:
'-1'
})
# print(args.get('hex'))
# return ocr2(args.get('hex'))
return
docr
(
args
.
get
(
'hex'
))
@
app
.
route
(
"/api/hex2img"
,
methods
=
[
'GET'
])
def
ocr_hex2img
():
try
:
args
=
request
.
args
if
not
args
.
get
(
'hex'
):
return
'缺少必传参数:hex!'
except
Exception
as
e
:
return
jsonify
({
'msg'
:
'非法调用'
,
'code'
:
'-1'
})
# print(args.get('hex'))
# return ocr2(args.get('hex'))
hex
=
args
.
get
(
'hex'
)
if
type
(
hex
)
==
list
:
hex
=
''
.
join
(
hex
)
img_bytes
=
hexStringTobytes
(
hex
)
resp
=
Response
(
img_bytes
,
mimetype
=
'image/jpeg'
)
return
resp
@
app
.
route
(
"/api/ocr_img"
,
methods
=
[
'POST'
])
def
ocr_img_fast
():
# print(request.values)
# print(request.files)
# print(request.data)
try
:
img_bytes
=
request
.
data
ret
=
ocr
.
classification
(
img_bytes
)
return
ret
# return jsonify({'ret':ret,'code':0,'msg':'识别完毕'})
except
Exception
as
e
:
return
''
# return jsonify({'msg':'请求出错','code':-1,'detail':f'{e}'})
def
test
():
pic
=
'yzm1.png'
# pic = '2.png'
with
open
(
pic
,
'rb'
)
as
f
:
img_bytes
=
f
.
read
()
res
=
ocr
.
classification
(
img_bytes
)
print
(
res
)
if
__name__
==
'__main__'
:
app
.
run
(
host
=
"0.0.0.0"
,
port
=
9000
)
# test()
\ No newline at end of file
utils/encode.py
浏览文件 @
38092847
...
...
@@ -10,7 +10,6 @@ import requests.utils
from
time
import
sleep
import
os
from
utils.web
import
UC_UA
,
PC_UA
# import ddddocr
def
getPreJs
():
base_path
=
os
.
path
.
dirname
(
os
.
path
.
abspath
(
os
.
path
.
dirname
(
__file__
)))
# 上级目
...
...
@@ -36,15 +35,28 @@ def getHome(url):
homeUrl
=
urls
[
0
]
+
'//'
+
urls
[
1
].
split
(
'/'
)[
0
]
return
homeUrl
def
verifyCode
(
url
,
headers
,
timeout
=
5
,
total_cnt
=
3
):
class
OcrApi
:
def
__init__
(
self
,
api
):
self
.
api
=
api
def
classification
(
self
,
img
):
try
:
code
=
requests
.
post
(
self
.
api
,
data
=
img
,
headers
=
{
'user-agent'
:
PC_UA
}).
text
except
Exception
as
e
:
print
(
f
'ocr识别发生错误:
{
e
}
'
)
code
=
''
return
code
def
verifyCode
(
url
,
headers
,
timeout
=
5
,
total_cnt
=
3
,
api
=
None
):
if
not
api
:
api
=
'http://192.168.3.224:9000/api/ocr_img'
lower_keys
=
list
(
map
(
lambda
x
:
x
.
lower
(),
headers
.
keys
()))
host
=
getHome
(
url
)
if
not
'referer'
in
lower_keys
:
headers
[
'Referer'
]
=
host
print
(
f
'开始自动过验证,请求头:
{
headers
}
'
)
cnt
=
0
import
ddddocr
ocr
=
ddddocr
.
DdddOcr
()
ocr
=
OcrApi
(
api
)
while
cnt
<
total_cnt
:
s
=
requests
.
session
()
try
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录