from gooey import GooeyParser,Gooey
from MyQR import myqr

@Gooey(encoding='utf-8',program_name='二维码生成器',program_description='生成动态、彩色、黑白二维码',language='chinese')
def main():
  parser = GooeyParser(description="生成动态、彩色、黑白二维码")
  parser.add_argument('文本', widget="TextField")
  parser.add_argument('文件', widget="FileChooser")   # 文件选择框
  parser.add_argument('样式选择', widget='Dropdown',choices={"黑白":"1","彩色":2,"动态":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.样式选择 == '动态':
    if args.文件.split('.')[-1] == 'gif':
      myqr.run(words=args.文本, version=9, save_name='qrcode.gif', save_dir=args.文件保存路径, picture=args.文件,colorized=True)
    else:
      print("图片格式错误,动态二维码图片必须为gif格式的")
if  __name__ == '__main__':
  main()