54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
import yaml
|
|
import pymysql.cursors
|
|
import asyncio
|
|
import json
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
with open("/home/phxntx5/public_html/Bonfire/config.yml", "r") as f:
|
|
global_config = yaml.load(f)
|
|
|
|
connection = None
|
|
|
|
botDescription = global_config.get("description")
|
|
commandPrefix = global_config.get("command_prefix")
|
|
|
|
battleWins = global_config.get("battleWins", [])
|
|
defaultStatus = global_config.get("default_status", "")
|
|
botToken = global_config.get("bot_token", "")
|
|
ownerID = global_config.get("owner_id", "")
|
|
|
|
|
|
def getCursor():
|
|
global connection
|
|
connection = pymysql.connect(host=global_config.get("db_host"), user=global_config.get("db_user"),
|
|
password=global_config.get("db_user_pass"), charset='utf8mb4',
|
|
cursorclass=pymysql.cursors.DictCursor)
|
|
return connection.cursor()
|
|
|
|
|
|
def closeConnection():
|
|
connection.commit()
|
|
connection.close()
|
|
|
|
def saveContent(key: str, content):
|
|
with open("/home/phxntx5/public_html/Bonfire/config.json", "r+") as jf:
|
|
data = json.load(jf)
|
|
jf.seek(0)
|
|
newData = dict(data)
|
|
newData[key] = content
|
|
jf.truncate()
|
|
try:
|
|
json.dump(newData, jf, indent=4)
|
|
except:
|
|
json.dump(data, jf, indent=4)
|
|
|
|
|
|
def getContent(key: str):
|
|
try:
|
|
with open("/home/phxntx5/public_html/Bonfire/config.json", "r+") as jf:
|
|
data = json.load(jf)
|
|
return data[key]
|
|
except KeyError:
|
|
return None
|