Как преобразовать строку в функцию в Python?

Question

Например, если у меня есть функция с именем add, например

def add(x,y):
    return x+y

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

w=raw_input('Please input the function you want to use')

или

w='add'

Есть ли способ использовать w для ссылки на функцию add?

Answer 1

Поскольку вы принимаете пользовательский ввод, самый безопасный способ - точно определить, что является допустимым:

dispatcher={'add':add}
w='add'
try:
    function=dispatcher[w]
except KeyError:
    raise ValueError('invalid input')

Если вы хотите оценить такие строки, как 'add(3,4)', вы можете использовать safe eval :

eval('add(3,4)',{'__builtins__':None},dispatcher)

eval в целом может быть опасным при применении к пользовательскому вводу. Вышеуказанное безопаснее, поскольку __builtins__ отключен, а locals ограничен dispatcher. Кто-то умнее, чем я, мог бы все еще вызывать проблемы, но я не мог сказать вам, как это сделать.

ПРЕДУПРЕЖДЕНИЕ: Четный eval(..., {'__builtins__':None}, dispatcher) является небезопасным для применения при вводе пользователем. Злонамеренный пользователь может запустить на вашем компьютере произвольные функции , если ему будет предоставлена ​​возможность оценить его строку по eval.

Answer 2

Одним из безопасных способов является сопоставление имен и функций. Это безопаснее, чем использовать eval.

function_mappings = {
        'add': add,
}

def select_function():
    while True:
        try:
            return function_mappings[raw_input('Please input the function you want to use')]
        except KeyError:
            print 'Invalid function, try again.'

Answer 3

Решение unutbu - это то, что я обычно использую, но для полноты:

Если вы указываете точное имя функции, вы можете использовать eval, хотя это крайне нежелательно, потому что люди могут совершать злые действия:

eval("add")(x,y)

Answer 4

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

Если имеется ограниченное количество предопределенных функций, вам следует избегать eval и использовать вместо этого таблицу поиска (т.е. Dict).Никогда не доверяйте своим пользователям.

Answer 5

Если вы реализуете подобное оболочке приложение, в котором пользователь вводит какую-то команду (например, add ), а приложение отвечает (возвращает сумму), вы можете использовать модуль cmd, обрабатывает все командные взаимодействия и диспетчеризацию для вас. Вот пример:

#!/usr/bin/env python

import cmd
import shlex
import sys

class MyCmd(cmd.Cmd):
    def do_add(self, arguments):
        '''add - Adds two numbers the print the sum'''
        x, y = shlex.split(arguments)
        x, y = int(x), int(y)
        print x + y

    def do_quit(self, s):
        '''quit - quit the program'''
        sys.exit(0)

if __name__ == '__main__':
    cmd = MyCmd()
    cmd.cmdloop('type help for a list of valid commands')

Вот пример запущенной сессии:

$ python cmd_tryout.py
введите help для получения списка допустимых команд
(Cmd) помогите добавить
add - добавляет два числа к распечатке суммы
(Cmd) добавить 5 3
8
(Cmd) выход

По приглашению (Cmd) вы можете ввести команду help, которую вы получаете бесплатно. Другие команды add и quit соответствуют функциям do_add() и do_quit().

Обратите внимание, что команда help отображает строку документации для вашей функции. Строка документа представляет собой строку, следующую непосредственно за объявлением функции (см., Например, do_add()).

Модуль cmd не разбивает и не разбирает аргументы, поэтому вам придется делать это самостоятельно. Функция do_add() иллюстрирует это.

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

Answer 6

Просто используйте ссылку на функцию:

def pwr(x, y):
    return x ** y

def add(x, y):
    return x + y

dispatcher = { 'pwr' : pwr, 'add' : add}

def call_func(x, y, func):
    try:
        return dispatcher[func](x, y)
    except:
        return "Invalid function"

call_func(2, 3, 'add')

Простой и безопасный.

Answer 7

У меня было много ситуаций, когда мне нужно было сравнить строку с int и наоборот в шаблоне Django.

Я создал фильтр, который позволил мне передать имя функции и с помощью eval () преобразовать его.

Пример:

Шаблон:

{% ifequal string int|convert:'str' %} do something {% endifequal %}

Фильтр шаблона (где я использую строку для вызова имени функции):

@register.filter
def convert(value, funcname):
    try:
        converted = eval(funcname)(value)
        return converted
    except:
        return value

Answer 8

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

with open("function.py",'w') as file:
    f=input('enter the funtion you want to draw example: 2*x+1 or e**x :\n')
    file.write("from math import *\ndef f(x):\n\treturn "+f)

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

from function import f

теперь вы можете использовать свою функцию как обычную функцию python, если хотите, вы также можете удалить файл, в котором вы сохранили свою функцию, используя os.remove:

import os
os.remove("function.py")

Чтобы помочь вам понять, вот моя программа для рисования математических функций:

import numpy
import cv2
import os
from math import *


def generate(f,a,b,min,max,functionname='noname'):
    ph=(b-a)/1920
    pv=(max-min)/1080
    picture=numpy.zeros((1080,1920))
    for i in range(0,1920):
        picture[1079-(int((f(a+(i+1)*ph)*1080/max))),i]=255
    for i in range(1920):
        picture[1079-(int((f(a+(i+1)*ph)*1080/max)))+1,i]=255
    cv2.imwrite(functionname+'.png',picture)


with open("function.py",'w') as file:
    f=input('enter the funtion you want to draw example: or e**x :\n')
    file.write("from math import *\ndef f(x):\n\treturn "+f)

from function import f
os.remove("function.py")
d=input('enter the interval ,min ,max and the image file name. Separate characters with spacebar. Example: 0 1 0 14 exponontielle :\n').split(" ")
generate(f,int(d[0]),int(d[1]),int(d[2]),int(d[3]),d[4])

Answer 9

[Я попал сюда через дублирующий вопрос. Моей первой мыслью было использовать argparse и shlex, и я не увидел этого здесь, поэтому я добавляю его в качестве другого варианта.]

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

import argparse
import shlex

def hello(name):
    print('hello,', name)

def main():
    parser = argparse.ArgumentParser()
    subparsers = parser.add_subparsers()

    hello_parser = subparsers.add_parser('hello')
    hello_parser.add_argument('name')
    hello_parser.set_defaults(func=hello)

    print('Enter q to quit')

    while True:
        command = input('command> ')
        command = command.strip()

        if not command:
            continue

        if command.lower() == 'q':
            break

        words = shlex.split(command)

        try:
            args = parser.parse_args(words)
        except SystemExit:
            # argparse will sys.exit() on -h and errors; prevent that
            continue

        func_args = {name: value for name, value in vars(args).items()}
        del func_args['func']
        args.func(**func_args)

if __name__ == '__main__':
    try:
        main()
    except KeyboardInterrupt:
        print()