from app.models import db, Type

def seed_types():
    admin = Type(type="Admin")
    client = Type(type="Client")
    lb = Type(type="Low Bandwidth")

    db.session.add(admin)
    db.session.add(client)
    db.session.add(lb)

    db.session.commit()

def undo_types():
    db.session.execute('TRUNCATE TABLE types RESTART IDENTITY CASCADE;')
    db.session.commit()

from app.models import db, Type

def seed_types():
    admin = Type(type="Admin")
    client = Type(type="Client")
    lb = Type(type="Low Bandwidth")

    db.session.add(admin)
    db.session.add(client)
    db.session.add(lb)

    db.session.commit()

def undo_types():
    db.session.execute('TRUNCATE TABLE types RESTART IDENTITY CASCADE;')
    db.session.commit()
