1
- from flask import Flask , render_template , request ,flash ,redirect ,send_from_directory
1
+ from flask import Flask , render_template , request , flash , redirect , send_from_directory
2
2
from scrapper import simple_image_download
3
3
from flask_bootstrap import Bootstrap
4
4
from flask_wtf import FlaskForm
5
- from wtforms import StringField , SubmitField , IntegerField ,SelectField
5
+ from wtforms import StringField , SubmitField , IntegerField , SelectField
6
6
from wtforms .validators import DataRequired , Email
7
7
import os
8
8
9
9
app = Flask (__name__ , template_folder = 'templates' )
10
- response = simple_image_download ()
10
+ response = simple_image_download ()
11
11
app .secret_key = 'tO$&!|0wkamvVia0?n$NqIRVWOG'
12
12
bootstrap = Bootstrap (app )
13
13
downloaded = [False ]
14
- image_request = {'name' :'' ,'number_of_images' :0 }
14
+ image_request = {'name' : '' , 'number_of_images' : 0 }
15
+
15
16
16
17
class ImageForm (FlaskForm ):
17
18
name = StringField ('name' , validators = [DataRequired ()])
18
19
number_of_images = IntegerField ('number_of_images' , validators = [DataRequired ()])
19
20
submit = SubmitField ('Submit' )
20
21
22
+
21
23
@app .route ('/' , methods = ['GET' , 'POST' ])
22
24
def index ():
23
25
form = ImageForm ()
@@ -27,13 +29,14 @@ def index():
27
29
flash ('Your images are being downloaded. Please wait.' )
28
30
downloaded [0 ] = True
29
31
return redirect ('/' )
30
-
32
+
31
33
if downloaded [0 ]:
32
- response .download (image_request ['name' ] , int (image_request ['number_of_images' ]))
34
+ response .download (image_request ['name' ], int (image_request ['number_of_images' ]))
33
35
flash ('All of your images have been downloaded' )
34
36
downloaded [0 ] = False
35
37
return redirect ('/' )
36
- return render_template ('index.html' , form = form )
38
+ return render_template ('index.html' , form = form )
39
+
37
40
38
41
if __name__ == '__main__' :
39
42
app .run (host = "0.0.0.0" , port = 8000 )
0 commit comments