# coding=utf-8

import os
from flask_sqlalchemy import SQLAlchemy
import glob
import traceback
from pyDes import des,PAD_PKCS5,ECB

import base64
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