-
Notifications
You must be signed in to change notification settings - Fork 0
/
add_extension.py
50 lines (42 loc) · 1.49 KB
/
add_extension.py
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
import pymysql
import redis
from mysqlcursor import cur, con
redis_client = redis.Redis(host='localhost', port=6379, db=0)
def option_6():
"This is to insert Extension Information into the database"
'''take the following as input
Extension_ID\n
SCOPE\n
MAG\n
GRIP\n
'''
try:
row = {}
Extension_ID = input("Enter Extension ID: ")
SCOPE = input("Enter SCOPE: ")
MAG = input("Enter MAG: ")
GRIP = input("Enter GRIP: ")
# Check if extension data exists in Redis cache
extension_key = f"extension:{Extension_ID}"
if redis_client.exists(extension_key):
print("Extension data found in cache:")
extension_data = redis_client.hgetall(extension_key)
print(extension_data)
else:
# Insert data into MySQL
query = "INSERT INTO Extension(Extension_ID, SCOPE, MAG, GRIP) VALUES (%s, %s, %s, %s)"
cur.execute(query, (Extension_ID, SCOPE, MAG, GRIP))
con.commit()
print("Inserted into database")
# Cache extension data in Redis
extension_data = {
'SCOPE': SCOPE,
'MAG': MAG,
'GRIP': GRIP
}
redis_client.hmset(extension_key, extension_data)
print("Extension data cached in Redis")
except Exception as e:
con.rollback()
print("Failed to insert into database")
print(">>>>>>>>>>>>>", e)