Random User Generation in SQLite Database

This function creates a temporary SQLite database, defines a User model, and generates a random user information, saving it to the database.

Technology Stack : SQLAlchemy, SQLite, declarative_base, Column, Integer, String, create_engine, sessionmaker, Session, random

Code Type : Function

Code Difficulty : Intermediate

                
                    
from sqlalchemy import create_engine, Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

Base = declarative_base()

class User(Base):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    name = Column(String)
    age = Column(Integer)

def generate_random_user():
    # Generate a random user and save it to the database
    engine = create_engine('sqlite:///:memory:')
    Base.metadata.create_all(engine)
    Session = sessionmaker(bind=engine)
    session = Session()

    # Generate a random name and age
    import random
    name = f"User{random.randint(1, 100)}"
    age = random.randint(18, 65)

    # Create a new User instance
    user = User(name=name, age=age)

    # Add the user to the session and commit
    session.add(user)
    session.commit()

    # Return the user
    return user