def check_flood(update, context) -> str: user = update.effective_user # type: Optional[User] chat = update.effective_chat # type: Optional[Chat] msg = update.effective_message # type: Optional[Message] if not user: # ignore channels return "" # ignore admins if is_user_admin(chat, user.id): sql.update_flood(chat.id, None) return "" should_ban = sql.update_flood(chat.id, user.id) if not should_ban: return "" try: getmode, getvalue = sql.get_flood_setting(chat.id) if getmode == 1: chat.kick_member(user.id) execstrings = tl(update.effective_message, "Keluar!") tag = "BANNED" elif getmode == 2: chat.kick_member(user.id) chat.unban_member(user.id) execstrings = tl(update.effective_message, "Keluar!") tag = "KICKED" elif getmode == 3: context.bot.restrict_chat_member(chat.id, user.id, permissions=ChatPermissions(can_send_messages=False)) execstrings = tl(update.effective_message, "Sekarang kamu diam!") tag = "MUTED" elif getmode == 4: bantime = extract_time(msg, getvalue) chat.kick_member(user.id, until_date=bantime) execstrings = tl(update.effective_message, "Keluar selama {}!").format(getvalue) tag = "TBAN" elif getmode == 5: mutetime = extract_time(msg, getvalue) context.bot.restrict_chat_member(chat.id, user.id, until_date=mutetime, permissions=ChatPermissions(can_send_messages=False)) execstrings = tl(update.effective_message, "Sekarang kamu diam selama {}!").format(getvalue) tag = "TMUTE" send_message(update.effective_message, tl(update.effective_message, "Saya tidak suka orang yang mengirim pesan beruntun. Tapi kamu hanya membuat " "saya kecewa. {}").format(execstrings)) return "<b>{}:</b>" \ "\n#{}" \ "\n<b>User:</b> {}" \ "\nFlooded the group.".format(tag, html.escape(chat.title), mention_html(user.id, user.first_name)) except BadRequest: send_message(update.effective_message, tl(update.effective_message, "Saya tidak bisa menendang orang di sini, beri saya izin terlebih dahulu! Sampai saat itu, saya akan menonaktifkan antiflood.")) sql.set_flood(chat.id, 0) return "<b>{}:</b>" \ "\n#INFO" \ "\n{}".format(chat.title, tl(update.effective_message, "Tidak memiliki izin kick, jadi secara otomatis menonaktifkan antiflood."))
def set_flood(update, context) -> str: chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] message = update.effective_message # type: Optional[Message] args = context.args conn = connected(context.bot, update, chat, user.id, need_admin=True) if conn: chat_id = conn chat_name = dispatcher.bot.getChat(conn).title else: if update.effective_message.chat.type == "private": send_message(update.effective_message, tl(update.effective_message, "Anda bisa lakukan command ini pada grup, bukan pada PM")) return "" chat_id = update.effective_chat.id chat_name = update.effective_message.chat.title if len(args) >= 1: val = args[0].lower() if val == "off" or val == "no" or val == "0": sql.set_flood(chat_id, 0) if conn: text = tl(update.effective_message, "Antiflood telah dinonaktifkan di *{}*.").format(chat_name) else: text = tl(update.effective_message, "Antiflood telah dinonaktifkan.") send_message(update.effective_message, text, parse_mode="markdown") elif val.isdigit(): amount = int(val) if amount <= 0: sql.set_flood(chat_id, 0) if conn: text = tl(update.effective_message, "Antiflood telah dinonaktifkan di *{}*.").format(chat_name) else: text = tl(update.effective_message, "Antiflood telah dinonaktifkan.") return "<b>{}:</b>" \ "\n#SETFLOOD" \ "\n<b>Admin:</b> {}" \ "\nDisable antiflood.".format(html.escape(chat_name), mention_html(user.id, user.first_name)) elif amount < 3: send_message(update.effective_message, tl(update.effective_message, "Antiflood harus baik 0 (dinonaktifkan), atau nomor lebih besar dari 3!")) return "" else: sql.set_flood(chat_id, amount) if conn: text = tl(update.effective_message, "Antiflood telah diperbarui dan diatur menjadi *{}* pada *{}*").format(amount, chat_name) else: text = tl(update.effective_message, "Antiflood telah diperbarui dan diatur menjadi *{}*").format(amount) send_message(update.effective_message, text, parse_mode="markdown") return "<b>{}:</b>" \ "\n#SETFLOOD" \ "\n<b>Admin:</b> {}" \ "\nSet antiflood to <code>{}</code>.".format(html.escape(chat_name), mention_html(user.id, user.first_name), amount) else: send_message(update.effective_message, tl(update.effective_message, "Argumen tidak dikenal - harap gunakan angka, 'off', atau 'no'.")) else: send_message(update.effective_message, tl(update.effective_message, "Gunakan `/setflood nomor` untuk menyetel anti pesan beruntun.\nAtau gunakan `/setflood off` untuk menonaktifkan anti pesan beruntun."), parse_mode="markdown") return ""
def FLOOD_EDITBTN(bot: Bot, update: Update): query = update.callback_query user = update.effective_user print("User {} clicked button FLOOD EDIT".format(user.id)) qdata = query.data.split("=")[1].split("|")[0] chat_id = query.data.split("|")[1] if qdata == "?": bot.answerCallbackQuery( query.id, "Batas dari pesan beruntun. Jika pengguna mengirim pesan lebih dari batas, maka akan langsung di banned.", show_alert=True) if qdata == "-": button = [] limit = sql.get_flood_limit(chat_id) limit = int(limit) - 1 if limit == 0: status = "❎ Tidak Aktif" else: status = "✅ Aktif" if limit <= 2: bot.answerCallbackQuery(query.id, "Batas limit Tidak boleh kurang dari 3", show_alert=True) return sql.set_flood(chat_id, int(limit)) chat = bot.get_chat(chat_id) text = "*{}* memiliki pengaturan berikut untuk modul *Anti Pesan Beruntun*:\n\n".format( escape_markdown(chat.title)) text += "Batas maksimal pesan beruntun telah di setel menjadi `{}`.".format( limit) button.append([ InlineKeyboardButton( text="➖", callback_data="set_flim=-|{}".format(chat_id)), InlineKeyboardButton( text="Limit {}".format(limit), callback_data="set_flim=?|{}".format(chat_id)), InlineKeyboardButton(text="➕", callback_data="set_flim=+|{}".format(chat_id)) ]) button.append([ InlineKeyboardButton( text="{}".format(status), callback_data="set_flim=exec|{}".format(chat_id)) ]) button.append([ InlineKeyboardButton( text="Kembali", callback_data="stngs_back({})".format(chat_id)) ]) query.message.edit_text(text=text, parse_mode=ParseMode.MARKDOWN, reply_markup=InlineKeyboardMarkup(button)) bot.answer_callback_query(query.id) if qdata == "+": button = [] limit = sql.get_flood_limit(chat_id) limit = int(limit) + 1 if limit == 0: status = "❎ Tidak Aktif" else: status = "✅ Aktif" if limit <= 0: bot.answerCallbackQuery(query.id, "Batas limit Tidak boleh kurang dari 0", show_alert=True) return sql.set_flood(chat_id, int(limit)) chat = bot.get_chat(chat_id) text = "*{}* memiliki pengaturan berikut untuk modul *Anti Pesan Beruntun*:\n\n".format( escape_markdown(chat.title)) text += "Batas maksimal pesan beruntun telah di setel menjadi `{}`.".format( limit) button.append([ InlineKeyboardButton( text="➖", callback_data="set_flim=-|{}".format(chat_id)), InlineKeyboardButton( text="Limit {}".format(limit), callback_data="set_flim=?|{}".format(chat_id)), InlineKeyboardButton(text="➕", callback_data="set_flim=+|{}".format(chat_id)) ]) button.append([ InlineKeyboardButton( text="{}".format(status), callback_data="set_flim=exec|{}".format(chat_id)) ]) button.append([ InlineKeyboardButton( text="Kembali", callback_data="stngs_back({})".format(chat_id)) ]) query.message.edit_text(text=text, parse_mode=ParseMode.MARKDOWN, reply_markup=InlineKeyboardMarkup(button)) bot.answer_callback_query(query.id) if qdata == "exec": button = [] limit = sql.get_flood_limit(chat_id) if limit == 0: sql.set_flood(chat_id, 3) limit = 3 status = "✅ Aktif ({})".format(limit) else: sql.set_flood(chat_id, 0) limit = 0 status = "❎ Tidak Aktif" chat = bot.get_chat(chat_id) text = "*{}* memiliki pengaturan berikut untuk modul *Anti Pesan Beruntun*:\n\n".format( escape_markdown(chat.title)) text += "Batas maksimal pesan beruntun telah di setel menjadi `{}`.".format( status) button.append([ InlineKeyboardButton( text="➖", callback_data="set_flim=-|{}".format(chat_id)), InlineKeyboardButton( text="Limit {}".format(limit), callback_data="set_flim=?|{}".format(chat_id)), InlineKeyboardButton(text="➕", callback_data="set_flim=+|{}".format(chat_id)) ]) button.append([ InlineKeyboardButton( text="{}".format(status), callback_data="set_flim=exec|{}".format(chat_id)) ]) button.append([ InlineKeyboardButton( text="Kembali", callback_data="stngs_back({})".format(chat_id)) ]) query.message.edit_text(text=text, parse_mode=ParseMode.MARKDOWN, reply_markup=InlineKeyboardMarkup(button)) bot.answer_callback_query(query.id)
def import_data(update, context): msg = update.effective_message # type: Optional[Message] chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] # TODO: allow uploading doc with command, not just as reply # only work with a doc 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": send_message(update.effective_message, tl(update.effective_message, "You can do this command in groups, not PM")) return "" chat = update.effective_chat chat_id = update.effective_chat.id chat_name = update.effective_message.chat.title if msg.reply_to_message and msg.reply_to_message.document: filetype = msg.reply_to_message.document.file_name if filetype.split('.')[-1] not in ("backup", "json", "txt"): send_message(update.effective_message, tl(update.effective_message, "Invalid backup file!")) return try: file_info = context.bot.get_file(msg.reply_to_message.document.file_id) except BadRequest: send_message(update.effective_message, tl(update.effective_message, "Try downloading and reuploading the file as yourself before importing - this one seems to be iffy!")) return with BytesIO() as file: file_info.download(out=file) file.seek(0) data = json.load(file) try: # If backup is from Emilia if data.get('bot_base') == "Emilia": imp_antiflood = False imp_blacklist = False imp_blacklist_count = 0 imp_blsticker = False imp_blsticker_count = 0 imp_disabled_count = 0 imp_filters_count = 0 imp_greet = False imp_gdbye = False imp_greet_pref = False imp_locks = False imp_notes = 0 imp_report = False imp_rules = False imp_lang = False imp_warn = False imp_warn_chat = 0 imp_warn_filter = 0 NOT_IMPORTED = "This cannot be imported because from other bot." NOT_IMPORTED_INT = 0 # If backup is from this bot, import all files if data.get('bot_id') == context.bot.id: is_self = True else: is_self = False if data.get('bot_base') == "Emilia": is_emilia = True else: is_emilia = False # Import antiflood if data.get('antiflood'): imp_antiflood = True flood_limit = data['antiflood'].get('flood_limit') flood_mode = data['antiflood'].get('flood_mode') flood_duration = data['antiflood'].get('flood_duration') # Add to db antifloodsql.set_flood(chat_id, int(flood_limit)) antifloodsql.set_flood_strength(chat_id, flood_mode, flood_duration) # Import blacklist if data.get('blacklists'): imp_blacklist = True blacklist_mode = data['blacklists'].get('blacklist_mode') blacklist_duration = data['blacklists'].get('blacklist_duration') blacklisted = data['blacklists'].get('blacklists') # Add to db blacklistsql.set_blacklist_strength(chat_id, blacklist_mode, blacklist_duration) if blacklisted: for x in blacklisted: blacklistsql.add_to_blacklist(chat_id, x.lower()) imp_blacklist_count += 1 # Import blacklist sticker if data.get('blstickers'): imp_blsticker = True blsticker_mode = data['blstickers'].get('blsticker_mode') blsticker_duration = data['blstickers'].get('blsticker_duration') blstickers = data['blstickers'].get('blstickers') # Add to db blackliststksql.set_blacklist_strength(chat_id, blsticker_mode, blsticker_duration) if blstickers: for x in blstickers: blackliststksql.add_to_stickers(chat_id, x.lower()) imp_blsticker_count += 1 # Import disabled if data.get('disabled'): candisable = disabledsql.get_disableable() if data['disabled'].get('disabled'): for listdisabled in data['disabled'].get('disabled'): if listdisabled in candisable: disabledsql.disable_command(chat_id, listdisabled) imp_disabled_count += 1 # Import filters if data.get('filters'): NOT_IMPORTED += "\n\nFilters:\n" for x in data['filters'].get('filters'): # If from self, import all if is_self: is_sticker = False is_document = False is_image = False is_audio = False is_voice = False is_video = False has_markdown = False universal = False if x['type'] == 1: is_sticker = True elif x['type'] == 2: is_document = True elif x['type'] == 3: is_image = True elif x['type'] == 4: is_audio = True elif x['type'] == 5: is_voice = True elif x['type'] == 6: is_video = True elif x['type'] == 0: has_markdown = True note_data, buttons = button_markdown_parser(x['reply'], entities=0) filtersql.add_filter(chat_id, x['name'], note_data, is_sticker, is_document, is_image, is_audio, is_voice, is_video, buttons) imp_filters_count += 1 elif is_emilia: is_sticker = False is_document = False is_image = False is_audio = False is_voice = False is_video = False has_markdown = False universal = False if x['type'] == 0: has_markdown = True else: NOT_IMPORTED += "- {}\n".format(x['name']) NOT_IMPORTED_INT += 1 continue note_data, buttons = button_markdown_parser(x['reply'], entities=0) filtersql.add_filter(chat_id, x['name'], note_data, is_sticker, is_document, is_image, is_audio, is_voice, is_video, buttons) imp_filters_count += 1 else: if x['has_markdown']: note_data, buttons = button_markdown_parser(x['reply'], entities=0) filtersql.add_filter(chat_id, x['name'], note_data, False, False, False, False, False, False, buttons) imp_filters_count += 1 else: NOT_IMPORTED += "- {}\n".format(x['name']) NOT_IMPORTED_INT += 1 # Import greetings if data.get('greetings'): if data['greetings'].get('welcome'): welcenable = data['greetings']['welcome'].get('enable') welcsql.set_welc_preference(str(chat_id), bool(welcenable)) welctext = data['greetings']['welcome'].get('text') welctype = data['greetings']['welcome'].get('type') if welctype == 0: welctype = Types.TEXT elif welctype == 1: welctype = Types.BUTTON_TEXT elif welctype == 2: welctype = Types.STICKER elif welctype == 3: welctype = Types.DOCUMENT elif welctype == 4: welctype = Types.PHOTO elif welctype == 5: welctype = Types.AUDIO elif welctype == 6: welctype = Types.VOICE elif welctype == 7: welctype = Types.VIDEO elif welctype == 8: welctype = Types.VIDEO_NOTE else: welctype = None welccontent = data['greetings']['welcome'].get('content') if welctext and welctype: note_data, buttons = button_markdown_parser(welctext, entities=0) welcsql.set_custom_welcome(chat_id, welccontent, note_data, welctype, buttons) imp_greet = True if data['greetings'].get('goodbye'): gdbyenable = data['greetings']['goodbye'].get('enable') welcsql.set_gdbye_preference(str(chat_id), bool(gdbyenable)) gdbytext = data['greetings']['goodbye'].get('text') gdbytype = data['greetings']['goodbye'].get('type') if gdbytype == 0: gdbytype = Types.TEXT elif gdbytype == 1: gdbytype = Types.BUTTON_TEXT elif gdbytype == 2: gdbytype = Types.STICKER elif gdbytype == 3: gdbytype = Types.DOCUMENT elif gdbytype == 4: gdbytype = Types.PHOTO elif gdbytype == 5: gdbytype = Types.AUDIO elif gdbytype == 6: gdbytype = Types.VOICE elif gdbytype == 7: gdbytype = Types.VIDEO elif gdbytype == 8: gdbytype = Types.VIDEO_NOTE else: gdbytype = None gdbycontent = data['greetings']['goodbye'].get('content') if welctext and gdbytype: note_data, buttons = button_markdown_parser(gdbytext, entities=0) welcsql.set_custom_gdbye(chat_id, gdbycontent, note_data, gdbytype, buttons) imp_gdbye = True # clean service cleanserv = data['greetings'].get('clean_service') welcsql.set_clean_service(chat_id, bool(cleanserv)) # security welcome if data['greetings'].get('security'): secenable = data['greetings']['security'].get('enable') secbtn = data['greetings']['security'].get('text') sectime = data['greetings']['security'].get('time') extra_verify = data['greetings']['security'].get('extra_verify') if not extra_verify: extra_verify = False timeout = data['greetings']['security'].get('timeout') if not timeout: timeout = "0" timeout_mode = data['greetings']['security'].get('timeout_mode') if not timeout_mode: timeout_mode = 1 welcsql.set_welcome_security(chat_id, extra_verify, bool(secenable), str(sectime), str(timeout), int(timeout_mode), str(secbtn)) imp_greet_pref = True # Import language if data['greetings'].get('language'): lang = data['language'].get('language') if lang: if lang in ('en', 'id'): langsql.set_lang(chat_id, lang) imp_lang = True # Import Locks if data.get('locks'): if data['locks'].get('lock_warn'): locksql.set_lockconf(chat_id, True) else: locksql.set_lockconf(chat_id, False) if data['locks'].get('locks'): for x in list(data['locks'].get('locks')): if x in LOCK_TYPES: is_locked = data['locks']['locks'].get('x') locksql.update_lock(chat_id, x, locked=is_locked) imp_locks = True # Import notes if data.get('notes'): allnotes = data['notes'] NOT_IMPORTED += "\n\nNotes:\n" for x in allnotes: # If from self, import all if is_self: note_data, buttons = button_markdown_parser(x['note_data'], entities=0) note_name = x['note_tag'] note_file = None note_type = x['note_type'] if x['note_file']: note_file = x['note_file'] if note_type == 0: note_type = Types.TEXT elif note_type == 1: note_type = Types.BUTTON_TEXT elif note_type == 2: note_type = Types.STICKER elif note_type == 3: note_type = Types.DOCUMENT elif note_type == 4: note_type = Types.PHOTO elif note_type == 5: note_type = Types.AUDIO elif note_type == 6: note_type = Types.VOICE elif note_type == 7: note_type = Types.VIDEO elif note_type == 8: note_type = Types.VIDEO_NOTE else: note_type = None if note_type <= 8: notesql.add_note_to_db(chat_id, note_name, note_data, note_type, buttons, note_file) imp_notes += 1 elif is_emilia: note_data, buttons = button_markdown_parser(x['note_data'], entities=0) note_name = x['note_tag'] note_file = None note_type = x['note_type'] if x['note_file']: note_file = x['note_file'] if note_type == 0: note_type = Types.TEXT elif note_type == 1: note_type = Types.BUTTON_TEXT else: NOT_IMPORTED += "- {}\n".format(x['note_tag']) NOT_IMPORTED_INT += 1 continue if note_type <= 8: notesql.add_note_to_db(chat_id, note_name, note_data, note_type, buttons, note_file) imp_notes += 1 else: # If this text if x['note_type'] == 0: note_data, buttons = button_markdown_parser(x['text'].replace("\\", ""), entities=0) note_name = x['name'] notesql.add_note_to_db(chat_id, note_name, note_data, Types.TEXT, buttons, None) imp_notes += 1 else: NOT_IMPORTED += "- {}\n".format(x['name']) NOT_IMPORTED_INT += 1 # Import reports if data.get('report'): reporting = data['report'].get('report') reportsql.set_chat_setting(chat_id, bool(reporting)) imp_report = True # Import rules if data.get('rules'): contrules = data['rules'].get('rules') if contrules: rulessql.set_rules(chat_id, contrules) imp_rules = True # Import warn config if data.get('warns'): warn_limit = data['warns'].get('warn_limit') if warn_limit >= 3: warnssql.set_warn_limit(chat_id, int(warn_limit)) warn_mode = data['warns'].get('warn_mode') if warn_mode: if warn_mode <= 3: warnssql.set_warn_mode(chat_id, int(warn_mode)) imp_warn = True # Import all warn filters if data['warns'].get('warn_filters'): for x in data['warns'].get('warn_filters'): warnssql.add_warn_filter(chat_id, x['name'], x['reason']) imp_warn_filter += 1 # Import all warn from backup chat, reset first for prevent overwarn if data['warns'].get('chat_warns'): for x in data['warns'].get('chat_warns'): # If this invaild if x['warns'] > warn_limit: break warnssql.reset_warns(x['user_id'], chat_id) warnssql.import_warns(x['user_id'], chat_id, int(x['warns']), x['reasons']) imp_warn_chat += 1 if conn: text = tl(update.effective_message, "Backup fully imported in *{}*. Welcome back! 😀").format(chat_name) else: text = tl(update.effective_message, "Backup fully imported. Welcome back! 😀").format(chat_name) text += tl(update.effective_message, "\n\nRestored:\n") if imp_antiflood: text += tl(update.effective_message, "- Antiflood Settings\n") if imp_blacklist: text += tl(update.effective_message, "- Blacklist Settings\n") if imp_blacklist_count: text += tl(update.effective_message, "- {} blacklists\n").format(imp_blacklist_count) if imp_blsticker: text += tl(update.effective_message, "- {} blacklisted stickers\n").format(imp_blsticker_count) if imp_disabled_count: text += tl(update.effective_message, "- {} cmd disabled\n").format(imp_disabled_count) if imp_filters_count: text += tl(update.effective_message, "- {} filters\n").format(imp_filters_count) if imp_greet_pref: text += tl(update.effective_message, "- Greeting settings\n") if imp_greet: text += tl(update.effective_message, "- Greetings message\n") if imp_gdbye: text += tl(update.effective_message, "- Goodbye message\n") if imp_locks: text += tl(update.effective_message, "- Locked\n") if imp_notes: text += tl(update.effective_message, "- {} Notes\n").format(imp_notes) if imp_report: text += tl(update.effective_message, "- Reporting settings\n") if imp_rules: text += tl(update.effective_message, "- Group rules\n") if imp_lang: text += tl(update.effective_message, "- Language settings\n") if imp_warn: text += tl(update.effective_message, "- Warn settings\n") if imp_warn_chat: text += tl(update.effective_message, "- {} user warnings\n").format(imp_warn_chat) if imp_warn_filter: text += tl(update.effective_message, "- {} warning filters\n").format(imp_warn_filter) try: send_message(update.effective_message, text, parse_mode="markdown") except BadRequest: send_message(update.effective_message, text, parse_mode="markdown", quote=False) if NOT_IMPORTED_INT: f = open("{}-notimported.txt".format(chat_id), "w") f.write(str(NOT_IMPORTED)) f.close() context.bot.sendDocument(chat_id, document=open('{}-notimported.txt'.format(chat_id), 'rb'), caption=tl(update.effective_message, "*Data yang tidak dapat di import*"), timeout=360, parse_mode=ParseMode.MARKDOWN) os.remove("{}-notimported.txt".format(chat_id)) return except Exception as err: send_message(update.effective_message, tl(update.effective_message, "An exception occured while restoring your data from Emilia backup!\n\nSorry."), parse_mode="markdown") LOGGER.exception("An error when importing from Emilia base!") return try: # If backup is from rose # doing manual lol if data.get('bot_id') == 609517172: imp_antiflood = False imp_blacklist = False imp_blacklist_count = 0 imp_disabled_count = 0 imp_filters_count = 0 imp_greet = False imp_gdbye = False imp_greet_pref = False imp_notes = 0 imp_report = False imp_rules = False imp_lang = False imp_warn = False NOT_IMPORTED = "This cannot be imported because from other bot." NOT_IMPORTED_INT = 0 if data.get('data'): # Import antiflood if data['data'].get('antiflood'): floodlimit = data['data']['antiflood'].get('flood_limit') action = data['data']['antiflood'].get('action') actionduration = data['data']['antiflood'].get('action_duration') act_dur = make_time(int(actionduration)) antifloodsql.set_flood(chat_id, int(floodlimit)) if action == "ban": antifloodsql.set_flood_strength(chat_id, 1, str(act_dur)) imp_antiflood = True elif action == "kick": antifloodsql.set_flood_strength(chat_id, 2, str(act_dur)) imp_antiflood = True elif action == "mute": antifloodsql.set_flood_strength(chat_id, 3, str(act_dur)) imp_antiflood = True # Import blacklist if data['data'].get('blacklists'): action = data['data']['blacklists'].get('action') actionduration = data['data']['blacklists'].get('action_duration') act_dur = make_time(int(actionduration)) strengthdone = False if action == "del": strengthdone = True blacklistsql.set_blacklist_strength(chat_id, 1, str(act_dur)) imp_blacklist = True elif action == "warn": strengthdone = True blacklistsql.set_blacklist_strength(chat_id, 2, str(act_dur)) imp_blacklist = True elif action == "mute": strengthdone = True blacklistsql.set_blacklist_strength(chat_id, 3, str(act_dur)) imp_blacklist = True elif action == "kick": strengthdone = True blacklistsql.set_blacklist_strength(chat_id, 4, str(act_dur)) imp_blacklist = True elif action == "ban": strengthdone = True blacklistsql.set_blacklist_strength(chat_id, 5, str(act_dur)) imp_blacklist = True else: if not strengthdone: action = data['data']['blacklists'].get('should_delete') if action: blacklistsql.set_blacklist_strength(chat_id, 1, "0") imp_blacklist = True blacklisted = data['data']['blacklists'].get('filters') if blacklisted: for x in blacklisted: blacklistsql.add_to_blacklist(chat_id, x['name'].lower()) imp_blacklist_count += 1 # Import disabled if data['data'].get('disabled'): if data['data']['disabled'].get('disabled'): candisable = disabledsql.get_disableable() for listdisabled in data['data']['disabled'].get('disabled'): if listdisabled in candisable: disabledsql.disable_command(chat_id, listdisabled) imp_disabled_count += 1 # Import filters if data['data'].get('filters'): NOT_IMPORTED += "\n\nFilters:\n" if data['data']['filters'].get('filters'): for x in data['data']['filters'].get('filters'): if x['type'] == 0: note_data, buttons = button_markdown_parser(x['text'].replace("\\", ""), entities=0) filtersql.add_filter(chat_id, x['name'], note_data, False, False, False, False, False, False, buttons) imp_filters_count += 1 else: NOT_IMPORTED += "- {}\n".format(x['name']) NOT_IMPORTED_INT += 1 # Import greetings if data['data'].get('greetings'): if data['data']['greetings'].get('welcome'): welctext = data['data']['greetings']['welcome'].get('text') if welctext: note_data, buttons = button_markdown_parser(welctext.replace("\\", ""), entities=0) welcsql.set_custom_welcome(chat_id, None, note_data, Types.TEXT, buttons) imp_greet = True if data['data']['greetings'].get('goodbye'): gdbytext = data['data']['greetings']['goodbye'].get('text') if welctext: note_data, buttons = button_markdown_parser(gdbytext.replace("\\", ""), entities=0) welcsql.set_custom_gdbye(chat_id, None, note_data, Types.TEXT, buttons) imp_gdbye = True # Welcome config if data['data']['greetings'].get('should_welcome'): welcsql.set_welc_preference(str(chat_id), True) else: welcsql.set_welc_preference(str(chat_id), False) # Goodbye config if data['data']['greetings'].get('should_goodbye'): welcsql.set_gdbye_preference(str(chat_id), True) else: welcsql.set_gdbye_preference(str(chat_id), False) # clean service if data['data']['greetings'].get('should_delete_service'): welcsql.set_clean_service(chat_id, True) else: welcsql.set_clean_service(chat_id, False) # custom mute btn if data['data']['greetings'].get('mute_text'): getcur, cur_value, extra_verify, timeout, timeout_mode, cust_text = welcsql.welcome_security(chat_id) welcsql.set_welcome_security(chat_id, getcur, extra_verify, cur_value, timeout, timeout_mode, data['data']['greetings'].get('mute_text')) imp_greet_pref = True # TODO parsing unix time and import that # TODO Locks # Import notes if data['data'].get('notes'): NOT_IMPORTED += "\n\nNotes:\n" allnotes = data['data']['notes']['notes'] for x in allnotes: # If this text if x['type'] == 0: note_data, buttons = button_markdown_parser(x['text'].replace("\\", ""), entities=0) note_name = x['name'] notesql.add_note_to_db(chat_id, note_name, note_data, Types.TEXT, buttons, None) imp_notes += 1 else: NOT_IMPORTED += "- {}\n".format(x['name']) NOT_IMPORTED_INT += 1 # Import reports if data['data'].get('reports'): if data['data']['reports'].get('disable_reports'): reporting = False else: reporting = True reportsql.set_chat_setting(chat_id, reporting) imp_report = True # Import rules if data['data'].get('rules'): contrules = data['data']['rules'].get('content') if contrules: rulessql.set_rules(chat_id, contrules.replace("\\", "")) imp_rules = True # Import current lang if data['data'].get('translations'): lang = data['data']['translations'].get('lang') if lang: if lang in ('en', 'id'): langsql.set_lang(chat_id, lang) imp_lang = True # Import warn if data['data'].get('warns'): action = data['data']['warns'].get('action') # actionduration = data['data']['warns'].get('action_duration') # act_dur = make_time(int(actionduration)) if action == "kick": warnssql.set_warn_mode(chat_id, 1) imp_warn = True elif action == "ban": warnssql.set_warn_mode(chat_id, 2) imp_warn = True elif action == "mute": warnssql.set_warn_mode(chat_id, 3) imp_warn = True if conn: text = tl(update.effective_message, "Cadangan sepenuhnya dikembalikan pada *{}*. Selamat datang kembali! 😀").format(chat_name) else: text = tl(update.effective_message, "Cadangan sepenuhnya dikembalikan. Selamat datang kembali! 😀").format(chat_name) text += tl(update.effective_message, "\n\nYang saya kembalikan:\n") if imp_antiflood: text += tl(update.effective_message, "- Pengaturan Antiflood\n") if imp_blacklist: text += tl(update.effective_message, "- Pengaturan Blacklist\n") if imp_blacklist_count: text += tl(update.effective_message, "- {} blacklists\n").format(imp_blacklist_count) if imp_disabled_count: text += tl(update.effective_message, "- {} cmd disabled\n").format(imp_disabled_count) if imp_filters_count: text += tl(update.effective_message, "- {} filters\n").format(imp_filters_count) if imp_greet_pref: text += tl(update.effective_message, "- Pengaturan salam\n") if imp_greet: text += tl(update.effective_message, "- Pesan salam\n") if imp_gdbye: text += tl(update.effective_message, "- Pesan selamat tinggal\n") if imp_notes: text += tl(update.effective_message, "- {} catatan\n").format(imp_notes) if imp_report: text += tl(update.effective_message, "- Pengaturan pelaporan\n") if imp_rules: text += tl(update.effective_message, "- Pesan peraturan grup\n") if imp_lang: text += tl(update.effective_message, "- Pengaturan bahasa\n") if imp_warn: text += tl(update.effective_message, "- Pengaturan peringatan\n") try: send_message(update.effective_message, text, parse_mode="markdown") except BadRequest: send_message(update.effective_message, text, parse_mode="markdown", quote=False) if NOT_IMPORTED_INT: f = open("{}-notimported.txt".format(chat_id), "w") f.write(str(NOT_IMPORTED)) f.close() context.bot.sendDocument(chat_id, document=open('{}-notimported.txt'.format(chat_id), 'rb'), caption=tl(update.effective_message, "*Data yang tidak dapat di import*"), timeout=360, parse_mode=ParseMode.MARKDOWN) os.remove("{}-notimported.txt".format(chat_id)) return except Exception as err: send_message(update.effective_message, tl(update.effective_message, "Telah terjadi kesalahan dalam import backup Rose!\nGabung ke [Grup support](https://t.me/joinchat/Fykz0VTMpqZvlkb8S0JevQ) kami untuk melaporkan dan mengatasi masalah ini!\n\nTerima kasih"), parse_mode="markdown") LOGGER.exception("An error when importing from Rose base!") return # only import one group if len(data) > 1 and str(chat_id) not in data: send_message(update.effective_message, tl(update.effective_message, "Ada lebih dari satu grup di file ini, dan tidak ada yang memiliki id obrolan yang sama dengan" "grup ini - bagaimana cara memilih apa yang akan diimpor?")) return # Check if backup is this chat try: if data.get(str(chat_id)) == None: if conn: text = tl(update.effective_message, "Backup berasal chat lain, Saya tidak bisa mengembalikan chat lain kedalam chat *{}*").format(chat_name) else: text = tl(update.effective_message, "Backup berasal chat lain, Saya tidak bisa mengembalikan chat lain kedalam chat ini") return send_message(update.effective_message, text, parse_mode="markdown") except: return send_message(update.effective_message, tl(update.effective_message, "Telah terjadi error dalam pengecekan data, silahkan laporkan kepada pembuat saya " "untuk masalah ini untuk membuat saya lebih baik! Terima kasih! 🙂")) # Check if backup is from self try: if str(bot.id) != str(data[str(chat_id)]['bot']): return send_message(update.effective_message, tl(update.effective_message, "Backup berasal dari bot lain, dokumen, foto, video, audio, suara tidak akan " "bekerja, jika file anda tidak ingin hilang, import dari bot yang dicadangkan." "jika masih tidak bekerja, laporkan pada pembuat bot tersebut untuk " "membuat saya lebih baik! Terima kasih! 🙂")) except: pass # Select data source if str(chat_id) in data: data = data[str(chat_id)]['hashes'] else: data = data[list(data.keys())[0]]['hashes'] try: for mod in DATA_IMPORT: mod.__import_data__(str(chat_id), data) except Exception: send_message(update.effective_message, tl(update.effective_message, "Kesalahan terjadi saat memulihkan data Anda. Prosesnya mungkin tidak lengkap. Jika " "Anda mengalami masalah dengan ini, pesan @AyraHikari dengan file cadangan Anda, jadi " "masalah bisa di-debug. Pemilik saya akan dengan senang hati membantu, dan setiap bug " "dilaporkan membuat saya lebih baik! Terima kasih! 🙂")) LOGGER.exception("Impor untuk id chat %s dengan nama %s gagal.", str(chat_id), str(chat.title)) return # TODO: some of that link logic # NOTE: consider default permissions stuff? if conn: text = tl(update.effective_message, "Cadangan sepenuhnya dikembalikan pada *{}*. Selamat datang kembali! 😀").format(chat_name) else: text = tl(update.effective_message, "Cadangan sepenuhnya dikembalikan. Selamat datang kembali! 😀").format(chat_name) send_message(update.effective_message, text, parse_mode="markdown")
def set_flood(bot: Bot, update: Update, args: List[str]) -> str: chat = update.effective_chat # type: Optional[Chat] user = update.effective_user # type: Optional[User] message = update.effective_message # type: Optional[Message] spam = spamfilters(update.effective_message.text, update.effective_message.from_user.id, update.effective_chat.id) if spam == True: return update.effective_message.reply_text( "Saya kecewa dengan anda, saya tidak akan mendengar kata-kata anda sekarang!" ) conn = connected(bot, update, chat, user.id, need_admin=True) if conn: chat_id = conn chat_name = dispatcher.bot.getChat(conn).title else: if update.effective_message.chat.type == "private": update.effective_message.reply_text( "Anda bisa lakukan command ini pada grup, bukan pada PM") return "" chat_id = update.effective_chat.id chat_name = update.effective_message.chat.title if len(args) >= 1: val = args[0].lower() if val == "off" or val == "no" or val == "0": sql.set_flood(chat_id, 0) if conn: text = "Antiflood telah dinonaktifkan di *{}*.".format( chat_name) else: text = "Antiflood telah dinonaktifkan." message.reply_text(text, parse_mode="markdown") elif val.isdigit(): amount = int(val) if amount <= 0: sql.set_flood(chat_id, 0) if conn: text = "Antiflood telah dinonaktifkan di *{}*.".format( chat_name) else: text = "Antiflood telah dinonaktifkan." return "<b>{}:</b>" \ "\n#SETFLOOD" \ "\n<b>Admin:</b> {}" \ "\nNonaktifkan antiflood.".format(html.escape(chat_name), mention_html(user.id, user.first_name)) elif amount < 3: message.reply_text( "Antiflood harus baik 0 (dinonaktifkan), atau nomor lebih besar dari 3!" ) return "" else: sql.set_flood(chat_id, amount) if conn: text = "Antiflood telah diperbarui dan diset menjadi *{}* pada *{}*".format( amount, chat_name) else: text = "Antiflood telah diperbarui dan diset menjadi *{}*".format( amount) message.reply_text(text, parse_mode="markdown") return "<b>{}:</b>" \ "\n#SETFLOOD" \ "\n<b>Admin:</b> {}" \ "\nSetel antiflood ke <code>{}</code>.".format(html.escape(chat_name), mention_html(user.id, user.first_name), amount) else: message.reply_text( "Argumen tidak dikenal - harap gunakan angka, 'off', atau 'no'." ) else: message.reply_text( "Gunakan `/setflood nomor` untuk menyetel anti pesan beruntun.\nAtau gunakan `/setflood off` untuk menonaktifkan anti pesan beruntun.", parse_mode="markdown") return ""