# 創建文件夾函數
def mkdir(path):
  isexists = os.path.exists(path) # 判斷文件夾路徑是否存在
  if not isexists: # 如果文件夾路徑不存在
    os.mkdir(path) # 創建要創建的文件夾


          

            
# 讀取文件內容函數
def openfile(filename):
  f = open(filename) # 打開指定文件
  fllist = f.read() # 讀取文件內容
  f.close() # 關閉文件
  return fllist # 返回讀取的文件內容


          

            
# 輸入驗證函數，showstr為input函數提供動態輸入提示文字，showorder提供驗證方式，length提供要求輸入數據的長度
def inputbox(showstr, showorder, length):
  instr = input(showstr) # 使用input函數要求用戶輸入信息，showstr為輸入提示文字
  if len(instr) != 0: # 輸入數據的長度不為零
    # 根據輸入數據的要求，分成三種驗證方式驗證，1：數字，不限位數；2：字母；3：數字且有位數要求
    if showorder == 1: # 驗證方式 ，數字格式，不限位數，大于零的整數
      if str.isdigit(instr): # 驗證是否為數字
        if instr == 0: # 驗證數字是否為0，如果是，要求重新輸入，返回值為0
          print("\033[1;31;40m 輸入為零，請重新輸入??！\033[0m") # 要求重新輸入，返回值為“0”
          return "0" # 函數返回值為“0”，為什么返回值為“0”呢？讀者思考一下
        else: # 如果輸入正確，返回輸入的數據給返回值
          return instr # 將輸入的數據傳給函數返回值
      else: # 如果輸入不是數字，要求用戶重新輸入，函數返回值為“0”
        print("\033[1;31;40m輸入非法，請重新輸入??！\033[0m") # 要求用戶重新輸入
        return "0" # 函數返回值為“0”
    if showorder == 2: # 驗證方式2 ，要求字母格式，且是三個字母
      if str.isalpha(instr): # 判斷輸入是否為字母
        if len(instr) != length: # 判斷輸入的是否為三個字母，如果不是，則要求重新輸入，返回值為“0”
          print("\033[1;31;40m必須輸入三個字母，請重新輸入！！\033[0m") # 要求重新輸入
          return "0" # 返回值為“0”
        else: # 如果輸入是三個字母，返回輸入的字母
          return instr # 將函數返回值設置為輸入的字母
      else: # 如果輸入不是字母
        print("\033[1;31;40m輸入非法，請重新輸入??！\033[0m") # 要求重新輸入
        return "0" # 返回值為“0”
    if showorder == 3: # 驗證方式3 ，要求數字格式，且輸入數字位數有要求
      if str.isdigit(instr): # 驗證是否為數字
        if len(instr) != length: # 驗證輸入數字是否為要求長度位數，如果不是3位數字，則要求重新輸入
          print("\033[1;31;40m必須輸入" + str(length) + "個數字，請重新輸入?。033[0m") # 要求重新輸入
          return "0" # 返回值為“0”
        else: # 輸入數字滿足要求，設置函數返回值為輸入信息
          return instr # 設置函數返回值為輸入信息
      else: # 如果輸入不是數字
        print("\033[1;31;40m輸入非法，請重新輸入！！\033[0m") # 提示輸入非法，要求重新輸入
        return "0" # 函數返回值為“0”
  else: # 如果沒有輸入任何內容，即輸入為空
    print("\033[1;31;40m輸入為空，請重新輸入！！\033[0m") # 提示輸入為空，要求重新輸入
    return "0" # 函數返回值為“0”


          

            
# 實現屏幕輸出和文件輸出編碼信息函數，# sstr參數為輸出防偽碼數據, sfile為輸出的文件名稱
# typeis設置輸出完成后是否通過信息框提示, smsg為信息提示框的提示文字，datapath 保存防偽碼的文件夾
def wfile(sstr, sfile, typeis, smsg, datapath):
  mkdir(datapath) # 調用該函數創建文件夾
  datafile = datapath + "\\" + sfile # 設置保存防偽碼的文件（包含路徑）
  file = open(datafile, 'w') # 打開保存防偽碼的文件，如果文件不存在，則創建該文件
  wrlist = sstr # 將防偽碼信息賦值給wrlist
  pdata = "" # 清空變量pdata，pdata存儲屏幕輸出的防偽碼信息
  wdata = "" # 清空變量 wdata ， wdata 存儲保存到文本文件的防偽碼信息
  for i in range(len(wrlist)): # 按條循環讀取防偽碼數據
    wdata = str(wrlist[i].replace('[', '')).replace(']', '') # 去掉字符的中括號
    wdata = wdata.replace(''''','').replace(''''', '') # 去掉字符的引號
    file.write(str(wdata)) # 寫入保存防偽碼的文件
    pdata = pdata + wdata # 將單條防偽碼存儲到pdata 變量
  file.close() # 關閉文件
  print("\033[1;31m" + pdata + "\033[0m") # 屏幕輸出生成的防偽碼信息
  if typeis != "no": # 是否顯示“輸出完成”的信息提示框。如果typeis的值為“no”,不現顯示
    # 顯示“輸出完成”的信息提示框。顯示信息包含防偽信息碼的保存路徑
    tkinter.messagebox.showinfo("提示", smsg + str(len(randstr)) + "\n 防偽碼文件存放位置：" + datafile)
    root.withdraw() # 關閉輔助窗口