You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
72 lines
2.7 KiB
72 lines
2.7 KiB
import asyncio
|
|
from cmath import inf
|
|
import aiomysql
|
|
from steam import SteamID
|
|
from colors import *
|
|
|
|
class Database:
|
|
database = None
|
|
permitions = {"VIP":("20:oa",2592000)}
|
|
|
|
def __init__(self, host, user, password, db):
|
|
self.host = host
|
|
self.user = user
|
|
self.password = password
|
|
self.db = db
|
|
|
|
async def connect(self):
|
|
info("Create database pool")
|
|
self.database = await aiomysql.create_pool(
|
|
host = self.host,
|
|
user = self.user,
|
|
password = self.password,
|
|
db=self.db
|
|
)
|
|
|
|
async def disconnect(self):
|
|
ok("Disconnect from database")
|
|
await self.database.wait_closed()
|
|
|
|
async def processing_vip(self, steamid: SteamID, amount, status = "VIP", who = "Donate.User"):
|
|
check_status = await self.check_update_VIP(steamid)
|
|
if check_status:
|
|
if check_status > 1:
|
|
info(f"Append more time to {steamid.community_url}")
|
|
return await self.update_user(steamid, amount)
|
|
else:
|
|
info(f"Create new vip user {steamid.community_url}")
|
|
return await self.add_new_user(steamid, amount, status, who)
|
|
else:
|
|
warning(f"{steamid.community_url} currently have ADMIN/MOD permitions")
|
|
return False
|
|
|
|
async def check_update_VIP(self, steamid: SteamID):
|
|
# 0 = mod/admin | -1 = not found | 1 = need update
|
|
query = f"SELECT * FROM `sm_admins` WHERE `identity` LIKE '{steamid.id2_zero}' ORDER BY `reg_date` DESC"
|
|
async with self.database.acquire() as connection:
|
|
async with connection.cursor(aiomysql.DictCursor) as cursor:
|
|
await cursor.execute(query)
|
|
for row in await cursor.fetchall():
|
|
if row["status"] == "VIP" and row["amount"] > 0:
|
|
return 1
|
|
else:
|
|
return 0
|
|
return -1
|
|
|
|
async def update_user(self, steamid: SteamID, amount):
|
|
query = f"UPDATE `sm_admins` SET `amount`=`amount`+{amount}, `reg_date`=`reg_date` WHERE `identity` LIKE '{steamid.id2_zero}'"
|
|
async with self.database.acquire() as connection:
|
|
async with connection.cursor(aiomysql.DictCursor) as cursor:
|
|
await cursor.execute(query)
|
|
await connection.commit()
|
|
return 101
|
|
|
|
async def add_new_user(self, steamid: SteamID, amount, status, who):
|
|
settings = self.permitions.get(status)
|
|
level, flags = settings[0].split(":")
|
|
query = f"INSERT INTO `sm_admins` (`id`, `authtype`, `identity`, `password`, `flags`, `name`, `immunity`, `comment`, `status`, `reg_date`, `amount`) VALUES (NULL, 'steam', '{steamid.id2_zero}', NULL, '{flags}', '', '{level}', '{who}', '{status}', CURRENT_TIMESTAMP,'{amount}')"
|
|
async with self.database.acquire() as connection:
|
|
async with connection.cursor(aiomysql.DictCursor) as cursor:
|
|
await cursor.execute(query)
|
|
await connection.commit()
|
|
return 100
|
|
|