python校验社会统一信用代码

# -*- coding: utf-8 -*-
import pandas as pd
import numpy as np
import pymysql
SOCIAL_CREDIT_CHECK_CODE_DICT = {
                '0':0,'1':1,'2':2,'3':3,'4':4,'5':5,'6':6,'7':7,'8':8,'9':9,
                'A':10,'B':11,'C':12, 'D':13, 'E':14, 'F':15, 'G':16, 'H':17, 'J':18, 'K':19, 'L':20, 'M':21, 'N':22, 'P':23, 'Q':24,
               'R':25, 'T':26, 'U':27, 'W':28, 'X':29, 'Y':30}
weighting_factor = [1, 3, 9, 27, 19, 26, 16, 17, 20, 29, 25, 13, 8, 24, 10, 30, 28]
# 本体代码
def check_social_credit_code(code):
    check_code = code[17]
    # 计算校验码公式:
    #     C9 = 31-mod(sum(Ci*Wi)，31)，其中Ci为组织机构代码的第i位字符,Wi为第i位置的加权因子,C9为校验码
    # 第i位置上的加权因子
    ontology_code = code[0:17]
    # 计算校验码
    tmp_check_code = gen_check_code(weighting_factor, ontology_code, 31, SOCIAL_CREDIT_CHECK_CODE_DICT)
    if tmp_check_code == check_code:
        return 1
    else:
        return 0
def gen_check_code(weighting_factor, ontology_code, modulus, check_code_dict):
    #     @param weighting_factor: 加权因子
    #     @param ontology_code:本体代码
    #     @param modulus:  模数
    #     @param check_code_dict: 字符字典
    total = 0
    for i in range(len(ontology_code)):
        if ontology_code[i].isdigit():
            total += int(ontology_code[i]) * weighting_factor[i]
        else:
            total += check_code_dict[ontology_code[i]] * weighting_factor[i]
#当MOD函数值为0时，校验码用0表示
    if modulus - total % modulus == 31:
        return list(check_code_dict.keys())[list(check_code_dict.values()).index(0)]
    else:
        diff = modulus - total % modulus
        print(diff)
        return list(check_code_dict.keys())[list(check_code_dict.values()).index(diff)]

dbconn = pymysql.connect(
    host="172.16.3.23",
    database="test_srz",
    user="test_srz",
    password="test_srz123",
    port=31229,
    charset='utf8'
)
sqlcmd = "select  qymc from shtyxydmyz"
a = pd.read_sql(sqlcmd, dbconn)
zchdjh = np.array(a)
print(zchdjh)
for n in range(len(zchdjh)):
    m = zchdjh[n]
    for o in range(len(m)):
        p = m[o]
        q = check_social_credit_code(p)
        cursor = dbconn.cursor()
#第一种结果处理方法，在原表基础上增加状态
        cursor.execute("update shtyxydmyz set zt = %s where qymc ='%s'" % (q, p))
#第二种方法，合格的导入emp2，不合格的导入emp3.
        # if q==1:
        #     cursor.execute("insert into emp2(xydm) values(%s)", (p))
        # else:
        #     cursor.execute("insert into emp3(xydm) values(%s)", (p))
        dbconn.commit()
        cursor.close()