How to pass a variable to a different function (a different stage in ConversationHandler?)

user_19031099 0
Viewed 63

So I have a bot which moves from stage to stage using the stage names as a product of return of each function. e.g.

STEP_1, STEP_2, STEP_3 = range(3)

def askforfile(update, context):
    chat_id = update.message.chat_id
    context.bot.send_message(chat_id=chat_id, text="give a file!")
    return STEP_1

def getfile(update, context):
    chat_id = update.message.chat_id
    filename = 'newfile'+str(chat_id)+ '.txt'
    f = open(filename, 'wb')
    context.bot.get_file(update.message.document.file_id).download(out=f)
    f.close()
    return STEP_2

def get_info(update, context):
    chat_id = update.message.chat_id
    info = update.message.text
    return STEP_3

def end(update, context):
    with open(filename, 'rb') as f:
       writecsv1 = some_module.some_function(filename, info)
    with open(writecsv1, 'rb') as doc:
        context.bot.send_document(chat_id=chat_id, document=doc)
    return ConversationHandler.END

So in the end() function I need to pass the variables from functions getfile() and get_info() in the some_function. But I have no clue how to do it, because returning multiple values doesn't work for me and even if it worked, obviously I can't call functions.

python-telegram-bot python
2 Answers

You could use global variables filename and info (maybe don't name them that) that you rewrite within your functions. Just define them with some default value above your function definitions. Right now they're just defined locally and disappear once the function returns.

user_106308390

I highly recommend not to use global variables. Global variables are considered bad practive in general and in this case they can easily get you into trouble when multiple users are using your bot at the same time and each overrides the global variables. python-telegram-bot comes with a built-in solution for storing data in memory. I recommend to leverage that.

Disclaimer: I'm currently the maintainer of python-telegram-bot.

user_106069620
Related