install psycopg2 https://pypi.org/project/psycopg2/
Psycopg is the most popular PostgreSQL database adapter for the Python programming language.
(venv) PS C:\flaskmyproject> pip install psycopg2
CREATE TABLE images (
id serial PRIMARY KEY,
file_name VARCHAR ( 100 ) NOT NULL,
uploaded_on TIMESTAMP
);
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | #app.pyfrom flask import Flask, render_template, redirect, request, flashfrom werkzeug.utils import secure_filenameimport os#import magicimport urllib.requestfrom datetime import datetimeimport psycopg2 #pip install psycopg2 import psycopg2.extras app = Flask(__name__) app.secret_key = "caircocoders-ednalan" DB_HOST = "localhost"DB_NAME = "sampledb"DB_USER = "postgres"DB_PASS = "admin" conn = psycopg2.connect(dbname=DB_NAME, user=DB_USER, password=DB_PASS, host=DB_HOST) UPLOAD_FOLDER = 'static/uploads'app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER ALLOWED_EXTENSIONS = set(['png', 'jpg', 'jpeg', 'gif']) def allowed_file(filename): return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS @app.route('/')def index(): return render_template('index.html') @app.route("/upload",methods=["POST","GET"])def upload(): cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor) now = datetime.now() print(now) if request.method == 'POST': files = request.files.getlist('files[]') #print(files) for file in files: if file and allowed_file(file.filename): filename = secure_filename(file.filename) file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename)) cur.execute("INSERT INTO images (file_name, uploaded_on) VALUES (%s, %s)",[filename, now]) conn.commit() print(file) cur.close() flash('File(s) successfully uploaded') return redirect('/') if __name__ == "__main__": app.run(debug=True) |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | //templates/index.html<!DOCTYPE html><html><head> <title>Upload Multiple Images using Python Flask PostgreSQL Database</title><link rel="stylesheet" href="https://maxcdn.bootstrapcdn.com/bootstrap/3.3.6/css/bootstrap.min.css" /> </head><body><div class="container"><h2>Upload Multiple Images using Python Flask PostgreSQL Database</h2> <div class="row"> <div class="col-lg-12"> <p> {% with messages = get_flashed_messages() %} {% if messages %} {% for message in messages %} <div class="alert alert-success"> <strong>{{ message }}</strong> </div> {% endfor %} {% endif %} {% endwith %} </p> <div> <!-- File upload form --> <form method="post" action="/upload" enctype="multipart/form-data" class="form-inline"> <div class="form-group"> <label>Choose Images: </label> <input type="file" name="files[]" id="fileInput" class="form-control" multiple > </div> <input type="submit" name="submit" class="btn btn-success" value="UPLOAD"/> </form> </div> </div> </div></div><script>$(document).ready(function(){ // File type validation $("#fileInput").change(function(){ var fileLength = this.files.length; var match= ["image/jpeg","image/png","image/jpg","image/gif"]; var i; for(i = 0; i < fileLength; i++){ var file = this.files[i]; var imagefile = file.type; if(!((imagefile==match[0]) || (imagefile==match[1]) || (imagefile==match[2]) || (imagefile==match[3]))){ alert('Please select a valid image file (JPEG/JPG/PNG/GIF).'); $("#fileInput").val(''); return false; } } });});</script></body></html> |
