sample_python_segno.py 339 B

12345678910111213
  1. import segno
  2. import os
  3. import tempfile
  4. qr = segno.make_qr('http:/www.example.org/')
  5. with tempfile.NamedTemporaryFile('wb', suffix='.png', delete=False) as f:
  6. fn = f.name
  7. qr.save(fn)
  8. expected = b'\211PNG\r\n\032\n' # PNG magic number
  9. with open(fn, mode='rb') as f:
  10. val = f.read(len(expected))
  11. os.unlink(fn)
  12. assert expected == val