-
-
Save otienoelvis/5cc9db4997a0ab4ccd39b9c11117605c to your computer and use it in GitHub Desktop.
Upload, read and save the content of a csv file to your model
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
from io import TextIOWrapper | |
import csv | |
from flask import Flask, request, redirect, url_for | |
from flask_sqlalchemy import SQLAlchemy | |
# Create Flaskk app, config the db and load the db object | |
# http://flask-sqlalchemy.pocoo.org/2.1/quickstart/#a-minimal-application | |
app = Flask(__name__) | |
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///test.db' | |
db = SQLAlchemy(app) | |
class User(db.Model): | |
id = db.Column(db.Integer, primary_key=True) | |
username = db.Column(db.String(80), unique=True) | |
email = db.Column(db.String(120), unique=True) | |
def __repr__(self): | |
return "<User: {}>".format(self.username) | |
@app.route('/', methods=['GET', 'POST']) | |
def upload_csv(): | |
if request.method == 'POST': | |
csv_file = request.files['file'] | |
csv_file = TextIOWrapper(csv_file, encoding='utf-8') | |
csv_reader = csv.reader(csv_file, delimiter=',') | |
for row in csv_reader: | |
user = User(username=row[0], email=row[1]) | |
db.session.add(user) | |
db.session.commit() | |
return redirect(url_for('upload_csv')) | |
return """ | |
<form method='post' action='/' enctype='multipart/form-data'> | |
Upload a csv file: <input type='file' name='file'> | |
<input type='submit' value='Upload'> | |
</form> | |
""" | |
if __name__ == '__main__': | |
db.create_all() | |
app.run() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment