mayureshagashe2105 commited on
Commit
6bf71bb
·
1 Parent(s): 5d8f196

Add init scripts

Browse files
Files changed (1) hide show
  1. scripts/test.py +55 -0
scripts/test.py ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from dotenv import load_dotenv
2
+ load_dotenv()
3
+ import os
4
+
5
+ import mysql.connector
6
+ from mysql.connector import errorcode
7
+
8
+ config={
9
+ 'host':os.environ.get("HOSTNAME"),
10
+ 'user':os.environ.get("UID"),
11
+ 'password':os.environ.get("PASSWORD"),
12
+ 'database':os.environ.get("DATABASE")
13
+ }
14
+
15
+ print(config)
16
+
17
+ try:
18
+ cnx = mysql.connector.connect(**config)
19
+ print("Connection established")
20
+ except mysql.connector.Error as err:
21
+ if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
22
+ print("Something is wrong with username or password")
23
+ elif err.errno == errorcode.ER_BAD_DB_ERROR:
24
+ print("Database does not exist")
25
+ else:
26
+ print(err)
27
+ else:
28
+ cursor = cnx.cursor()
29
+
30
+ cursor.execute("DROP TABLE IF EXISTS api_key")
31
+ cursor.execute("DROP TABLE IF EXISTS auth")
32
+ cursor.execute("CREATE TABLE IF NOT EXISTS auth(username VARCHAR(15) PRIMARY KEY, password TEXT, email VARCHAR(50))")
33
+ cursor.execute("CREATE TABLE IF NOT EXISTS api_key(username VARCHAR(15),apikey TEXT, FOREIGN KEY (username) REFERENCES auth(username))")
34
+
35
+ QUERY = ('INSERT INTO {coll_name} '
36
+ '(username, password, email) '
37
+ 'VALUES '
38
+ '(%s, %s, %s)').format(coll_name="auth")
39
+
40
+ testlist=[("test2","test2","[email protected]"),("test1","test1","[email protected]")]
41
+ cursor.executemany(QUERY, testlist)
42
+
43
+ QUERY = ('SELECT {cols} FROM {table_name} WHERE email="[email protected]"').format(cols="*", table_name="auth")
44
+ cursor.execute(QUERY)
45
+ for i in cursor.fetchall():
46
+ print(i)
47
+
48
+
49
+
50
+
51
+ cnx.commit()
52
+ cursor.close()
53
+ cnx.close()
54
+ # from jose import jwt
55
+ # print(jwt.encode("bruhh"))