def clearall_btn(update: Update, context: CallbackContext):
query = update.callback_query
chat = update.effective_chat
message = update.effective_message
member = chat.get_member(query.from_user.id)
if query.data == "notes_rmall":
if member.status == "creator" or query.from_user.id in SUDO_USERS:
note_list = sql.get_all_chat_notes(chat.id)
try:
for notename in note_list:
note = notename.name.lower()
sql.rm_note(chat.id, note)
message.edit_text("Deleted all notes.")
except BadRequest:
return
if member.status == "administrator":
query.answer("Only owner of the chat can do this.")
if member.status == "member":
query.answer("You need to be admin to do this.")
elif query.data == "notes_cancel":
if member.status == "creator" or query.from_user.id in SUDO_USERS:
message.edit_text("Clearing of all notes has been cancelled.")
return
if member.status == "administrator":
query.answer("Only owner of the chat can do this.")
if member.status == "member":
query.answer("You need to be admin to do this.")
def list_notes(update: Update, context: CallbackContext):
chat_id = update.effective_chat.id
note_list = sql.get_all_chat_notes(chat_id)
notes = len(note_list) + 1
msg = "Get note by `/notenumber` or `#notename` \n\n *ID* *Note* \n"
for note_id, note in zip(range(1, notes), note_list):
if note_id < 10:
note_name = f"`{note_id:2}.` `#{(note.name.lower())}`\n"
else:
note_name = f"`{note_id}.` `#{(note.name.lower())}`\n"
if len(msg) + len(note_name) > MAX_MESSAGE_LENGTH:
update.effective_message.reply_text(msg,
parse_mode=ParseMode.MARKDOWN)
msg = ""
msg += note_name
if not note_list:
try:
update.effective_message.reply_text("No notes in this chat!")
except BadRequest:
update.effective_message.reply_text("No notes in this chat!",
quote=False)
elif len(msg) != 0:
update.effective_message.reply_text(msg, parse_mode=ParseMode.MARKDOWN)
def slash_get(update: Update, context: CallbackContext):
message, chat_id = update.effective_message.text, update.effective_chat.id
no_slash = message[1:]
note_list = sql.get_all_chat_notes(chat_id)
try:
noteid = note_list[int(no_slash) - 1]
note_name = str(noteid).strip(">").split()[1]
get(update, context, note_name, show_none=False)
except IndexError:
update.effective_message.reply_text("Wrong Note ID 😾")
def export_data(update, context):
chat_data = context.chat_data
msg = update.effective_message # type: Optional[Message]
user = update.effective_user # type: Optional[User]
chat_id = update.effective_chat.id
chat = update.effective_chat
current_chat_id = update.effective_chat.id
conn = connected(context.bot, update, chat, user.id, need_admin=True)
if conn:
chat = dispatcher.bot.getChat(conn)
chat_id = conn
# chat_name = dispatcher.bot.getChat(conn).title
else:
if update.effective_message.chat.type == "private":
update.effective_message.reply_text(
"This is a group only command!")
return ""
chat = update.effective_chat
chat_id = update.effective_chat.id
# chat_name = update.effective_message.chat.title
jam = time.time()
new_jam = jam + 10800
checkchat = get_chat(chat_id, chat_data)
if checkchat.get("status"):
if jam <= int(checkchat.get("value")):
timeformatt = time.strftime("%H:%M:%S %d/%m/%Y",
time.localtime(checkchat.get("value")))
update.effective_message.reply_text(
"You can only backup once a day!\nYou can backup again in about `{}`"
.format(timeformatt),
parse_mode=ParseMode.MARKDOWN,
)
return
else:
if user.id != OWNER_ID:
put_chat(chat_id, new_jam, chat_data)
else:
if user.id != OWNER_ID:
put_chat(chat_id, new_jam, chat_data)
note_list = sql.get_all_chat_notes(chat_id)
backup = {}
# button = ""
buttonlist = []
namacat = ""
isicat = ""
rules = ""
count = 0
countbtn = 0
# Notes
for note in note_list:
count += 1
# getnote = sql.get_note(chat_id, note.name)
namacat += "{}<###splitter###>".format(note.name)
if note.msgtype == 1:
tombol = sql.get_buttons(chat_id, note.name)
# keyb = []
for btn in tombol:
countbtn += 1
if btn.same_line:
buttonlist.append(
("{}".format(btn.name), "{}".format(btn.url), True))
else:
buttonlist.append(
("{}".format(btn.name), "{}".format(btn.url), False))
isicat += "###button###: {}<###button###>{}<###splitter###>".format(
note.value, str(buttonlist))
buttonlist.clear()
elif note.msgtype == 2:
isicat += "###sticker###:{}<###splitter###>".format(note.file)
elif note.msgtype == 3:
isicat += "###file###:{}<###TYPESPLIT###>{}<###splitter###>".format(
note.file, note.value)
elif note.msgtype == 4:
isicat += "###photo###:{}<###TYPESPLIT###>{}<###splitter###>".format(
note.file, note.value)
elif note.msgtype == 5:
isicat += "###audio###:{}<###TYPESPLIT###>{}<###splitter###>".format(
note.file, note.value)
elif note.msgtype == 6:
isicat += "###voice###:{}<###TYPESPLIT###>{}<###splitter###>".format(
note.file, note.value)
elif note.msgtype == 7:
isicat += "###video###:{}<###TYPESPLIT###>{}<###splitter###>".format(
note.file, note.value)
elif note.msgtype == 8:
isicat += "###video_note###:{}<###TYPESPLIT###>{}<###splitter###>".format(
note.file, note.value)
else:
isicat += "{}<###splitter###>".format(note.value)
notes = {
"#{}".format(namacat.split("<###splitter###>")[x]):
"{}".format(isicat.split("<###splitter###>")[x])
for x in range(count)
}
# Rules
rules = rulessql.get_rules(chat_id)
# Blacklist
bl = list(blacklistsql.get_chat_blacklist(chat_id))
# Disabled command
disabledcmd = list(disabledsql.get_all_disabled(chat_id))
# Filters (TODO)
"""
all_filters = list(filtersql.get_chat_triggers(chat_id))
export_filters = {}
for filters in all_filters:
filt = filtersql.get_filter(chat_id, filters)
# print(vars(filt))
if filt.is_sticker:
tipefilt = "sticker"
elif filt.is_document:
tipefilt = "doc"
elif filt.is_image:
tipefilt = "img"
elif filt.is_audio:
tipefilt = "audio"
elif filt.is_voice:
tipefilt = "voice"
elif filt.is_video:
tipefilt = "video"
elif filt.has_buttons:
tipefilt = "button"
buttons = filtersql.get_buttons(chat.id, filt.keyword)
print(vars(buttons))
elif filt.has_markdown:
tipefilt = "text"
if tipefilt == "button":
content = "{}#=#{}|btn|{}".format(tipefilt, filt.reply, buttons)
else:
content = "{}#=#{}".format(tipefilt, filt.reply)
print(content)
export_filters[filters] = content
print(export_filters)
"""
# Welcome (TODO)
# welc = welcsql.get_welc_pref(chat_id)
# Locked
curr_locks = locksql.get_locks(chat_id)
curr_restr = locksql.get_restr(chat_id)
if curr_locks:
locked_lock = {
"sticker": curr_locks.sticker,
"audio": curr_locks.audio,
"voice": curr_locks.voice,
"document": curr_locks.document,
"video": curr_locks.video,
"contact": curr_locks.contact,
"photo": curr_locks.photo,
"gif": curr_locks.gif,
"url": curr_locks.url,
"bots": curr_locks.bots,
"forward": curr_locks.forward,
"game": curr_locks.game,
"location": curr_locks.location,
"rtl": curr_locks.rtl,
}
else:
locked_lock = {}
if curr_restr:
locked_restr = {
"messages":
curr_restr.messages,
"media":
curr_restr.media,
"other":
curr_restr.other,
"previews":
curr_restr.preview,
"all":
all([
curr_restr.messages,
curr_restr.media,
curr_restr.other,
curr_restr.preview,
]),
}
else:
locked_restr = {}
locks = {"locks": locked_lock, "restrict": locked_restr}
# Warns (TODO)
# warns = warnssql.get_warns(chat_id)
# Backing up
backup[chat_id] = {
"bot": context.bot.id,
"hashes": {
"info": {
"rules": rules
},
"extra": notes,
"blacklist": bl,
"disabled": disabledcmd,
"locks": locks,
},
}
baccinfo = json.dumps(backup, indent=4)
with open("Yui_GBot{}.backup".format(chat_id), "w") as f:
f.write(str(baccinfo))
context.bot.sendChatAction(current_chat_id, "upload_document")
tgl = time.strftime("%H:%M:%S - %d/%m/%Y", time.localtime(time.time()))
try:
context.bot.sendMessage(
JOIN_LOGGER,
"*Successfully imported backup:*\nChat: `{}`\nChat ID: `{}`\nOn: `{}`"
.format(chat.title, chat_id, tgl),
parse_mode=ParseMode.MARKDOWN,
)
except BadRequest:
pass
context.bot.sendDocument(
current_chat_id,
document=open("Yui_GBot{}.backup".format(chat_id), "rb"),
caption=
"*Successfully Exported backup:*\nChat: `{}`\nChat ID: `{}`\nOn: `{}`\n\nNote: This `Yui_GBot-Backup` was specially made for notes."
.format(chat.title, chat_id, tgl),
timeout=360,
reply_to_message_id=msg.message_id,
parse_mode=ParseMode.MARKDOWN,
)
os.remove("Yui_GBot{}.backup".format(chat_id)) # Cleaning file
def __chat_settings__(chat_id, user_id):
notes = sql.get_all_chat_notes(chat_id)
return f"There are `{len(notes)}` notes in this chat."