bogus.py

import psycopg2

db = psycopg2.connect('postgresql:///mastodon_development')

cur = db.cursor()

for i in range(500):
    screenname = "bogus_{}".format(i)
    email = "bogus_{}@m.codl.fr".format(i)
    acc_to_follow = 1
    
    try:
        cur.execute('''
        INSERT INTO accounts (username, created_at, updated_at)
        VALUES (%s, now(), now())
        RETURNING id;
        ''', (screenname, ))

        acc_id = cur.fetchone()[0]

        cur.execute('''
        INSERT INTO users (email, account_id, created_at, updated_at, confirmed_at)
        VALUES (%s, %s, now(), now(), now());
        ''', (email, acc_id))

        cur.execute('''
        INSERT INTO follows (account_id, target_account_id, created_at, updated_at)
        VALUES (%s, %s, now(), now());
        ''', (acc_id, acc_to_follow))

        db.commit()
        print("{} created".format(screenname))
    
    except psycopg2.IntegrityError:
        print("{} already exists".format(screenname))
        db.rollback()