social-engineer-toolkit/src/qrcode/qrgenerator.py

18 lines
615 B
Python
Raw Normal View History

2012-12-31 17:11:37 -05:00
from src.qrcode.qrcode import *
from src.core.setcore import *
import os
# generate the qrcode and save it definition
2012-12-31 17:11:37 -05:00
def gen_qrcode(url):
# generate the qrcode
qr = QRCode(5, QRErrorCorrectLevel.L)
qr.addData(url)
qr.make()
im = qr.makeImage()
time.sleep(1)
2013-04-15 10:26:00 -04:00
if os.path.isfile(setdir + "/reports/qrcode_attack.png"): os.remove(setdir + "/reports/qrcode_attack.png")
# save the image out
2013-04-15 10:26:00 -04:00
im.save(setdir + "/reports/qrcode_attack.png", format='png')
# print that its been successful
print_status("QRCode has been generated under %s/reports/qrcode_attack.png!" % (setdir))