mirror of
https://github.com/Xtao-Labs/iShotaBot.git
synced 2024-11-22 15:36:55 +00:00
58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
import pickle
|
|
from os import sep
|
|
from os.path import exists
|
|
|
|
|
|
def init() -> None:
|
|
data = {}
|
|
if not exists(f"data{sep}anti_channel.pkl"):
|
|
with open(f"data{sep}anti_channel.pkl", "wb") as f:
|
|
pickle.dump(data, f, pickle.HIGHEST_PROTOCOL)
|
|
|
|
|
|
def add(gid: int, cid: int) -> None:
|
|
with open(f"data{sep}anti_channel.pkl", "rb") as f:
|
|
data = pickle.load(f)
|
|
try:
|
|
if cid in data[gid]:
|
|
return
|
|
data[gid].append(cid)
|
|
except KeyError:
|
|
data[gid] = [cid]
|
|
with open(f"data{sep}anti_channel.pkl", "wb") as f:
|
|
pickle.dump(data, f, pickle.HIGHEST_PROTOCOL)
|
|
|
|
|
|
def clean(gid: int) -> None:
|
|
with open(f"data{sep}anti_channel.pkl", "rb") as f:
|
|
data = pickle.load(f)
|
|
try:
|
|
data[gid] = []
|
|
except KeyError:
|
|
return
|
|
with open(f"data{sep}anti_channel.pkl", "wb") as f:
|
|
pickle.dump(data, f, pickle.HIGHEST_PROTOCOL)
|
|
|
|
|
|
def get(gid: int) -> list:
|
|
with open(f"data{sep}anti_channel.pkl", "rb") as f:
|
|
data = pickle.load(f)
|
|
try:
|
|
return data[gid]
|
|
except KeyError:
|
|
return []
|
|
|
|
|
|
def get_status(gid: int) -> bool:
|
|
if len(get(gid)) == 0:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
|
|
def check_status(gid: int, cid: int) -> bool:
|
|
if cid in get(gid):
|
|
return True
|
|
else:
|
|
return False
|