srbhavya01 commited on
Commit
6f24918
·
verified ·
1 Parent(s): c23e255

Create database.py

Browse files
Files changed (1) hide show
  1. database.py +36 -0
database.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sqlite3
2
+
3
+ def create_table():
4
+ conn = sqlite3.connect("users.db")
5
+ c = conn.cursor()
6
+
7
+ c.execute("""
8
+ CREATE TABLE IF NOT EXISTS users(
9
+ email TEXT PRIMARY KEY
10
+ )
11
+ """)
12
+
13
+ conn.commit()
14
+ conn.close()
15
+
16
+
17
+ def add_user(email):
18
+ conn = sqlite3.connect("users.db")
19
+ c = conn.cursor()
20
+
21
+ c.execute("INSERT OR IGNORE INTO users VALUES (?)", (email,))
22
+
23
+ conn.commit()
24
+ conn.close()
25
+
26
+
27
+ def check_user(email):
28
+ conn = sqlite3.connect("users.db")
29
+ c = conn.cursor()
30
+
31
+ c.execute("SELECT * FROM users WHERE email=?", (email,))
32
+ user = c.fetchone()
33
+
34
+ conn.close()
35
+
36
+ return uoy