Раздеделение главного скрипта на под-скрипты для удобства разработки.
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
import os
|
||||
import json
|
||||
|
||||
# Создаём БД, если её нету
|
||||
if not os.path.exists('db.json'):
|
||||
db = {"token": "None"}
|
||||
js = json.dumps(db, indent=2)
|
||||
@@ -11,12 +12,57 @@ if not os.path.exists('db.json'):
|
||||
exit()
|
||||
|
||||
|
||||
# raw_db = {
|
||||
# "Anon": {"id": 2045634, pkey: "fuD2d", channel: "AnotherUser", avatar: "♿️", blocks: [5375652, 436432], keys: {AnotherUser: "dDH73s"}},
|
||||
# "2045634": "Anon"
|
||||
# }
|
||||
def read_db(file = 'db.json'):
|
||||
with open(file, "r", encoding="utf-8") as openfile:
|
||||
db = json.load(openfile)
|
||||
return db
|
||||
raw_db = json.load(openfile)
|
||||
return raw_db
|
||||
|
||||
def write_db(db, file = 'db.json'):
|
||||
js = json.dumps(db, indent=2, ensure_ascii=False)
|
||||
def write_db(raw_db, file = 'db.json'):
|
||||
js = json.dumps(raw_db, indent=2, ensure_ascii=False)
|
||||
with open(file, "w", encoding="utf-8") as outfile:
|
||||
outfile.write(js)
|
||||
|
||||
|
||||
|
||||
from user import *
|
||||
# db = {
|
||||
# "Anon": user.*,
|
||||
# "2045634": "Anon"
|
||||
# }
|
||||
|
||||
def is_num(str):
|
||||
try:
|
||||
int(str)
|
||||
return True
|
||||
except:
|
||||
return False
|
||||
|
||||
def load():
|
||||
raw_db = read_db()
|
||||
db = {}
|
||||
for i in raw_db:
|
||||
if is_num(i) == True:
|
||||
db[i] = raw_db[i]
|
||||
elif "token" == i:
|
||||
db["token"] = raw_db["token"]
|
||||
else:
|
||||
id, pkey, channel, avatar, blocks, keys = raw_db[i]["id"], raw_db[i]["pkey"], raw_db[i]["channel"], raw_db[i]["avatar"], raw_db[i]["blocks"], raw_db[i]["keys"],
|
||||
user = user_(id, pkey, channel, avatar, blocks, keys)
|
||||
db[i] = user
|
||||
|
||||
return db
|
||||
|
||||
def save(db):
|
||||
raw_db = {}
|
||||
for i in db:
|
||||
if is_num(i) == True:
|
||||
raw_db[i] = db[i]
|
||||
elif "token" == i:
|
||||
raw_db["token"] = db["token"]
|
||||
else:
|
||||
raw_db[i] = {"id": db[i].id, "pkey": db[i].pkey, "channel": db[i].channel, "avatar": db[i].avatar, "blocks": db[i].blocks, "keys": db[i].keys}
|
||||
write_db(raw_db)
|
||||
|
||||
Reference in New Issue
Block a user