用友T+ OpenAPI 企业自建应用对接指南

from fastapi import FastAPI
from fastapi.responses import FileResponse
import uvicorn
import os.path
import base64
from Crypto.Cipher import AES  
from pydantic import BaseModel
from datetime import datetime

class Item(BaseModel):
    encryptMsg: str

#https://open.chanjet.com/console/selfApp/14281/message
#该值来源于消息配置里的消息密钥
SECRET_KEY='1234567890123456'

#该class来自于 cnblogs.com/gqv2009/p/14365911.html
class AES_ENCRYPT(object):
    def __init__(self):
        self.key = SECRET_KEY
        self.mode = AES.MODE_ECB

    def pading(self, text):
        """对加密字符的处理"""
        return text + (len(self.key) - len(text) % len(self.key)) * chr(len(self.key) - len(text) % len(self.key))

    def unpading(self, text):
        """对解密字符的处理"""
        return text[0:-ord(text[-1:])]

    def getKey(self, key):
        """对key的处理,key 的长度 16，24，32"""
        key_len = len(key)
        if key_len <= 16:
            key += "0" * (16 - key_len)
        elif 16 < key_len <= 24:
            key += "0" * (24 - key_len)
        elif key_len <= 32:
            key += "0" * (32 - key_len)
        else:
            key = key[:32]
        return key

    # 加密函数
    def encrypt(self, text):
        cryptor = AES.new(self.key.encode("utf-8"), self.mode)  # ECB 模式
        self.ciphertext = cryptor.encrypt(bytes(self.pading(text), encoding="utf-8"))
        encrypt_string = str(base64.b64encode(self.ciphertext)).lstrip("b")
        return encrypt_string[1:len(encrypt_string)-1] #去掉头尾的引号

    # 解密函数
    def decrypt(self, text):
        decode = base64.b64decode(text)
        cryptor = AES.new(self.key.encode("utf-8"), self.mode)  # ECB 模式
        plain_text = cryptor.decrypt(decode)
        decrypt_string = str(self.unpading(plain_text)).lstrip("b")
        return decrypt_string[1:len(decrypt_string)-1] #去掉头尾的引号

app = FastAPI()


@app.get("/")
def read_root_get():
    return {"welcome":"T+"}

@app.post("/")
def read_root(item: Item):
    encryptMsg = item.encryptMsg
    aes_encrypt = AES_ENCRYPT()
    msg = aes_encrypt.decrypt(encryptMsg)
    print(f"{datetime.now()} {msg}")

    return {"result":"success"}
            

@app.get("/CHANJET_CHECK.txt")
def get_CHANJET_CHECK():
    file_object = open("CHANJET_CHECK.txt", 'w')
    #https://open.chanjet.com/console/domainManage/domainManage
    #该值来自于可信域名管理页面上CHANJET_CHECK.txt的内容
    file_object.write('NTRiYTI0OTZlMTIwNDkzYTllNmMzODA2YTJkNDNlYzI=')
    file_object.close()
    
    f = os.path.join('./', "CHANJET_CHECK.txt")
    print(f)
    return FileResponse(path=f,
        filename="CHANJET_CHECK.txt",)

@app.post("/api/retn")
def read_tokenPost(code: str= '', state: str=''):
    print(f'code:{code}, state:{state}')
    file_object = open("token.txt", 'a+')
    file_object.write(f'code:{code}, state:{state} \n')
    return {'code':code, 'state':state}

@app.get("/api/retn")
def read_tokenGet(code: str= '', state: str=''):
    print(f'code:{code}, state:{state}')
    file_object = open("token.txt", 'a+')
    file_object.write(f'code:{code}, state:{state} \n')
    return {'code':code, 'state':state}

if __name__ == '__main__':
    uvicorn.run(app='getTPtoken:app', host="0.0.0.0", port=8080, reload=True)