import memory_pic  # 无需像原博客那样一个个导入
import os
import base64

base_path = 'data'


def prepare_data():
    if not os.path.exists(base_path) or os.path.isfile(base_path):
        os.makedirs(base_path)
    # pics = ["bank2.png", "id_back.png", "id_front.png", "license.png", "banknametobin.json",
    #         "districtcode.txt"]
    create_file(memory_pic.bank2_png, base_path+"/bank2.png")
    create_file(memory_pic.id_back_png, base_path+"/id_back.png")
    create_file(memory_pic.id_front_png, base_path+"/id_front.png")
    create_file(memory_pic.license_png, base_path+"/license.png")
    create_file(memory_pic.banknametobin_json, base_path+"/banknametobin.json")
    create_file(memory_pic.districtcode_txt, base_path+"/districtcode.txt")


def create_file(data, file):
    if not os.path.exists(file) or not os.path.isfile(file):
        get_pic(data, file)


def get_pic(pic_code, pic_name):
    image = open(pic_name, 'wb')
    image.write(base64.b64decode(pic_code))
    image.close()