models.py
2.0 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
# coding=utf-8
import os
from flask_sqlalchemy import SQLAlchemy
import glob
import traceback
from pyDes import des, PAD_PKCS5, ECB
import base64
from gmssl import sm3, func, sm2
class SM3():
@classmethod
def encode(self, data=""):
by_str = bytes(data, 'utf-8')
result = (sm3.sm3_hash(func.bytes_to_list(by_str)))
return result
class SM2():
privatekey = "EOpBYj1pH54b8BsAlkON71Q2c2FXEm+VTPNCgxz4+gVZ2C/pF/Bv152Qj3QH7cBRCUlNeO5SkI02DcwTASmVwESzR4F9IbMvUNxkDngINdcwaSWCTaaRTstNd4FIlJ0CqhDl/TPHFuBoRNBgUDfPJFfl7XYHs4VmOcrp02aXv8Q="
publickey = "MFkwEwYHKoZIzj0CAQYIKoEcz1UBgi0DQgAEsENe/8Sft6rK0h59t2a/6Pf4nzvxvJzc1SzalZe1xSDtUqW/10E/e+RjRbvykyjKzu3vAY1lwv9aU2Tnp6+isw=="
sm2_crypt = sm2.CryptSM2(public_key=publickey, private_key=privatekey)
@classmethod
def decode(self, ciphertxt):
return self.sm2_crypt.decrypt(ciphertxt)
@classmethod
def encode(self, plaintxt):
return self.sm2_crypt.encrypt(plaintxt)
class DES():
'''
DES密码加解密
'''
Des: des = des("Chinadci", ECB, "\0\0\0\0\0\0\0\0",
pad=None, padmode=PAD_PKCS5)
@classmethod
def encode(cls, data):
return str(base64.b64encode(cls.Des.encrypt(data)), encoding="utf8")
@classmethod
def decode(cls, data):
if data:
return str(cls.Des.decrypt(base64.b64decode(data.encode("utf-8"))), encoding="utf8")
else:
return data
db = SQLAlchemy()
# 动态加载Model
current_dir = os.path.abspath(os.path.dirname(__file__))
pkgs = list(glob.glob('%s/modules/*/models' % (current_dir)))
pkgs.extend(list(glob.glob('%s/modules/*/*/models' % (current_dir))))
for pkg in pkgs:
pkg = os.path.normpath(pkg)
node_list = pkg.split(os.path.sep)
pkg_name = "app.{}".format(
".".join(node_list[node_list.index("modules"):]))
try:
if pkg_name.__contains__("models"):
__import__(pkg_name)
except Exception as e:
print(traceback.format_exc())
pass