def cmd_subscribe(message):
try:
user_objects = Dialogs.objects(index=message.from_user.id)
if len(user_objects) > 0:
user = user_objects[0]
else:
user = Dialogs(index=message.from_user.id)
if not user.subscribe:
user.subscribe = True
user.save()
try:
bot.send_message(
message.from_user.id,
'Теперь Вам будут приходить уведомления об изменении кармы.'
)
if message.location != 'private':
bot.reply_to(
message,
'Теперь Вам будут приходить уведомления об изменении кармы.'
)
except:
bot.reply_to(message, 'Сначала напиши мне в ЛС.')
else:
bot.reply_to(
message, 'Вы уже подписаны на уведомления об изменении кармы.')
except:
crash_message(message)
def cmd_start(message, user, chat):
typing(message)
if message.chat.type == 'private':
bot.send_message(
message.chat.id,
_('hello {user_name}', locale=user.locale).format(
user_name=get_username_or_name(message.from_user)))
def cmd_top(message):
try:
typing(message)
karma_records = Karma.objects(chat=message.chat.id)
users = {}
for karma in karma_records:
if karma.rollback:
continue
if karma.from_user and karma.transfer:
users[karma.from_user] = users.get(karma.from_user,
0) - karma.amount
if karma.to_user:
users[karma.to_user] = users.get(karma.to_user,
0) + karma.amount
text = ['Статистика:']
if bot_id in users:
users.pop(bot_id)
for index, (user, karma) in enumerate(sorted(users.items(),
key=lambda x: x[1],
reverse=True),
start=1):
if index > 10:
break
text.append('{}) {}: {}'.format(
index, get_username_or_name(bot.get_chat(user)), karma))
bot.send_message(message.chat.id,
'\n'.join(text),
disable_notification=True)
except:
crash_message(message)
def query_settings(callbackquery):
try:
key, data = parse_inline_data(callbackquery.data)
karma = Karma.get_transaction(data[0])
if karma:
if not sender_is_admin(karma.chat, callbackquery.from_user.id):
return bot.answer_callback_query(callbackquery.id,
'You shall not pass!')
karma.cancel()
if karma.rollback:
bot.answer_callback_query(
callbackquery.id,
'Transaction ' + data[0] + ' is canceled!')
text = '#karma #WARNING #Transaction\nCanceled: #{}\nby {} ({})'.format(
data[0], get_username_or_name(callbackquery.from_user),
callbackquery.from_user.id)
bot.send_message(LOGGING_CHAT, text)
notify_chat_admins(karma.chat, text)
else:
bot.answer_callback_query(
callbackquery.id,
'Transaction ' + data[0] + ' is activated!')
text = '#karma #WARNING #Transaction\nRestored: #{}\nby {} ({})'.format(
data[0], get_username_or_name(callbackquery.from_user),
callbackquery.from_user.id)
bot.send_message(LOGGING_CHAT, text)
notify_chat_admins(karma.chat, text)
bot.answer_callback_query(callbackquery.id, 'Wrong transaction!')
except:
with CrashReport(*sys.exc_info()) as c:
log.warning(c.formatted_traceback)
c.save()
def emit(self, record):
try:
if LOGGING_TO_CHAT:
bot.send_message(self.chat_id,
self.format(record),
parse_mode='html',
disable_web_page_preview=True)
except:
print(sys.exc_info())
def notify_chat_admins(chat_id, text, markup=None):
for user_id in AdminSubscribe.get_chat_subscribers(chat_id=chat_id):
try:
bot.send_message(user_id,
text,
parse_mode='html',
reply_markup=markup)
except:
pass
def crash_message(message):
c = CrashReport(*sys.exc_info())
try:
c.save()
telebot.logger.error('\n\t'.join(c.formatted_traceback))
bot.send_message(message.chat.id, 'Во время выполнения команды произошла ошибка :с\n'
'Если не сложно, перешли, пожалуйста это сообщение разработчику '
'{user}\n'
'Report: {id}-{file}'.format(user=get_username_or_name(root_user),
id=c.id, file=c.filename))
chat_log.error('Crash report: <b>{id}</b>\n'
'File: <b>{file}</b>\n'
'User: {login} ({uid})\n'
'<pre>{traceback}</pre>'.format(id=c.id, file=c.filename,
login=get_username_or_name(message.from_user),
uid=message.from_user.id,
traceback='\n'.join(c.formatted_traceback)))
except:
cc = CrashReport(*sys.exc_info())
print('\n'.join(c.formatted_traceback), file=sys.stderr)
print('\n'.join(cc.formatted_traceback), file=sys.stderr)
def cmd_messages_count(message):
try:
typing(message)
date = datetime.now(TIMEZONE) - timedelta(days=1)
users = Messages.calculate(message.chat.id, date)
text = ['Топ флудеров за последние 24 часа:']
if bot_id in users:
users.pop(bot_id)
for index, (user, karma) in enumerate(sorted(users.items(),
key=lambda x: x[1],
reverse=True),
start=1):
if index > 10:
break
text.append('{}) {}: {} сбщ.'.format(
index, get_username_or_name(bot.get_chat(user)), karma))
bot.send_message(message.chat.id,
'\n'.join(text),
disable_notification=True)
except:
crash_message(message)
def log_transaction(transaction,
chat=0,
from_user=0,
to_user=0,
amount=0,
description='',
transfer=True):
try:
if type(chat) is telebot.types.Chat:
chat = chat
else:
chat = bot.get_chat(chat)
except:
chat = None
try:
if type(from_user) is telebot.types.User:
from_user = from_user
else:
from_user = bot.get_chat(from_user)
except:
from_user = None
try:
if type(to_user) is telebot.types.User:
to_user = to_user
else:
to_user = bot.get_chat(to_user)
except:
to_user = None
message = ['#karma #INFO #Transaction ID: #{}'.format(transaction)]
if chat:
message.append('Dialog: {} (<code>{}</code>)'.format(
chat.title or get_username_or_name(chat), chat.id))
if from_user:
message.append('From: {} (<code>{}</code>)'.format(
get_username_or_name(from_user), from_user.id))
if transfer:
message.append('\tResult amount: <i>{}</i>'.format(
get_cached_user_chat(from_user.id,
chat.id).get(KARMA_FIELD_NAME, 0)))
if to_user:
message.append('To: {} (<code>{}</code>)'.format(
get_username_or_name(to_user), to_user.id))
message.append('\tResult amount: <i>{}</i>'.format(
get_cached_user_chat(to_user.id, chat.id).get(KARMA_FIELD_NAME,
0)))
else:
message.append('To: <code>/dev/null</code>')
message.append('Description: <code>{}</code>'.format(description))
message.append('Amount: <code>{}</code>'.format(amount))
markup = telebot.types.InlineKeyboardMarkup()
markup.add(
telebot.types.InlineKeyboardButton('Cancel',
callback_data=generate_inline_data(
'CANCEL_TRANSACTION',
[str(transaction)])))
bot.send_message(LOGGING_CHAT,
'\n'.join(message),
parse_mode='html',
reply_markup=markup)
notify_chat_admins(chat.id, '\n'.join(message), markup=markup)
def vote_message(message, description='', amount=1):
# TODO: anti spam
if bool(
message.reply_to_message
) and message.reply_to_message.from_user.id != message.from_user.id:
user_cache = get_cached_user_chat(message.from_user.id,
message.chat.id)
last_message = user_cache.get('last_message', 0)
timeout = user_cache.get('karma_change_timeout', ANTI_FLOOD_TIMEOUT)
warn_count = user_cache.get('karma_change_warn', 0) + 1
update_cached_user(
message.from_user.id, message.chat.id, {
'last_message':
time.time(),
'karma_change_timeout':
time.time() - last_message + ANTI_FLOOD_TIMEOUT * warn_count,
'karma_change_warn':
warn_count
})
if time.time() - last_message < timeout:
log.warning(
'#flood {user} (<code>{user_id}</code>) is flooding <b>x{count}</b>!\n'
'Chat "{chat}" (<code>{chat_id}</code>)\n'
'Set timeout: {timeout:.1f}'.format(
user=get_username_or_name(message.from_user),
user_id=message.from_user.id,
count=warn_count,
chat=message.chat.title,
chat_id=message.chat.id,
timeout=get_cached_user_chat(message.from_user.id,
message.chat.id).get(
'karma_change_timeout',
ANTI_FLOOD_TIMEOUT)))
return bot.reply_to(
message,
_('anti spam karma {timeout:.0f}s').format(
timeout=timeout - (time.time() - last_message)))
update_cached_user(message.from_user.id, message.chat.id, {
'karma_change_timeout': ANTI_FLOOD_TIMEOUT,
'karma_change_warn': 0
})
karma_transaction(message.chat,
message.from_user,
message.reply_to_message.from_user,
amount=amount,
description=bool_to_str(amount > 0, 'good message',
'bad message'),
transfer=False)
user = get_dialog_object(message.reply_to_message.from_user.id)
if user.subscribe:
try:
if amount > 0:
text = _('{user} thanked you chatting {chat}',
locale=user.locale)
elif amount < 0:
text = _(
'{user} did not like your message in the chat {chat}',
locale=user.locale)
else:
text = _(
'User {user} is indifferent to your message in the chat {chat}',
locale=user.locale)
if len(description):
text += '\n' + _('Comment: {text}')
bot.send_message(
message.reply_to_message.from_user.id,
text.format(user=get_username_or_name(message.from_user),
chat=get_chat_url_or_title(message),
text=description),
disable_web_page_preview=True,
parse_mode='markdown')
bot.forward_message(message.reply_to_message.from_user.id,
message.chat.id,
message.reply_to_message.message_id,
disable_notification=True)
except:
pass
