from gooey import GooeyParser,Gooey
from MyQR import myqr

@Gooey(encoding='utf-8',program_name='二維碼生成器',program_description='生成動(dòng)態(tài)、彩色、黑白二維碼',language='chinese')
def main():
  parser = GooeyParser(description="生成動(dòng)態(tài)、彩色、黑白二維碼")
  parser.add_argument('文本', widget="TextField")
  parser.add_argument('文件', widget="FileChooser")   # 文件選擇框
  parser.add_argument('樣式選擇', widget='Dropdown',choices={"黑白":"1","彩色":2,"動(dòng)態(tài)":3})
  parser.add_argument('文件保存路徑',widget='DirChooser')
  args = parser.parse_args()
  if args.樣式選擇 == '黑白':
    # 生成二維碼
    myqr.run(words=args.文本,version=9,save_name='qr.png',save_dir=args.文件保存路徑)
    print('成功')
  elif args.樣式選擇 == '彩色':
    if args.文件.split('.')[-1] == 'gif' or args.文件.split('.')[-1] == 'png' or args.文件.split('.')[-1] == 'jpeg':
      myqr.run(words=args.文本, version=9, save_name='qr.png', save_dir=args.文件保存路徑, picture=f'{args.文件}',colorized=True)
      print('成功')
  elif args.樣式選擇 == '動(dòng)態(tài)':
    if args.文件.split('.')[-1] == 'gif':
      myqr.run(words=args.文本, version=9, save_name='qrcode.gif', save_dir=args.文件保存路徑, picture=args.文件,colorized=True)
    else:
      print("圖片格式錯(cuò)誤,動(dòng)態(tài)二維碼圖片必須為gif格式的")
if  __name__ == '__main__':
  main()