Flask-SQLAlchemy User Management Functions

  • Share this:

Code introduction


The functions in the code are used to add, search by email, delete by ID, and update the email address of a user. The Flask-SQLAlchemy ORM features are used to interact with the database.


Technology Stack : Flask, Flask-SQLAlchemy, SQLAlchemy

Code Type : The type of code

Code Difficulty :


                
                    
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///example.db'
db = SQLAlchemy(app)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50), nullable=False)
    email = db.Column(db.String(120), unique=True, nullable=False)

def add_user(name, email):
    new_user = User(name=name, email=email)
    db.session.add(new_user)
    db.session.commit()

def get_user_by_email(email):
    return User.query.filter_by(email=email).first()

def delete_user_by_id(user_id):
    user_to_delete = User.query.get(user_id)
    if user_to_delete:
        db.session.delete(user_to_delete)
        db.session.commit()

def update_user_email(user_id, new_email):
    user_to_update = User.query.get(user_id)
    if user_to_update:
        user_to_update.email = new_email
        db.session.commit()