Init db.py
This commit is contained in:
commit
cf6521ff4a
|
@ -0,0 +1,29 @@
|
||||||
|
#!/usr/bin/python3
|
||||||
|
# -- coding: utf-8 --
|
||||||
|
|
||||||
|
import pickle, os
|
||||||
|
|
||||||
|
PATH = "/etc/bot/"
|
||||||
|
AUTH_FILE= "auth.pkl"
|
||||||
|
DIR_FILE = "dir.pkl"
|
||||||
|
|
||||||
|
def check(type,FILE):
|
||||||
|
if type == 'dir':
|
||||||
|
if os.path.exists(PATH) == False:
|
||||||
|
os.mkdir(PATH)
|
||||||
|
return True
|
||||||
|
elif type == 'obj':
|
||||||
|
if os.path.exists(f'{PATH}{FILE}'):
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|
||||||
|
def write(obj,FILE):
|
||||||
|
if check('dir',None):
|
||||||
|
with open(f'{PATH}{FILE}',"wb") as file:
|
||||||
|
pickle.dump(obj,file)
|
||||||
|
|
||||||
|
def read(FILE):
|
||||||
|
with open(f'{PATH}{FILE}',"rb") as file:
|
||||||
|
obj = pickle.load(file)
|
||||||
|
return obj
|
Reference in New Issue