def initial_handler(self, msg, chat_handler):
if len(Mute.select().where(Mute.user == chat_handler.user, Mute.until > datetime.datetime.now())) > 0:
groups = "Please respond with the name of the group you would like to mute\n\n"
groups += "Currently you are muting the following groups:\n"
options = []
for group_mem in Mute.select().where(Mute.user == chat_handler.user, Mute.until > datetime.datetime.now()):
groups += " " + group_mem.group.group_name + "\n"
options.append(group_mem.group.group_name)
self.current_handler = self.handle_group
yield from chat_handler.sender.sendMessage(groups, reply_markup={"keyboard": [options]})
else:
self.finished()
yield from chat_handler.sender.sendMessage(_s["msg_notmuting"])
def handle_time(self, msg, chat_handler):
if (yield from assert_text(msg,chat_handler)):
try:
mute_until = datetime.datetime.now() + datetime.timedelta(minutes=int(msg["text"]))
Mute.create(
user=chat_handler.user,
group=self._group,
until=mute_until)
self.finished()
yield from chat_handler.sender.sendMessage(_s["msg_mutegroupuntil"]+str(mute_until),
reply_markup={'hide_keyboard': True})
except ValueError:
yield from chat_handler.sender.sendMessage(_s["msg_expectint"])
def handle_time(self, msg, chat_handler):
if (yield from assert_text(msg, chat_handler)):
try:
mute_until = datetime.datetime.now() + datetime.timedelta(
minutes=int(msg["text"]))
Mute.create(user=chat_handler.user,
group=self._group,
until=mute_until)
self.finished()
yield from chat_handler.sender.sendMessage(
_s["msg_mutegroupuntil"] + str(mute_until),
reply_markup={'hide_keyboard': True})
except ValueError:
yield from chat_handler.sender.sendMessage(_s["msg_expectint"])
def initial_handler(self, msg, chat_handler):
if len(Mute.select().where(Mute.user == chat_handler.user,
Mute.until > datetime.datetime.now())) > 0:
groups = "Please respond with the name of the group you would like to mute\n\n"
groups += "Currently you are muting the following groups:\n"
options = []
for group_mem in Mute.select().where(
Mute.user == chat_handler.user,
Mute.until > datetime.datetime.now()):
groups += " " + group_mem.group.group_name + "\n"
options.append(group_mem.group.group_name)
self.current_handler = self.handle_group
yield from chat_handler.sender.sendMessage(
groups, reply_markup={'keyboard': [options]})
else:
self.finished()
yield from chat_handler.sender.sendMessage(_s["msg_notmuting"])
def handle_message(self, msg, chat_handler):
content_type, chat_type, chat_id = telepot.glance2(msg)
tasks = []
success = 0
failure = 0
muted = 0
for group_membership in GroupMembership.select().where(GroupMembership.group == self._group):
try:
if not Mute.select().where(Mute.group == self._group, Mute.user==group_membership.user, Mute.until > datetime.datetime.now()):
telegram_id = group_membership.user.telegram_id
main_character_name = chat_handler.user.main_character.name
group_name = self._group.group_name
if content_type == "text":
tasks.append(self.safe_send(telegram_id, main_character_name, group_name, msg))
elif content_type == "photo":
tasks.append(self.bot.sendMessage(telegram_id,"Ping from:" + main_character_name + " to " + group_name))
tasks.append(self.bot.sendPhoto(telegram_id, msg["photo"][1]["file_id"], caption=msg["caption"]))
elif content_type == "document":
tasks.append(self.bot.sendMessage(telegram_id,"Ping from:" + main_character_name + " to " + group_name))
tasks.append(self.bot.sendDocument(telegram_id, msg["document"]["file_id"]))
elif content_type == "voice":
tasks.append(self.bot.sendMessage(telegram_id,"Ping from:" + main_character_name + " to " + group_name))
tasks.append(self.bot.sendVoice(telegram_id, msg["voice"]["file_id"]))
elif content_type == "video":
tasks.append(self.bot.sendMessage(telegram_id,"Ping from:" + main_character_name + " to " + group_name))
tasks.append(self.bot.sendVideo(telegram_id, msg["video"]["file_id"]))
elif content_type == "sticker":
tasks.append(self.bot.sendMessage(telegram_id,"Ping from:" + main_character_name + " to " + group_name))
tasks.append(self.bot.sendSticker(telegram_id, msg["sticker"]["file_id"]))
success += 1
else:
muted += 1
except:
failure += 1
self.finished()
start_time = time.time()
yield from chat_handler.throttle(tasks)
elapsed = time.time() - start_time
rounded = math.floor(elapsed*100)/100
yield from chat_handler.sender.sendMessage(
"Ping sent to "+str(success)+" users in "+str(rounded)+" seconds"
)
def handle_group(self, msg, chat_handler):
if (yield from assert_text(msg, chat_handler)):
group_name = msg["text"]
if len(Group.select().where(Group.group_name == group_name)) > 0:
self._group = Group.select().where(Group.group_name == group_name).get()
mute_instance = Mute.select().where(Mute.user == chat_handler.user, Mute.group == self._group).get()
mute_instance.delete_instance()
self.finished()
yield from chat_handler.sender.sendMessage(_s["msg_groupunmuted"], reply_markup={"hide_keyboard": True})
else:
yield from chat_handler.sender.sendMessage(_s["msg_notagroup"])
def handle_group(self, msg, chat_handler):
if (yield from assert_text(msg,chat_handler)):
group_name = msg["text"]
if len(Group.select().where(Group.group_name == group_name)) > 0:
self._group = Group.select().where(Group.group_name == group_name).get()
if len(Mute.select().where(Mute.group == self._group, Mute.user == chat_handler.user)) == 0:
show_keyboard = {'keyboard': [['60','1440'],['2160','10080']]}
self.current_handler = self.handle_time
yield from chat_handler.sender.sendMessage(_s["msg_howlongmute"], reply_markup=show_keyboard)
else:
self.cancel()
yield from chat_handler.sender.sendMessage("You are already muting this group",reply_markup={'hide_keyboard': True})
else:
yield from chat_handler.sender.sendMessage(_s["msg_notagroup"])
def handle_group(self, msg, chat_handler):
if (yield from assert_text(msg, chat_handler)):
group_name = msg["text"]
if len(Group.select().where(Group.group_name == group_name)) > 0:
self._group = Group.select().where(
Group.group_name == group_name).get()
mute_instance = Mute.select().where(
Mute.user == chat_handler.user,
Mute.group == self._group).get()
mute_instance.delete_instance()
self.finished()
yield from chat_handler.sender.sendMessage(
_s["msg_groupunmuted"],
reply_markup={'hide_keyboard': True})
else:
yield from chat_handler.sender.sendMessage(_s["msg_notagroup"])
def handle_group(self, msg, chat_handler):
if (yield from assert_text(msg, chat_handler)):
group_name = msg["text"]
if len(Group.select().where(Group.group_name == group_name)) > 0:
self._group = Group.select().where(
Group.group_name == group_name).get()
if len(Mute.select().where(
Mute.group == self._group,
Mute.user == chat_handler.user)) == 0:
show_keyboard = {
'keyboard': [['60', '1440'], ['2160', '10080']]
}
self.current_handler = self.handle_time
yield from chat_handler.sender.sendMessage(
_s["msg_howlongmute"], reply_markup=show_keyboard)
else:
self.cancel()
yield from chat_handler.sender.sendMessage(
"You are already muting this group",
reply_markup={'hide_keyboard': True})
else:
yield from chat_handler.sender.sendMessage(_s["msg_notagroup"])
def handle_message(self, msg, chat_handler):
content_type, chat_type, chat_id = telepot.glance2(msg)
tasks = []
success = 0
failure = 0
muted = 0
for group_membership in GroupMembership.select().where(
GroupMembership.group == self._group):
try:
if not Mute.select().where(
Mute.group == self._group, Mute.user
== group_membership.user,
Mute.until > datetime.datetime.now()):
telegram_id = group_membership.user.telegram_id
main_character_name = chat_handler.user.main_character.name
group_name = self._group.group_name
if content_type == "text":
tasks.append(
self.safe_send(telegram_id, main_character_name,
group_name, msg))
elif content_type == "photo":
tasks.append(
self.bot.sendMessage(
telegram_id, "Ping from:" +
main_character_name + " to " + group_name))
tasks.append(
self.bot.sendPhoto(telegram_id,
msg["photo"][1]["file_id"],
caption=msg["caption"]))
elif content_type == "document":
tasks.append(
self.bot.sendMessage(
telegram_id, "Ping from:" +
main_character_name + " to " + group_name))
tasks.append(
self.bot.sendDocument(telegram_id,
msg["document"]["file_id"]))
elif content_type == "voice":
tasks.append(
self.bot.sendMessage(
telegram_id, "Ping from:" +
main_character_name + " to " + group_name))
tasks.append(
self.bot.sendVoice(telegram_id,
msg["voice"]["file_id"]))
elif content_type == "video":
tasks.append(
self.bot.sendMessage(
telegram_id, "Ping from:" +
main_character_name + " to " + group_name))
tasks.append(
self.bot.sendVideo(telegram_id,
msg["video"]["file_id"]))
elif content_type == "sticker":
tasks.append(
self.bot.sendMessage(
telegram_id, "Ping from:" +
main_character_name + " to " + group_name))
tasks.append(
self.bot.sendSticker(telegram_id,
msg["sticker"]["file_id"]))
success += 1
else:
muted += 1
except:
failure += 1
self.finished()
start_time = time.time()
yield from chat_handler.throttle(tasks)
elapsed = time.time() - start_time
rounded = math.floor(elapsed * 100) / 100
yield from chat_handler.sender.sendMessage("Ping sent to " +
str(success) +
" users in " +
str(rounded) + " seconds")
