Python ConversationHandler.END属性代码示例

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def polo(self, bot, update, user_data):
        msg = update.message.text.upper()
        conn = sqlite3.connect(self.mount_point + 'coders1.db')
        c = conn.cursor()
        c.execute("SELECT name FROM handles WHERE name=(?)", (msg,))
        if c.fetchone():
            keyboard = [[InlineKeyboardButton("Hackerearth", callback_data='HElist8'),
                         InlineKeyboardButton("Hackerrank", callback_data='HRlist8')],
                        [InlineKeyboardButton("Codechef", callback_data='CClist8'),
                         InlineKeyboardButton("Spoj", callback_data='SPlist8')],
                        [InlineKeyboardButton("Codeforces", callback_data='CFlist8'),
                         InlineKeyboardButton("ALL", callback_data='ALLlist8')]]
            reply_markup = InlineKeyboardMarkup(keyboard)
            update.message.reply_text('please select the judge or select all for showing all',
                                      reply_markup=reply_markup)
            user_data['name1'] = msg
            conn.close()
            return XOLO
        else:
            conn.close()
            update.message.reply_text("Sorry this name is not registered with me.")
            return ConversationHandler.END

    # FUNCTION TO SHOW THE KIND OF RANKLIST USER WANTS 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def code(bot, update, user_data):
        query = update.callback_query
        val = query.data
        val = str(val).replace("so1", "")
        if val == "code":
            bot.edit_message_text(
                text="selected",
                chat_id=query.message.chat_id, message_id=query.message.message_id)
            bot.send_message(chat_id=query.message.chat_id, text="please enter your code\n"
                                                                  "Please make sure that "
                                                                  "the first line is not a comment line",
                             reply_markup=ForceReply(True))
            return DECODE
        elif val == "file":
            bot.edit_message_text(text="please send your .txt file\nMaximum size 2mb", chat_id=query.message.chat_id,
                                  message_id=query.message.message_id)
            return FILE
        else:
            return ConversationHandler.END

    # FUNCTION TO GET TESTCASE FILE 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def filetest(self, bot, update, user_data):
        file_id = update.message.document.file_id
        if ComHandler.check_file_size(update):
            return ConversationHandler.END
        newFile = bot.get_file(file_id)
        newFile.download('test.txt')
        with open('test.txt', 'rt') as f:
            source = f.read()
        s1 = (str(user_data['code'])).replace("«", "<<").replace("»", ">>")
        result = self.compiler.run({'source': s1,
                               'lang': user_data['lang'],
                               'testcases': [source]
                               })
        Utility.paginate(bot, update, result)
        user_data.clear()
        os.remove('test.txt')
        return ConversationHandler.END 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def filer(bot, update, user_data):
        file_id = update.message.document.file_id
        if ComHandler.check_file_size(update):
            return ConversationHandler.END
        newFile = bot.get_file(file_id)
        newFile.download('abcd.txt')
        with open('abcd.txt', 'r') as f:
            source = f.read()
        user_data['code'] = source
        custom_keyboard = [['#no test case', '#send a .txt file']]
        reply_markup = ReplyKeyboardMarkup(custom_keyboard, one_time_keyboard=True, resize_keybord=True)
        update.message.reply_text(
            'Please send test cases together as you would do in online ide\nIf you dont want to provide test cases select #no test case\n I you want to send test cases as .txt file select #send a .txt file',
            reply_markup=reply_markup)
        # REMOVING THE FILE AFTER PROCESS IS COMPLETE
        os.remove('abcd.txt')
        return TESTCASES

    # FUNCTION TO GET THE SOURCE CODE SENT BY USER 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def qselcf(self, bot, update):
        query = update.callback_query
        val = query.data
        if val == 'Acf1':
            strn = self.get_random_ques('A')
        elif val == 'Bcf1':
            strn = self.get_random_ques('B')
        elif val == 'Ccf1':
            strn = self.get_random_ques('C')
        elif val == 'Dcf1':
            strn = self.get_random_ques('D')
        elif val == 'Ecf1':
            strn = self.get_random_ques('E')
        elif val == 'Fcf1':
            strn = self.get_random_ques('F')
        elif val == 'OTHERScf1':
            strn = self.get_random_ques('OTHERS')
        val = str(val).replace("cf1", "")
        bot.edit_message_text(
            text="Random " + val + " question from codeforces\n\n" + strn,
            chat_id=query.message.chat_id,
            message_id=query.message.message_id)
        return ConversationHandler.END
        # END OF CONVERSATION HANDLER FOR GETTING RANDOM QUESTION FROM CODEFORCES 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def broadcast_message(self, bot, update):
        message = update.message.text
        conn = sqlite3.connect(self.mount_point + 'coders1.db')
        c = conn.cursor()
        c.execute('select id from handles union select id from subscribers')
        for row in c.fetchall():
            try:
                bot.send_message(text=message, chat_id=row[0])
            except:
                pass
            time.sleep(1)
        c.close()
        conn.close()
        return ConversationHandler.END

    # END OF ADMIN CONVERSATION HANDLER TO BROADCAST MESSAGE

    # START OF ADMIN CONVERSATION HANDLER TO REPLACE THE DATABASE 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def register(self, bot,update):
		try:
			with closing(self.conn.cursor()) as cur:
				uid = update.message.from_user.id
				cur.execute("""SELECT telegram_id FROM User WHERE telegram_id = %s""",(uid,))
				if cur.rowcount == 0:
					message = "Registering you in my database! \n"
					message += "Can I please have the generation index?\n"
					message += "This bot will search for the generation index less than or equals to the number you input\n"
					message += "If you feel threatened at any point of time, do a /cancel to abort this conversation."
					cur.execute("""INSERT INTO User VALUES(%s,NULL,NULL,NULL,NULL,'No')""",(uid,))
					update.message.reply_text(message,parse_mode='HTML')
					return GENERATION
				else:
					message = "You are already registered in my database. To remove your details, do a /forget"
					update.message.reply_text(message,parse_mode='HTML')
					return ConversationHandler.END


		except Exception as e:
			catcherror = traceback.format_exc()
			bot.sendMessage(chat_id=Tokens().error_channel(),text=catcherror,parse_mode='HTML') 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def attribute_list(self,bot,update):
		try:
			with closing(self.conn.cursor()) as cur:
				uid = update.message.from_user.id
				if str.lower(update.message.text) == "end":
					message = "Thanks for registering =)\n"
					message = "If you want to toggle 10 minute scans, please do a /alert"
					update.message.reply_text(message,parse_mode='HTML')
					return ConversationHandler.END
				else:
					cur.execute("""INSERT INTO Attribute VALUES(%s,%s)""",(uid,update.message.text,))
					message= update.message.text
					message += " has been added as an attribute. Please enter the next attribute \n"
					message += "If you're done with adding your cattributes, please reply with end"
					update.message.reply_text(message,parse_mode='HTML')
					return ATTLIST

		except Exception as e:
			catcherror = traceback.format_exc()
			bot.sendMessage(chat_id=Tokens().error_channel(),text=catcherror,parse_mode='HTML') 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def send_broadcast(bot, update, user_data):
    uid = update.effective_user.id

    try:
        bc = user_data['broadcast']
        text = bc['text']
        recipient = bc['target_chat_id']
        mode = bc.get('mode', 'just_send')
    except AttributeError:
        bot.formatter.send_failure(uid, "Missing attributes for broadcast. Aborting...")
        return ConversationHandler.END

    mid = bc.get('reply_to_message_id')

    if mode == 'replying':
        msg = util.send_md_message(bot, recipient, text, reply_to_message_id=mid)
    elif mode == 'editing':
        msg = bot.formatter.send_or_edit(recipient, text, to_edit=mid)
    else:
        msg = util.send_md_message(bot, recipient, text)

    # Post actions
    buttons = [
        InlineKeyboardButton(captions.PIN,
                             callback_data=util.callback_for_action('pin_message', {'mid': msg.message_id})),
        InlineKeyboardButton('Add "Thank You" counter',
                             callback_data=util.callback_for_action('add_thank_you',
                                                                    {'cid': recipient, 'mid': msg.message_id})),
    ]
    reply_markup = InlineKeyboardMarkup(util.build_menu(buttons, 1))
    mid = util.mid_from_update(update)
    action_taken = "edited" if mode == 'editing' else "broadcasted"
    bot.formatter.send_or_edit(uid, mdformat.success("Message {}.".format(action_taken)), mid, reply_markup=reply_markup) 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def cancel(bot, update, user_data):
        update.message.reply_text('Cancelled')
        user_data.clear()
        return ConversationHandler.END

    # FUNCTION FOR LOGGING ALL KINDS OF ERRORS 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def getCf(self, bot, update):
        if not str(update.message.chat_id) in self.admin_list:
            update.message.reply_text("sorry you are not an admin")
            return ConversationHandler.END
        update.message.reply_text("send your json file")
        return self.CF 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def receive_cf(self, bot, update):
        file_id = update.message.document.file_id
        newFile = bot.get_file(file_id)
        newFile.download(self.mount_point + 'codeforces.json')
        update.message.reply_text("saved")
        with open(self.mount_point + 'codeforces.json', 'r') as code_json:
            self.cf.change_cf(json.load(code_json))
        return ConversationHandler.END

    # END OF ADMIN CONVERSATION HANDLER TO REPLACE THE CODEFORCES JSON 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def testcases(self, bot, update, user_data):
        markup = ReplyKeyboardRemove()
        s = update.message.text
        if s == "#send a .txt file":
            update.message.reply_text("Please send your testcases as a .txt file\nMaximum size 2mb",
                                      reply_markup=markup)
            return FILETEST
            # CONVERTING UNICODE CHARACTER TO DOUBLE GREATER THAN OR LESS THAN
            # WEIRD
        s1 = (str(user_data['code'])).replace("«", "<<").replace("»", ">>")
        if s == "#no test case":
            # USING COMPILER FUNCTION FROM helper.py script
            result = self.compiler.run({'source': s1,
                                   'lang': user_data['lang']
                                   })
            # GETTING OUTPUT FROM result CLASS in helper.py script
            Utility.paginate(bot, update, result)
        else:
            # AGAIN THE SAME DRILL
            result = self.compiler.run({'source': s1,
                                   'lang': user_data['lang'],
                                   'testcases': [s]
                                   })
            Utility.paginate(bot, update, result)
        user_data.clear()
        return ConversationHandler.END

    # FUNCTION FOR THE CASE WHERE USER HAD SELECTED OTHER 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def solo(self, bot, update):
        query = update.callback_query
        val = query.data
        choices = ['HElist7', 'HRlist7', 'CClist7', 'SPlist7', 'CFlist7', 'ALLlist7']
        if val not in choices:
            return ConversationHandler.END
        val = str(val).replace("list7", "")
        conn = sqlite3.connect(self.mount_point + 'coders1.db')
        c = conn.cursor()
        if val == "ALL":
            a = str(query.from_user.id)
            bot.edit_message_text(text='Sending please wait',
                                  chat_id=query.message.chat_id,
                                  message_id=query.message.message_id)
            mysel = c.execute("SELECT name, HE, HR, SP, CC, CF FROM datas WHERE id=" + a)
            self.utility.xlsx_creator(mysel, 'me.xlsx')
            bot.send_document(chat_id=query.message.chat_id, document=open('me.xlsx', 'rb'))
            os.remove('me.xlsx')
        else:
            a = str(query.from_user.id)
            c.execute("SELECT " + val + " FROM datas WHERE id=" + a)
            for i in c.fetchall():
                if i[0] is None or i[0] == "":
                    bot.edit_message_text(text="NOT REGISTERED",
                                          chat_id=query.message.chat_id,
                                          message_id=query.message.message_id)
                else:
                    bot.edit_message_text(text="" + i[0],
                                          chat_id=query.message.chat_id,
                                          message_id=query.message.message_id)
        conn.commit()
        conn.close()
        return ConversationHandler.END

    # FUNCTION TO GET THE RANKLIST MENU OF THE USER BY SEARCHING HIS NAME 

# 需要导入模块: from telegram.ext import ConversationHandler [as 别名]
# 或者: from telegram.ext.ConversationHandler import END [as 别名]
def xolo(self, bot, update, user_data):
        query = update.callback_query
        val = query.data
        choices = ['HElist8', 'HRlist8', 'CClist8', 'SPlist8', 'CFlist8', 'ALLlist8']
        if val not in choices:
            return ConversationHandler.END
        val = str(val).replace("list8", "")
        name1 = user_data['name1']
        conn = sqlite3.connect(self.mount_point + 'coders1.db')
        c = conn.cursor()
        if val == "ALL":
            bot.edit_message_text(text='Sending please wait',
                                  chat_id=query.message.chat_id,
                                  message_id=query.message.message_id)
            mysel = c.execute("SELECT name, HE, HR, SP, CC, CF FROM datas WHERE name=(?)", (name1,))
            self.utility.xlsx_creator(mysel, 'det.xlsx')
            bot.send_document(chat_id=query.message.chat_id, document=open('det.xlsx', 'rb'))
            os.remove('det.xlsx')
        else:
            c.execute("SELECT " + val + " FROM datas WHERE name=(?)", (name1,))
            for i in c.fetchall():
                if i[0] is None or i[0] == "":
                    bot.edit_message_text(text="Not Registered",
                                          chat_id=query.message.chat_id,
                                          message_id=query.message.message_id)
                else:
                    bot.edit_message_text(text="" + i[0],
                                          chat_id=query.message.chat_id,
                                          message_id=query.message.message_id)
        user_data.clear()
        conn.close()
        return ConversationHandler.END

    # FUNCTION TO SHOW THE RANKLIST OF ALL THE PEOPLE