tcutils/pybccclt.py

# -*- coding: utf-8 -*-

from ctypes import *

bccclt_dll = None
xpack_path = 'cpack.dat'

cpack_fields = [
"scust_no",
"scust_no2",
"sholder_ac_no",
"sholder_ac_no2",
"sholder_type",
"sholder_type2",
"sname",
"sname2",
"sall_name",
"smarket_code",
"smarket_code2",
"sdate0",
"sdate1",
"sdate2",
"sdate3",
"stime0",
"stime1",
"stime2",
"stime3",
"lvol0",
"lvol1",
"lvol2",
"lvol3",
"lvol4",
"lvol5",
"lvol6",
"lvol7",
"lvol8",
"lvol9",
"lvol10",
"lvol11",
"lvol12",
"damt0",
"damt1",
"damt2",
"damt3",
"damt4",
"damt5",
"damt6",
"damt7",
"damt8",
"damt9",
"damt10",
"damt11",
"damt12",
"damt13",
"damt14",
"damt15",
"damt16",
"damt17",
"damt18",
"damt19",
"damt20",
"damt21",
"damt22",
"damt23",
"damt24",
"damt25",
"damt26",
"damt27",
"damt28",
"damt29",
"damt30",
"damt31",
"damt32",
"damt33",
"sstock_code",
"sstock_code2",
"scust_type",
"scust_type2",
"sstat_type",
"sstat_type2",
"sroom_no",
"sroom_no2",
"sopen_emp",
"sclose_emp",
"schange_emp",
"scheck_emp",
"semp",
"snation_code",
"lcert_code",
"stx_pwd",
"stx_pwd2",
"swithdraw_pwd",
"swithdraw_pwd2",
"semp_pwd",
"semp_pwd2",
"sbank_pwd",
"sbank_pwd2",
"scust_auth",
"scust_auth2",
"scust_limit",
"scust_limit2",
"lsafe_level",
"lsafe_level2",
"spost_code",
"spost_code2",
"sphone",
"sphone2",
"sphone3",
"spager",
"semail",
"semail2",
"snote",
"snote2",
"scert_no",
"scert_no2",
"scert_addr",
"sstatus0",
"sstatus1",
"sstatus2",
"sstatus3",
"sstatus4",
"lwithdraw_flag",
"saddr",
"saddr2",
"sserial0",
"sserial1",
"sserial2",
"sserial3",
"sserial4",
"scurrency_type",
"scurrency_type2",
"sbranch_code0",
"sbranch_code1",
"sbranch_code2",
"usset0",
"usset1",
"usset2",
"usset3",
"usset4",
"usset5",
"usset6",
"sstation0",
"sstation1",
"sbank_acc",
"sbank_acc2",
"lbank_acc_type",
"lbank_acc_type2",
"smain_flag",
"smain_flag2",
"sbank_code",
"sbank_code2",
"semp_no",
"semp_no2",
"drate0",
"drate1",
"lserial0",
"lserial1",
"sbankname",
"sbankname2",
"scard0",
"scard1",
"sorder0",
"sorder1",
"sorder2",
"scusttypes",
"ssectypes",
"vsmess",
"vsvarstr0",
"vsvarstr1",
"vsvarstr2"
]

def load_bccclt():
    global bccclt_dll
    bccclt_dll = cdll.LoadLibrary('bccclt.dll')
    
  
load_bccclt()  

class pyBccclt:
    drtp_no = -1
    
    def setup(ip,port,xpack='cpack.dat',debug=0):
        ''' 初始化 '''
        global xpack_path
        xpack_path = xpack
        bccclt_dll.SetDebugSwitch( debug )
        r = bccclt_dll.BCCCLTInit(1)
        if not r:
            raise RuntimeError(u"初始化连接失败")
        no = bccclt_dll.AddDrtpNode(ip,port)
        pyBccclt.drtp_no = no
        if pyBccclt.drtp_no < 0:
            raise RuntimeError(u'初始化通讯平台连接失败')
            
        
        
    setup = staticmethod(setup)
        
    def __init__(self,mainfunc):
        self.handle = bccclt_dll.NewXpackHandle( xpack_path )
        self.mainfunc = mainfunc
        self.reset()
        
    def get_errormsg(self):
        return self.error_msg.value
        
    def reset(self):
        self.request_row = 0
        self.error_code = c_int(0)
        self.error_msg = create_string_buffer('\000' * 1024)
        self.return_code = c_int(0)
        self.record_count = 0
        self.record = None
        self.timeout = 5000
        self.response_row = 0
        bccclt_dll.ResetPackHandle(self.handle)
        
    def close(self):
        if self.handle <> None:
            bccclt_dll.DeleteXpackHandle( self.handle )
            self.handle = None
        
    def callsvr(self,func,request,timeout = 1000):
        self.reset()
        self.timeout = timeout
        for k,v in request.iteritems():
            if k not in cpack_fields:
                raise ValueError(u"字段 %s 不存在" % k)
            
            if isinstance(v,int):
                bccclt_dll.SetIntFieldByName(self.handle,self.request_row,k,v)
            elif isinstance(v,float):
                bccclt_dll.SetDoubleFieldByName(self.handle,self.request_row,k,v)
            elif isinstance(v,str):
                bccclt_dll.SetStringFieldByName(self.handle,self.request_row,k,v)
            else:
                raise ValueError(u"字段 %s 数据类型错误" % k)
               
        bccclt_dll.SetRequestType(self.handle,func)
        #print "CallRequest %d:%d" % (self.mainfunc,func)
        if not bccclt_dll.CallRequest(self.handle,pyBccclt.drtp_no,0,self.mainfunc,timeout,byref(self.error_code),self.error_msg):
            #print "CallRequest error "
            return False
            
        self.get_return_error()
        
        cnt = c_int(0)
        bccclt_dll.GetRecordCount(self.handle,byref(cnt))
        self.record_count = cnt.value
        return True
                
    def get_retcode(self):
        return self.return_code.value
        
    def get_return_error(self):
        r = bccclt_dll.GetRetCode(self.handle,byref(self.return_code))
        if not r:
            raise RuntimeError(u'取返回码错误')
        if self.get_retcode() <> 0:
            bccclt_dll.GetStringFieldByName(self.handle,0,"vsmess",self.error_msg)
        return self.return_code.value
        
    def has_more_record(self):
        #print "resp:%d, record: %d" % (self.response_row , self.record_count)
        if self.response_row < self.record_count:
            return True
        elif not bccclt_dll.HaveNextPack( self.handle ):
            return False
        elif not bccclt_dll.CallNext(self.handle,self.timeout,byref(self.error_code),self.error_msg):
            raise RuntimeError(u'获取后续数据包异常')
        else:
            self.response_row = 0
            self.record_count = 0
            self.get_return_error()
            if self.get_retcode() <> 0:
                raise RuntimeError(u'取后续数据返回错误，ret=%d' % self.get_retcode())
            cnt = c_int(0)
            bccclt_dll.GetRecordCount(self.handle,byref(cnt))
            self.record_count = cnt.value
            if self.response_row  < self.record_count:
                return True
            return False
    
    def convert_ascii(self,data):
        i = 0
        datalen = len(data)
        result = ''
        while i < datalen:
            t = data[i:i+2]
            t1 = int(t,16) & 0xFF
            if (t1 < ord('0') or t1 > ord('9')) and (t1 < ord('A') or t1 > ord('F')):
                raise ValueError("Value is non-hexdicimal %s:%d" % (chr(t1),t1))
            result += chr(t1)
            i += 2
        return result
        
    def next_record(self,fields):
        if self.response_row >= self.record_count:
            raise ValueError(u'无可用记录')
        
        self.record = {}
        for f in fields:
            #print "获取字段 ",f
            if f not in cpack_fields:
                raise ValueError(u"字段 %s 不存在" % f)
            
            ctype_field = create_string_buffer(f)
            if f[0] == 'l':
                v = c_int(0)
                r = bccclt_dll.GetIntFieldByName(self.handle,self.response_row,ctype_field,byref(v))
                if not r:
                    raise ValueError(u"字段 %s 未返回" % f)
                self.record[f] = v.value
            elif f[0] == 's':
                v = create_string_buffer('\000' * 512)
                r = bccclt_dll.GetStringFieldByName(self.handle,self.response_row,ctype_field,v,511)
                if not r:
                    raise ValueError(u"字段 %s 未返回" % f)
                #print "%s: %s" % (f,v.value)
                self.record[f] = v.value
            elif f[0] == 'd':
                v = c_double(0.0)
                r = bccclt_dll.GetDoubleFieldByName(self.handle,self.response_row,ctype_field,byref(v))
                if not r:
                    raise ValueError(u"字段 %s 未返回" % f)
                self.record[f] = v.value
            elif f[0] == 'u':
                v = create_string_buffer('\000' * 1024)
                r = bccclt_dll.GetStringFieldByName(self.handle,self.response_row,ctype_field,v,1023)
                if not r:
                    raise ValueError(u"字段 %s 未返回" % f)
                    
                data = v.value
                data = data[2:]
                data = self.convert_ascii(data)
                self.record[f] = data
            else:
                raise ValueError(u'字段类型错误 %s' % f)
        self.response_row += 1
        #print "fetch one record , " ,self.record
        return self.record
        
    def fetch_record(self,fields):
        while self.has_more_record():
            ret = self.next_record(fields)
            yield ret