more fixes from Gamerbot
parent
4989f51b24
commit
8f67a175ae
@ -1,125 +0,0 @@
|
||||
import os, sys, shutil
|
||||
|
||||
# Nonstandard to avoid depend loop
|
||||
from logger import BotLogger
|
||||
|
||||
syslog = BotLogger("system_log.txt")
|
||||
|
||||
|
||||
def check(fn):
|
||||
if os.path.exists(fn):
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
|
||||
class simpledb:
|
||||
def __init__(self):
|
||||
self.dbroot = "server-data/"
|
||||
if not check(self.dbroot):
|
||||
os.makedirs(self.dbroot)
|
||||
|
||||
def get(self, server, key):
|
||||
serverdb = self.dbroot + server
|
||||
keyp = serverdb + "/" + key
|
||||
if not check(serverdb):
|
||||
return "NOT SET"
|
||||
else:
|
||||
if not check(keyp):
|
||||
return "NOT SET"
|
||||
else:
|
||||
with open(keyp) as f:
|
||||
return str(f.read())
|
||||
|
||||
def set(self, server, key, value):
|
||||
serverdb = self.dbroot + server
|
||||
keyp = serverdb + "/" + key
|
||||
if not check(serverdb):
|
||||
os.makedirs(serverdb)
|
||||
with open(keyp, "w") as f:
|
||||
f.write(str(value))
|
||||
|
||||
def exists(self, server):
|
||||
serverdb = self.dbroot + server
|
||||
if check(serverdb):
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
def erase(self, server):
|
||||
serverdb = self.dbroot + server
|
||||
if self.exists(server):
|
||||
shutil.rmtree(serverdb)
|
||||
|
||||
def listdata(self, server):
|
||||
serverdb = self.dbroot + server
|
||||
if self.exists(server):
|
||||
data = ""
|
||||
for f in os.listdir(serverdb):
|
||||
if f != "blank":
|
||||
data += f + " : " + open(serverdb + "/" + f).read() + "\n"
|
||||
return data
|
||||
else:
|
||||
return "No DB for " + server
|
||||
|
||||
|
||||
# ____________________________________________
|
||||
|
||||
|
||||
class serverconfig:
|
||||
def __init__(self):
|
||||
self.db = simpledb()
|
||||
|
||||
def showdata(self, server):
|
||||
server = str(server)
|
||||
data = self.db.listdata(server)
|
||||
out = "Data for `" + server + "`:\n```" + data + "```"
|
||||
return out
|
||||
|
||||
def checkinit(self, server):
|
||||
server = str(server)
|
||||
if self.db.exists(server):
|
||||
return True
|
||||
else:
|
||||
self.db.set(server, "blank", "blank")
|
||||
return False
|
||||
|
||||
def getstring(self, server, key):
|
||||
server = str(server)
|
||||
return self.db.get(server, key)
|
||||
|
||||
def getint(self, server, key):
|
||||
server = str(server)
|
||||
val = self.db.get(server, key)
|
||||
if val == "NOT SET":
|
||||
return -1
|
||||
else:
|
||||
return int(self.db.get(server, key))
|
||||
|
||||
def getbool(self, server, key):
|
||||
server = str(server)
|
||||
val = self.db.get(server, key)
|
||||
if val == "NOT SET":
|
||||
return False
|
||||
else:
|
||||
if "true" in self.db.get(server, key):
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
def set(self, server, key, value):
|
||||
server = str(server)
|
||||
key = str(key)
|
||||
value = str(value)
|
||||
self.db.set(server, key, value)
|
||||
|
||||
def rs(self, server):
|
||||
server = str(server)
|
||||
self.db.erase(server)
|
||||
|
||||
|
||||
# GUILD ID:
|
||||
# - mods
|
||||
# - swears_censored
|
||||
# - require_privileges
|
||||
# - announcements
|
Loading…
Reference in New Issue