CODEWITHSUNDEEP
flaskflask-sqlalchemy
Riskhan
Riskhan

Reputation: 4470

How to automaticaly insert a row in flask-sqlalchemy while creating the database in flask

I created a database script using python flask. I used flask-sqlalchemy. I want to insert a record in the database with immediate after creating the database. Kindly guide me to do. I made the below code from scratch.

from flask import Flask, request, flash, url_for, redirect, render_template
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///students.sqlite3'
app.config['SECRET_KEY'] = "random string"

db = SQLAlchemy(app)

class students(db.Model):
    id = db.Column('student_id', db.Integer, primary_key = True)
    name = db.Column(db.String(100))
    city = db.Column(db.String(50))
    addr = db.Column(db.String(200)) 
    pin = db.Column(db.String(10))

    def __init__(self, name, city, addr,pin):
        self.name = name
        self.city = city
        self.addr = addr
        self.pin = pin

@app.route('/')
def show_all():
   return render_template('show_all.html', students = students.query.all() )

@app.route('/new', methods = ['GET', 'POST'])
def new():
   if request.method == 'POST':
      if not request.form['name'] or not request.form['city'] or not request.form['addr']:
         flash('Please enter all the fields', 'error')
      else:
         student = students(request.form['name'], request.form['city'],
            request.form['addr'], request.form['pin'])
         
         db.session.add(student)
         db.session.commit()
         flash('Record was successfully added')
         return redirect(url_for('show_all'))
   return render_template('new.html')

if __name__ == '__main__':
   db.create_all()
   app.run(debug = True)

Upvotes: 1

Views: 1583

Answers (1)

gittert
gittert

Reputation: 1308

you could use @app.before_first_request like this example. Btw it is common practice to start class name in upper class. see https://flask.palletsprojects.com/en/1.1.x/api/#flask.Flask.before_first_request

@app.before_first_request
def populate_db():
    student = students(
        name = 'Rishkan', 
        city = 'New York',
        addr = 'test street 10',
        pin = '1234')
    
    db.session.add(student)
    db.session.commit()

Upvotes: 2

Related Questions