105 lines
3.8 KiB
Python
105 lines
3.8 KiB
Python
|
import sqlite3
|
||
|
|
from datetime import datetime
|
||
|
|
from config import DATABASE_PATH
|
||
|
|
|
||
|
|
# Règles de mercy pour stockage
|
||
|
|
MERCY_RULES = {
|
||
|
|
"ancient": {"threshold": 200, "increment": 0.5, "base": 0},
|
||
|
|
"void": {"threshold": 200, "increment": 0.5, "base": 0},
|
||
|
|
"sacred": {"threshold": 12, "increment": 2, "base": 0},
|
||
|
|
"primal_legendary": {"threshold": 75, "increment": 1, "base": 0},
|
||
|
|
"primal_mythical": {"threshold": 200, "increment": 10, "base": 0},
|
||
|
|
"remnant": {"threshold": 24, "increment": 1, "base": 0},
|
||
|
|
}
|
||
|
|
|
||
|
|
class MercyManager:
|
||
|
|
def __init__(self, db_path=DATABASE_PATH):
|
||
|
|
self.db_path = db_path
|
||
|
|
self.init_table()
|
||
|
|
|
||
|
|
def init_table(self):
|
||
|
|
conn = sqlite3.connect(self.db_path)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute("""
|
||
|
|
CREATE TABLE IF NOT EXISTS mercy_counters (
|
||
|
|
user_id TEXT,
|
||
|
|
shard_type TEXT,
|
||
|
|
pulls INTEGER DEFAULT 0,
|
||
|
|
last_reset TIMESTAMP,
|
||
|
|
PRIMARY KEY(user_id, shard_type)
|
||
|
|
)
|
||
|
|
""")
|
||
|
|
conn.commit()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
def get_pulls(self, user_id, shard_type):
|
||
|
|
conn = sqlite3.connect(self.db_path)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute(
|
||
|
|
"SELECT pulls FROM mercy_counters WHERE user_id = ? AND shard_type = ?",
|
||
|
|
(user_id, shard_type)
|
||
|
|
)
|
||
|
|
row = cursor.fetchone()
|
||
|
|
conn.close()
|
||
|
|
return row[0] if row else 0
|
||
|
|
|
||
|
|
def add_pulls(self, user_id, shard_type, pulls):
|
||
|
|
current = self.get_pulls(user_id, shard_type)
|
||
|
|
conn = sqlite3.connect(self.db_path)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
if current:
|
||
|
|
cursor.execute(
|
||
|
|
"UPDATE mercy_counters SET pulls = ?, last_reset = ? WHERE user_id = ? AND shard_type = ?",
|
||
|
|
(current + pulls, datetime.utcnow(), user_id, shard_type)
|
||
|
|
)
|
||
|
|
else:
|
||
|
|
cursor.execute(
|
||
|
|
"INSERT INTO mercy_counters (user_id, shard_type, pulls, last_reset) VALUES (?, ?, ?, ?)",
|
||
|
|
(user_id, shard_type, pulls, datetime.utcnow())
|
||
|
|
)
|
||
|
|
conn.commit()
|
||
|
|
conn.close()
|
||
|
|
return current + pulls
|
||
|
|
|
||
|
|
def reset_pulls(self, user_id, shard_type):
|
||
|
|
conn = sqlite3.connect(self.db_path)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute(
|
||
|
|
"UPDATE mercy_counters SET pulls = 0, last_reset = ? WHERE user_id = ? AND shard_type = ?",
|
||
|
|
(datetime.utcnow(), user_id, shard_type)
|
||
|
|
)
|
||
|
|
conn.commit()
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
def get_all_pulls(self, user_id):
|
||
|
|
conn = sqlite3.connect(self.db_path)
|
||
|
|
cursor = conn.cursor()
|
||
|
|
cursor.execute(
|
||
|
|
"SELECT shard_type, pulls FROM mercy_counters WHERE user_id = ?",
|
||
|
|
(user_id,)
|
||
|
|
)
|
||
|
|
rows = cursor.fetchall()
|
||
|
|
conn.close()
|
||
|
|
return {shard_type: pulls for shard_type, pulls in rows}
|
||
|
|
|
||
|
|
def get_mercy_chance(self, shard_type, pulls):
|
||
|
|
rule = MERCY_RULES[shard_type]
|
||
|
|
if pulls <= rule["threshold"]:
|
||
|
|
return rule["base"]
|
||
|
|
return rule["base"] + (pulls - rule["threshold"]) * rule["increment"]
|
||
|
|
|
||
|
|
def get_guaranteed_pull(self, shard_type):
|
||
|
|
"""Retourne le pull où la loot est garanti à 100%"""
|
||
|
|
rules = {
|
||
|
|
"ancient": {"start": 200, "increment": 5, "base": 0.5},
|
||
|
|
"void": {"start": 200, "increment": 5, "base": 0.5},
|
||
|
|
"sacred": {"start": 12, "increment": 2, "base": 6},
|
||
|
|
"primal_legendary": {"start": 75, "increment": 1, "base": 1},
|
||
|
|
"primal_mythical": {"start": 200, "increment": 10, "base": 0.5},
|
||
|
|
"remnant": {"start": 24, "increment": 1, "base": 0},
|
||
|
|
}
|
||
|
|
if shard_type not in rules:
|
||
|
|
return None
|
||
|
|
rule = rules[shard_type]
|
||
|
|
return int(rule["start"] + (100 - rule["base"]) / rule["increment"])
|