Есть ли способ проверить, назначен ли вывод функции переменной в Python?

Новое решение

Это новое решение, которое определяет, когда результат функции используется для присвоения, путем проверки его собственного байт-кода. Запись байт-кода не производится, и он даже должен быть совместим с будущими версиями Python, потому что он использует модуль кода операции для определений.

import inspect, dis, opcode

def check_status():

    try:
        frame = inspect.currentframe().f_back
        next_opcode = opcode.opname[ord(frame.f_code.co_code[frame.f_lasti+3])]
        if next_opcode == "POP_TOP": 
            # or next_opcode == "RETURN_VALUE":
            # include the above line in the if statement if you consider "return check_status()" to be assignment
            print "I was not assigned"
            print "Pretty printer status check 0.02v"
            print "NOTE: This is so totally not written for giant robots"
            return
    finally:
        del frame    

    # do normal routine

    info = {'cond_op': 1, 't_canoncharge': 1342, 'stage_booster': 5}

    return info

# no assignment    
def test1():
    check_status()

# assignment
def test2():
    a = check_status()

# could be assignment (check above for options)
def test3():
    return check_status()

# assignment
def test4():
    a = []
    a.append(check_status())
    return a

Решение 1

Это старое решение, которое обнаруживает всякий раз, когда вы вызываете функцию при отладке с использованием python -i или PDB.

import inspect

def check_status():
    frame = inspect.currentframe()
    try:
        if frame.f_back.f_code.co_name == "<module>" and frame.f_back.f_code.co_filename == "<stdin>":
            print "Pretty printer status check 0.02v"
            print "NOTE: This is so totally not written for giant robots"
    finally:
        del frame

    # do regular stuff   
    return {'cond_op': 1, 't_canoncharge': 1342, 'stage_booster': 5}

def test():
    check_status()


>>> check_status()
Pretty printer status check 0.02v
NOTE: This is so totally not written for giant robots
{'cond_op': 1, 't_canoncharge': 1342, 'stage_booster': 5}

>>> a=check_status()
Pretty printer status check 0.02v
NOTE: This is so totally not written for giant robots

>>> a
{'cond_op': 1, 't_canoncharge': 1342, 'stage_booster': 5}

test()
>>>

Однако я также хотел бы передать вывод в виде списка

Вы имеете в виду "вернуть результат в виде словаря" - будьте осторожны ;-)

Одна вещь, которую вы могли бы сделать, - это использовать способность интерпретатора Python автоматически преобразовывать в строку результат любого выражения. Для этого создайте собственный подкласс dict, который, когда его попросят преобразовать себя в строку, выполняет любое красивое форматирование, которое вы хотите. Например,

class PrettyDict(dict):
    def __str__(self):
        return '''Pretty printer status check 0.02v
NOTE: This is so totally not written for giant robots
=================================
System operational: ... %s
Time to ion canon charge is %dm %ds
Booster rocket in %s state
 (other stuff)
''' % (self.conf_op and 'ok' or 'OMGPANIC!!!',
       self.t_canoncharge / 60, self.t_canoncharge % 60, 
       BOOSTER_STATES[self.booster_charge],
       ... )

Конечно, вы могли бы придумать более красивый способ написания кода, но основная идея заключается в том, что метод __str__ создает красиво напечатанную строку, которая представляет состояние объекта и возвращает его. Затем, если вы вернете PrettyDict из своей check_status() функции, при вводе

>>> check_status()

вы бы увидели

Pretty printer status check 0.02v
NOTE: This is so totally not written for giant robots
=================================
System operational: ... ok
Time to ion canon charge is 9m 21s
Booster rocket in AFTERBURNER state
Range check is optimal
Rocket fuel is 10h 19m 40s to depletion
Beer served is type WICKSE LAGER, chill optimal
Suggested catchphrase is 01_FIGHTING_SPIRIT_GOGOGO
Virtual ... on

Единственная загвоздка в том, что

>>> not_robot_stat = check_status()
>>> print not_robot_stat

даст вам то же самое, потому что такое же преобразование в строку происходит как часть функции print. Но для использования этого в каком-то реальном приложении я сомневаюсь, что это будет иметь значение. Если вы действительно хотите видеть возвращаемое значение как чистый dict, вы можете сделать

>>> print repr(not_robot_stat)

вместо этого, и он должен показать вам

{'cond_op': 1, 't_canoncharge': 1342, 'stage_booster': 5, 'range_est_sigma': 0.023, 'fuel_est': 32557154, 'beer_type': 31007, 'beer_temp': 2, 'catchphrase_suggestion': 1023, 'virtual_on': 'hell yes'}

Дело в том, что, как говорили другие плакаты, функция в Python не может знать, что будет сделано с ее возвращаемым значением (EDIT: хорошо, может быть, будет какой-то странный байт-код -hacking way, но не делайте этого) - но вы можете обойти это в тех случаях, когда это важно.

Для этого нет варианта использования. Python присваивает все интерактивные результаты специальной переменной с именем _.

Вы можете сделать следующее в интерактивном режиме. Работает отлично. Никаких забавных дел.

>>> check_status()
{'cond_op': 1, 't_canoncharge': 1342, 'stage_booster': 5, 'range_est_sigma': 0.023, 'fuel_est': 32557154, 'beer_type': 31007, 'beer_temp': 2, 'catchphrase_suggestion': 1023, 'virtual_on': 'hell yes'}

>>> pprint.pprint( _ )
{'beer_temp': 2,
 'beer_type': 31007,
 'catchphrase_suggestion': 1023,
 'cond_op': 1,
 'fuel_est': 32557154,
 'range_est_sigma': 0.023,
 'stage_booster': 5,
 't_canoncharge': 1342,
 'virtual_on': 'hell yes'}

Функция не может узнать, как используется ее возвращаемое значение.

Что ж, как вы упомянули, вопиющие взломы байт-кода могли бы это сделать, но это было бы действительно сложно и, вероятно, хрупко. Пойдите более простым явным путем.

Даже если есть способ сделать это, это плохая идея. Представьте, что вы пытаетесь отладить что-то, что ведет себя по-разному в зависимости от контекста, из которого оно вызывается. А теперь представьте, что через полгода все это похоронено в части какой-то системы, которая слишком велика, чтобы держать ее в голове сразу.

Будь проще. Явное лучше, чем неявное. Просто создайте функцию красивой печати (или используйте модуль pprint) и вызовите ее по результату. В интерактивном сеансе Pythn вы можете использовать _, чтобы получить значение последнего выражения.

Невозможно сделать это, по крайней мере, с помощью обычных синтаксических процедур, потому что вызов функции и присвоение являются полностью независимыми операциями, которые не знают друг друга.

Самый простой обходной путь, который я могу найти для этого, - передать флаг в качестве аргумента вашей функции check_status и обработать его соответствующим образом.

def check_status(return_dict=False) :
    if return_dict :
        # Return stuff here.
    # Pretty Print stuff here.

А потом...

check_status() # Pretty print
not_robot_stat = check_status(True) # Get the dict.

РЕДАКТИРОВАТЬ: Я предполагал, что вы будете печатать чаще, чем назначили. Если это не так, поменяйте местами значение по умолчанию и значение, которое вы передаете.

Единственный способ сделать то, что вы хотите, - это действительно «поиграть с байт-кодом» - другого способа восстановить эту информацию нет. Гораздо лучше, конечно, предоставить две отдельные функции: одну для получения статуса как dict, а другую для вызова первой и ее форматирования.

В качестве альтернативы (но не в отличной архитектуре) у вас может быть одна функция, которая принимает необязательный параметр, чтобы вести себя этими двумя совершенно разными способами - это не отлично, потому что функция должна иметь ОДНУ функцию, т.е. в основном делать ОДНУ вещь, а не два разных такие как эти.