turdbot/py/storage.py

32 lines
1.0 KiB
Python
Raw Normal View History

2022-10-25 16:21:21 -06:00
# Copyright (c) 2022, Ian Burgess
# All rights reserved.
#
# This source code is licensed under the GPLv3 license. A copy of this license can be found in the LICENSE file in the root directory of this source tree.
2022-10-25 17:14:44 -06:00
import shelve
2022-10-25 16:21:21 -06:00
server = "default"
def config(value,db="config",serverid=server,mode="r"):
default = {
"replytobot":"false",
"triggerbotenabled":"true",
"quotebotenabled":"true",
"triggerbottriggers":["hello","hi","howdy"],
"quotequeue":"1010042640508669982",
}
2022-11-08 14:47:25 -07:00
try:
with shelve.open("bot.shlf",writeback=True) as data:
if mode == "r":
try:
return data[db][serverid][value]
except:
return default[value]
elif mode == "w":
data[db][serverid] = value
return("success")
else:
print("error")
except:
print("a storage error occured so default was returned")
return default[value]