gen_units.py 832 B

123456789101112131415161718192021222324252627282930313233
  1. import random
  2. import qrcode
  3. import os
  4. # 生成二维码图片
  5. def gen_qr_code(data):
  6. qr = qrcode.QRCode(
  7. version=1,
  8. error_correction=qrcode.constants.ERROR_CORRECT_L,
  9. box_size=10,
  10. border=4,
  11. )
  12. qr.add_data(data)
  13. qr.make(fit=True)
  14. return qr.make_image(fill_color="black", back_color="white")
  15. def gen_file_url(file, url):
  16. file_name = str(random.randint(10000, 99999))
  17. # file_name = file.filename
  18. file_ext = os.path.splitext(file.filename)[1]
  19. path = 'uploads/'
  20. if file_ext == '.json':
  21. path += 'json/' + file_name + file_ext
  22. url += "json/" + file_name
  23. elif file_ext == '.html':
  24. path += 'html/' + file_name + file_ext
  25. url += "html/" + file_name
  26. else:
  27. path += file_name + file_ext
  28. file.save(path)
  29. return url