4
1
Fork 1

Init db.py

This commit is contained in:
trueold89 2023-10-31 01:00:58 +03:00
commit cf6521ff4a
Signed by: trueold89
GPG Key ID: C122E85DD49E6B30
1 changed files with 29 additions and 0 deletions

29
db.py Normal file
View File

@ -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