populate_db.py 文件源码

def main(total_users, host, user, password, db_name):
    engine = sqlalchemy.create_engine(
        'mysql+pymysql://{user}:{password}@{host}/{db_name}'.format(
            user=user,
            password=parse.quote_plus(password),
            host=host,
            db_name=db_name))
    session = create_session(engine)

    try:
        populate_db(session, total_users)
    finally:
        session.close()