コード例 #1
0
ファイル: seed.py プロジェクト: amnh-library/qr_iousities
# @app.teardown_appcontext
# def close_connection(exception):
# db.close_connection(exception)

if __name__ == '__main__':
    db.execute('''DROP TABLE IF EXISTS items''')
    db.execute('''DROP TABLE IF EXISTS stats''')
    db.execute(
        '''CREATE TABLE items(TableID INTEGER PRIMARY KEY, ItemID text, Name text, Descriptor text, ShortUrl text, LongUrl text)'''
    )
    db.execute('''CREATE TABLE stats(TableID INTEGER, CreatedAt INTEGER)''')

    items = [
        ('BHL-001', 'T-Shirt', 'Super Item', 'rndmstrng',
         'http://nytimes.com'),
        ('BHL-002', 'Book', 'Super Books', 'tneoius', 'http://twitter.com'),
        ('BHL-003', 'Also Something', 'Super super super', 'shrtrl',
         'http://short.com'),
    ]

    db.executemany(
        '''INSERT INTO items(ItemID, Name, Descriptor, ShortUrl, LongUrl) VALUES (?,?,?,?,?)''',
        items)

    stats = [(randint(1, 6), int(time.time()) - randint(1, 86400000))
             for i in range(100006)]

    db.executemany('''INSERT INTO stats(TableId, CreatedAt) VALUES (?,?)''',
                   stats)