如何根据参数调用特定的函数来装饰器

时间:2013-07-01 22:00:41

标签: python design-patterns decorator

免责声明:我显然对装饰者来说很陌生

在python中创建和装饰函数是非常简单和直接的,而这个excellent answer(具有最多upvotes的那个)给出了非常好的介绍并展示了如何嵌套装饰器。这一切都很好,花花公子。但是我还要弄清楚的是有多少(python)-web-frameworks(flask,django等)设法调用,我只能猜测是基于的装饰函数 参数传递给装饰器

一个例子(使用Flask,但在许多框架中类似)向您展示我的意思。

@application.route('/page/a_page')
def show_a_page():
    return html_content

@application.route('/page/another_page')
def show_another_page():
    return html_content

现在,如果我向mysite.com/page/a_page瓶子发出请求,我们会以某种方式确定它应该调用show_a_page,如果请求是show_another_page,那么mysite.com/page/a_page也是如此。dir(module_name)

我想知道如何在我自己的项目中实现类似的功能?

我认为存在类似于使用{{1}}提取有关每个函数的修饰(?)的信息的内容?

3 个答案:

答案 0 :(得分:0)

除了包装函数

之外,装饰者没有理由不能做其他的事情
>>> def print_stuff(stuff):
...     def decorator(f):
...         print stuff
...         return f
...     return decorator
...
>>> @print_stuff('Hello, world!')
... def my_func():
...     pass
...
Hello, world!

在这个例子中,我们只是在定义函数时打印出传递给装饰器构造函数的参数。请注意,我们打印了“Hello,world!”没有实际调用my_func - 那是因为打印发生在构造装饰器时,而不是装饰器本身。

发生了什么,application.route本身不是装饰者。相反,它是一个获取路径的函数,并生成将应用于视图函数的装饰器,以及在该路径上注册视图函数。在flask中,装饰器构造函数可以访问路径和视图函数,因此它可以在应用程序对象上的该路径上注册视图函数。如果您有兴趣,可以查看Github上的Flask源代码:

https://github.com/mitsuhiko/flask/blob/master/flask/app.py

答案 1 :(得分:0)

装饰器对功能的作用取决于它。它可以将它包装在另一个函数中,向其属性添加一些元信息,甚至将它存储在字典中。

这是一个如何将多个函数保存到字典中的示例,以后可以用它来查找这些函数:

function_table = {}

def add_to_table(name):
    def dec(func):
        def inner_func(*args, **kwargs):
            return func(*args, **kwargs)
        function_table[name] = inner_func
        return inner_func
    return dec

@add_to_table("my_addition_function")
def myfunction(a, b):
    return a + b

@add_to_table("my_subtraction_function")
def myfunction2(a, b):
    return a - b

print myfunction(1, 3)
# 4
print function_table["my_addition_function"](1, 4)
# 5
print function_table["my_subtraction_function"](1, 4)
# -3

这是Flask正在做的非常简单的版本:根据正在使用的路径存储要调用的函数的表。

答案 2 :(得分:0)

这是一个使用BeautifulSoup的粗略示例,它建立在David Robinson的回答之上。装饰器使用传递给它的字符串作为对应于函数的字典的键。这将通过关键字参数传递给调用它的装饰函数。

import os
import sys

# Import System libraries
from copy import deepcopy

# Import Custom libraries
from BeautifulSoup import BeautifulSoup, Tag

page_base_str = \
'''
<!doctype html>
<html>
<head>
    <title>Example Domain</title>

    <meta charset="utf-8" />
    <meta http-equiv="Content-type" content="text/html; charset=utf-8" />
    <meta name="viewport" content="width=device-width, initial-scale=1" />
    <style type="text/css">
    body {
        background-color: #f0f0f2;
        margin: 0;
        padding: 0;
        font-family: "Open Sans", "Helvetica Neue", Helvetica, Arial, sans-serif;

    }
    div {
        width: 600px;
        margin: 5em auto;
        padding: 50px;
        background-color: #fff;
        border-radius: 1em;
    }
    a:link, a:visited {
        color: #38488f;
        text-decoration: none;
    }
    @media (max-width: 700px) {
        body {
            background-color: #fff;
        }
        div {
            width: auto;
            margin: 0 auto;
            border-radius: 0;
            padding: 1em;
        }
    }
    </style>    
</head>

<body>
<div>
    <h1>Example Domain</h1>
    <div id="body">
        <p>This domain is established to be used for illustrative examples in documents. You may use this
        domain in examples without prior coordination or asking for permission.</p>
        <p><a href="http://www.iana.org/domains/example">More information...</a></p>
    </div>
</div>
</body>
</html>
'''
page_base_tag = BeautifulSoup(page_base_str)

def default_gen(*args):
    return page_base_tag.prettify()

def test_gen(**kwargs):
    copy_tag = deepcopy(page_base_tag)

    title_change_locations = \
    [
        lambda x: x.name == u"title",
        lambda x: x.name == u"h1"
    ]
    title = kwargs.get("title", "")
    if(title):
        for location in title_change_locations:
            search_list = copy_tag.findAll(location)
            if(not search_list):
                continue
            tag_handle = search_list[0]
            tag_handle.clear()
            tag_handle.insert(0, title)

    body_change_locations = \
    [
        lambda x: x.name == "div" and set([(u"id", u"body")]) <= set(x.attrs)
    ]
    body = kwargs.get("body", "")
    if(body):
        for location in body_change_locations:
            search_list = copy_tag.findAll(location)
            if(not search_list):
                continue
            tag_handle = search_list[0]
            tag_handle.clear()
            tag_handle.insert(0, body)

    return copy_tag.prettify()

page_gens = \
{
    "TEST" : test_gen
}

def page_gen(name = ""):
    def dec(func):
        def inner_func(**kwargs):
            kwargs["PAGE_FUNC"] = page_gens.get(name, default_gen)
            return func(**kwargs)
        return inner_func
    return dec

@page_gen("TEST")
def test_page_01(**kwargs):
    content = kwargs["PAGE_FUNC"](title = "Page 01", body = "Page 01 body")
    return content

@page_gen("TEST")
def test_page_02(**kwargs):
    content = kwargs["PAGE_FUNC"](title = "Page 02", body = "Page 02 body")
    return content

@page_gen()
def a_page(**kwargs):
    content = kwargs["PAGE_FUNC"]()
    return content

print test_page_01()
print test_page_02()
print a_page()