Beispiel #1
0
    def close_spider(self, spider):
        student_misc = Misc.query.filter(Misc.chatID == spider.chatID).first()
        try:
            bot.send_photo(
                chat_id=spider.chatID,
                photo=open("files/{}_attendance.png".format(spider.username),
                           'rb'),
                caption='Attendance Report for {}'.format(spider.username))
            if student_misc is not None and student_misc.attendance_target is not None:
                target = student_misc.attendance_target
                no_of_lectures = int(until_x(spider.chatID, target))
                if no_of_lectures > 0:
                    messageContent = "You need to attend {} lectures to meet your target of {}%".format(
                        no_of_lectures, target)
                    bot.sendMessage(chat_id=spider.chatID, text=messageContent)

            remove('files/{}_attendance.png'.format(
                spider.username))  #Delete saved image
            mp.track(spider.username, 'Attendance')
        except IOError:
            bot.sendMessage(
                chat_id=spider.chatID,
                text=
                'The bot is experiencing some issues. Please try again later.')
            logger.warning(
                "Attendance screenshot failed! Check if site is blocking us or if Splash is up."
            )
            mp.track(spider.username, 'Error', {'type': 'Site down?'})
        except TelegramError as te:
            logger.warning("TelegramError: {}".format(str(te)))
            mp.track(spider.username, 'Error', {
                'type': 'TelegramError',
                'error': str(te)
            })
Beispiel #2
0
def attendance_target(bot, update):
    """Like :func:`until_eighty`, but with user specified target attendance percentage
    which is stored in the ``Misc`` table.
    If target isn't set, asks users whether they'd like to and passes control to 
    :py:func:`select_yn`
    
    :param bot: Telegram Bot object
    :type bot: telegram.bot.Bot
    :param update: Telegram Update object
    :type update: telegram.update.Update
    :return: SELECT_YN
    :rtype: int
    """

    bot.send_chat_action(chat_id=update.message.chat_id, action='typing')

    student_misc = Misc.query.filter(
        Misc.chatID == update.message.chat_id).first()

    if student_misc is None:
        new_misc_record = Misc(chatID=update.message.chat_id)
        db_session.add(new_misc_record)
        db_session.commit()
        username = get_user_info(update.message.chat_id)['PID']
        logger.info("Created new Misc record for {}".format(username))
        student_misc = Misc.query.filter(
            Misc.chatID == update.message.chat_id).first()

    target = student_misc.attendance_target

    if target is None:

        messageContent = textwrap.dedent("""
        You have not set a target yet. Would you like to set it now?
        You can change it anytime using /edit_target
        """)
        keyboard = [['Yes'], ['No']]
        reply_markup = ReplyKeyboardMarkup(keyboard)
        bot.sendMessage(chat_id=update.message.chat_id,
                        text=messageContent,
                        reply_markup=reply_markup)
        return SELECT_YN

    no_of_lectures = int(until_x(update.message.chat_id, target))
    if no_of_lectures < 0:
        messageContent = "Your attendance is already over {}%. Maybe set it higher? Use /edit_target to change it.".format(
            target)
        bot.sendMessage(chat_id=update.message.chat_id, text=messageContent)
        return ConversationHandler.END

    messageContent = "You need to attend {} lectures to meet your target of {}%".format(
        no_of_lectures, target)
    bot.sendMessage(chat_id=update.message.chat_id, text=messageContent)
    return ConversationHandler.END
Beispiel #3
0
def until_eighty(bot, update):
    """Calculate number of lectures you must consecutively attend before you attendance is 80%
    
    If :py:func:`misbot.mis_utils.until_x` returns a negative number, attendance is already over 80%

    :param bot: Telegram Bot object
    :type bot: telegram.bot.Bot
    :param update: Telegram Update object
    :type update: telegram.update.Update    
    """
    bot.send_chat_action(chat_id=update.message.chat_id, action='typing')
    no_of_lectures = int(until_x(update.message.chat_id, 80))
    if no_of_lectures < 0:
        bot.sendMessage(chat_id=update.message.chat_id,
                        text="Your attendance is already over 80%. Relax.")
    else:
        messageContent = "No. of lectures to attend: {}".format(no_of_lectures)
        bot.sendMessage(chat_id=update.message.chat_id, text=messageContent)
Beispiel #4
0
def until(bot, update, args):
    """Like :py:func:`until_eighty` but user supplies the number.

    :param bot: Telegram Bot object
    :type bot: telegram.bot.Bot
    :param update: Telegram Update object
    :type update: telegram.update.Update
    :param args: User supplied arguments
    :type args: tuple
    :return: None
    :rtype: None
    """
    if len(args) == 0:
        messageContent = textwrap.dedent("""
        You must specify a number after the command to use this feature.

        E.g: `/until 75`
        """)
        bot.sendMessage(chat_id=update.message.chat_id,
                        text=messageContent,
                        parse_mode='markdown')
        return

    try:
        figure = float(args[0])
    except (ValueError, IndexError):
        bot.sendMessage(chat_id=update.message.chat_id,
                        text="You must send a number between 1-99.")
        return

    if figure > 99:
        bot.sendMessage(chat_id=update.message.chat_id,
                        text="You must send a number between 1-99.")
        return

    no_of_lectures = int(until_x(update.message.chat_id, figure))
    if no_of_lectures < 0:
        bot.sendMessage(
            chat_id=update.message.chat_id,
            text="Your attendance is already over {}%. Relax.".format(figure))
    else:
        messageContent = "No. of lectures to attend: {}".format(no_of_lectures)
        bot.sendMessage(chat_id=update.message.chat_id, text=messageContent)
Beispiel #5
0
    def close_spider(self, spider):
        student_misc = Misc.query.filter(Misc.chatID == spider.chatID).first()
        try:
            bot.send_photo(
                chat_id=spider.chatID,
                photo=open("files/{}_attendance.png".format(spider.username),
                           'rb'),
                caption='Attendance Report for {}'.format(spider.username))
            if student_misc is not None and student_misc.attendance_target is not None:
                target = student_misc.attendance_target
                no_of_lectures = int(until_x(spider.chatID, target))
                if no_of_lectures > 0:
                    messageContent = "You need to attend {} lectures to meet your target of {}%".format(
                        no_of_lectures, target)
                    bot.sendMessage(chat_id=spider.chatID, text=messageContent)

            remove('files/{}_attendance.png'.format(
                spider.username))  #Delete saved image
        except IOError:
            bot.sendMessage(chat_id=spider.chatID,
                            text='There were some errors.')
            logger.warning(
                "Attendance screenshot failed! Check if site is blocking us or if Splash is up."
            )