'''generate a captchaes'''
import random
import os
from captcha.image import ImageCaptcha
'''generating list'''
digits=['0','1','2','3','4','5','6','7','8','9']
'''generate a text with four digits'''
def gen_random_text(field=digits,size=4):
# create a list
res=[]
# for _ in size:
for _ in range(size):
# append to the list
digit=random.choice(digits)
res.append(digit)
# return the list
return res
'''create an image'''
image=ImageCaptcha()
captcha_txt=''.join(gen_random_text())
image_data=image.generate(captcha_txt)
'''store the image'''
# dest dir
captcha_dir='captcha'
# check dir
if not os.path.exists(captcha_dir):
os.makedirs(captcha_dir)
filename=captcha_txt+'.png'
image.write(captcha_txt,os.path.join(captcha_dir,filename))