-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathgenerate_users.py
More file actions
57 lines (46 loc) · 1.35 KB
/
generate_users.py
File metadata and controls
57 lines (46 loc) · 1.35 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
import sqlite3
from contextlib import closing
from faker import Faker
SEED = 0
FILENAME = "users.db"
COUNT = 50
def generate_user(faker):
return {
"first_name": faker.first_name(),
"last_name": faker.last_name(),
"address": faker.address(),
"email": faker.email(),
}
def populate_database(db, faker, count):
with closing(db.cursor()) as cursor:
cursor.execute(
"""
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
first_name TEXT,
last_name TEXT,
address TEXT,
email TEXT
)
"""
)
cursor.execute("DELETE FROM users")
cursor.execute("DELETE FROM sqlite_sequence WHERE name='users'")
for _ in range(count):
user = generate_user(faker)
cursor.execute(
"""
INSERT INTO users (first_name, last_name, address, email)
VALUES (?, ?, ?, ?)
""",
(user["first_name"], user["last_name"], user["address"], user["email"]),
)
if __name__ == "__main__":
faker = Faker()
faker.seed_instance(SEED)
db = sqlite3.connect(FILENAME)
try:
with db:
populate_database(db, faker, COUNT)
finally:
db.close()