utils.py样本

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/sjpljr/article/details/80737911

#!/usr/bin/env python
# -*- coding:utf-8 -*- 

'''
Created on 2016年10月1日
@author: Jimmy Liu
@group : waditu
@contact: [email protected]
'''
import sys
sys.path.append(r"d:/pdsi")
from pdsi import cropimage,clrfiles,get_feature, svm_model_test
import os
import json
import time
import six
from tushare.trader import vars as vs
import pandas as pd
def nowtime_str():
    return time.time() * 1000


def get_jdata(txtdata):
    txtdata = txtdata.content
    if six.PY3:
        txtdata = txtdata.decode('utf-8')
    jsonobj = json.loads(txtdata)
    return jsonobj
        
        
def get_vcode(broker, res):
    from PIL import Image
    import pytesseract as pt
    import io
    if broker == 'csc':
        imgdata = res.content
        #----------------
        
        
        
        #---------------------------
        img = Image.open(io.BytesIO(imgdata)).convert("1")
        img.save(r'd:/pdsi/img/imgs/0image.png')
        print(type(img),img)
        vcode = pt.image_to_string(img)
        clrfiles(r'./img/img_split/')
        cropimage(img,0,4)
        df_a=pd.DataFrame()
        for imgname in os.listdir('d:/pdsi/img/img_split/'):
            aa=get_feature(Image.open('d:/pdsi/img/img_split/'+str(imgname)))
            aa.insert(0,"2")
            df_a=df_a.append([aa])
        df_a.to_csv(r'./img/last_test_pix_xy_new.txt', index=False, sep=r' ', header=False)
        aaa=svm_model_test()
        print('验证码', aaa)
        vcode=str(aaa)


        return vcode