-
Notifications
You must be signed in to change notification settings - Fork 0
/
retrieve_guns.py
58 lines (50 loc) · 2.11 KB
/
retrieve_guns.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
50
51
52
53
54
55
56
57
import pymysql
import redis
from mysqlcursor import cur, con
# Initialize Redis client
redis_client = redis.Redis(host='localhost', port=6379, db=0)
def option_21():
"This is to retrieve guns for a particular extension"
'''
Take the following as input
SCOPE\n
Or MAG\n
Or GRIP\n
'''
try:
row = {}
Extension_ID = input("Enter Extension ID: ")
cache_key = f"guns_for_extension:{Extension_ID}"
# Check if the result is cached in Redis
if redis_client.exists(cache_key):
print("Data found in cache:")
cached_data = redis_client.hgetall(cache_key)
print("Weapon_ID\tWeapon_Name\tAmmo\tFire_Rate\tDamage\tExtension_ID")
for key, value in cached_data.items():
print(value.decode('utf-8'))
else:
query = """
SELECT Weapons.Weapon_ID, Weapons.Weapon_Name, Weapons.Ammo, Weapons.Fire_Rate, Weapons.Damage, Weapons.Extension_ID
FROM Weapons
JOIN Extension ON Weapons.Extension_ID = Extension.Extension_ID
WHERE Extension.SCOPE = %s OR Extension.MAG = %s OR Extension.GRIP = %s
"""
cur.execute(query, (Extension_ID, Extension_ID, Extension_ID))
weapons = cur.fetchall()
if weapons:
print("Weapon_ID\tWeapon_Name\tAmmo\tFire_Rate\tDamage\tExtension_ID")
weapon_data = {}
for idx, row in enumerate(weapons):
weapon_str = f"{row['Weapon_ID']}\t{row['Weapon_Name']}\t{row['Ammo']}\t{row['Fire_Rate']}\t{row['Damage']}\t{row['Extension_ID']}"
print(weapon_str)
weapon_data[f"weapon_{idx}"] = weapon_str
# Cache the result in Redis
redis_client.hmset(cache_key, weapon_data)
print("Data cached in Redis")
else:
print("No guns found for the given extension")
con.commit()
except Exception as e:
con.rollback()
print("Failed to fetch from database")
print(">>>>>>>>>>>>>", e)