添加参考域

本教程的目的是说明角色、指令和域。完成后,我们将能够使用此扩展来描述食谱,并在文档的其他地方引用该食谱。

注意

本教程基于最初在 opensource.com 上发布的指南,并经原作者许可在此提供。

概述

我们希望扩展为 Sphinx 添加以下内容:

  • 一个 recipe 指令,包含一些描述食谱步骤的内容,以及一个 :contains: 选项,突出显示食谱的主要成分。

  • 一个 ref 角色,它提供对食谱本身的交叉引用。

  • 一个 recipe ,它允许我们将上述角色和域,以及索引等内容联系在一起。

为此,我们需要向 Sphinx 添加以下元素:

  • 一个名为 recipe 的新指令

  • 新的索引,以允许我们引用成分和食谱

  • 一个名为 recipe 的新域,它将包含 recipe 指令和 ref 角色

先决条件

我们需要与 之前的扩展 相同的设置。这次,我们将把扩展放在一个名为 recipe.py 的文件中。

以下是一个您可能获得的文件夹结构示例:

└── source
    ├── _ext
    │   └── recipe.py
    ├── conf.py
    └── index.rst

编写扩展

打开 recipe.py 并粘贴以下代码,我们将很快详细解释所有代码。

  1from collections import defaultdict
  2
  3from docutils.parsers.rst import directives
  4
  5from sphinx import addnodes
  6from sphinx.application import Sphinx
  7from sphinx.directives import ObjectDescription
  8from sphinx.domains import Domain, Index
  9from sphinx.roles import XRefRole
 10from sphinx.util.nodes import make_refnode
 11from sphinx.util.typing import ExtensionMetadata
 12
 13
 14class RecipeDirective(ObjectDescription):
 15    """A custom directive that describes a recipe."""
 16
 17    has_content = True
 18    required_arguments = 1
 19    option_spec = {
 20        'contains': directives.unchanged_required,
 21    }
 22
 23    def handle_signature(self, sig, signode):
 24        signode += addnodes.desc_name(text=sig)
 25        return sig
 26
 27    def add_target_and_index(self, name_cls, sig, signode):
 28        signode['ids'].append('recipe' + '-' + sig)
 29        if 'contains' in self.options:
 30            ingredients = [x.strip() for x in self.options.get('contains').split(',')]
 31
 32            recipes = self.env.get_domain('recipe')
 33            recipes.add_recipe(sig, ingredients)
 34
 35
 36class IngredientIndex(Index):
 37    """A custom index that creates an ingredient matrix."""
 38
 39    name = 'ingredient'
 40    localname = 'Ingredient Index'
 41    shortname = 'Ingredient'
 42
 43    def generate(self, docnames=None):
 44        content = defaultdict(list)
 45
 46        recipes = {
 47            name: (dispname, typ, docname, anchor)
 48            for name, dispname, typ, docname, anchor, _ in self.domain.get_objects()
 49        }
 50        recipe_ingredients = self.domain.data['recipe_ingredients']
 51        ingredient_recipes = defaultdict(list)
 52
 53        # flip from recipe_ingredients to ingredient_recipes
 54        for recipe_name, ingredients in recipe_ingredients.items():
 55            for ingredient in ingredients:
 56                ingredient_recipes[ingredient].append(recipe_name)
 57
 58        # convert the mapping of ingredient to recipes to produce the expected
 59        # output, shown below, using the ingredient name as a key to group
 60        #
 61        # name, subtype, docname, anchor, extra, qualifier, description
 62        for ingredient, recipe_names in ingredient_recipes.items():
 63            for recipe_name in recipe_names:
 64                dispname, typ, docname, anchor = recipes[recipe_name]
 65                content[ingredient].append((
 66                    dispname,
 67                    0,
 68                    docname,
 69                    anchor,
 70                    docname,
 71                    '',
 72                    typ,
 73                ))
 74
 75        # convert the dict to the sorted list of tuples expected
 76        content = sorted(content.items())
 77
 78        return content, True
 79
 80
 81class RecipeIndex(Index):
 82    """A custom index that creates an recipe matrix."""
 83
 84    name = 'recipe'
 85    localname = 'Recipe Index'
 86    shortname = 'Recipe'
 87
 88    def generate(self, docnames=None):
 89        content = defaultdict(list)
 90
 91        # sort the list of recipes in alphabetical order
 92        recipes = self.domain.get_objects()
 93        recipes = sorted(recipes, key=lambda recipe: recipe[0])
 94
 95        # generate the expected output, shown below, from the above using the
 96        # first letter of the recipe as a key to group thing
 97        #
 98        # name, subtype, docname, anchor, extra, qualifier, description
 99        for _name, dispname, typ, docname, anchor, _priority in recipes:
100            content[dispname[0].lower()].append((
101                dispname,
102                0,
103                docname,
104                anchor,
105                docname,
106                '',
107                typ,
108            ))
109
110        # convert the dict to the sorted list of tuples expected
111        content = sorted(content.items())
112
113        return content, True
114
115
116class RecipeDomain(Domain):
117    name = 'recipe'
118    label = 'Recipe Sample'
119    roles = {
120        'ref': XRefRole(),
121    }
122    directives = {
123        'recipe': RecipeDirective,
124    }
125    indices = {
126        RecipeIndex,
127        IngredientIndex,
128    }
129    initial_data = {
130        'recipes': [],  # object list
131        'recipe_ingredients': {},  # name -> object
132    }
133    data_version = 0
134
135    def get_full_qualified_name(self, node):
136        return f'recipe.{node.arguments[0]}'
137
138    def get_objects(self):
139        yield from self.data['recipes']
140
141    def resolve_xref(self, env, fromdocname, builder, typ, target, node, contnode):
142        match = [
143            (docname, anchor)
144            for name, sig, typ, docname, anchor, prio in self.get_objects()
145            if sig == target
146        ]
147
148        if len(match) > 0:
149            todocname = match[0][0]
150            targ = match[0][1]
151
152            return make_refnode(builder, fromdocname, todocname, targ, contnode, targ)
153        else:
154            print('Awww, found nothing')
155            return None
156
157    def add_recipe(self, signature, ingredients):
158        """Add a new recipe to the domain."""
159        name = f'recipe.{signature}'
160        anchor = f'recipe-{signature}'
161
162        self.data['recipe_ingredients'][name] = ingredients
163        # name, dispname, type, docname, anchor, priority
164        self.data['recipes'].append((
165            name,
166            signature,
167            'Recipe',
168            self.env.docname,
169            anchor,
170            0,
171        ))
172
173
174def setup(app: Sphinx) -> ExtensionMetadata:
175    app.add_domain(RecipeDomain)
176
177    return {
178        'version': '0.1',
179        'parallel_read_safe': True,
180        'parallel_write_safe': True,
181    }

让我们一步一步地查看扩展的每个部分,以解释正在发生的事情。

指令类

首先要检查的是 RecipeDirective 指令

 1class RecipeDirective(ObjectDescription):
 2    """A custom directive that describes a recipe."""
 3
 4    has_content = True
 5    required_arguments = 1
 6    option_spec = {
 7        'contains': directives.unchanged_required,
 8    }
 9
10    def handle_signature(self, sig, signode):
11        signode += addnodes.desc_name(text=sig)
12        return sig
13
14    def add_target_and_index(self, name_cls, sig, signode):
15        signode['ids'].append('recipe' + '-' + sig)
16        if 'contains' in self.options:
17            ingredients = [x.strip() for x in self.options.get('contains').split(',')]
18
19            recipes = self.env.get_domain('recipe')
20            recipes.add_recipe(sig, ingredients)

使用角色和指令扩展语法扩展构建过程 不同,此指令不继承自 docutils.parsers.rst.Directive 并且不定义 run 方法。相反,它继承自 sphinx.directives.ObjectDescription 并定义了 handle_signatureadd_target_and_index 方法。这是因为 ObjectDescription 是一个特殊用途的指令,旨在用于描述诸如类、函数或在本例中为食谱之类的东西。更具体地说,handle_signature 实现解析指令的签名,并将对象的名称和类型传递给它的超类,而 add_target_and_index 为此节点添加一个目标(以链接到)和一个索引条目。

我们还看到此指令定义了 has_contentrequired_argumentsoption_spec。与在 上一个教程 中添加的 TodoDirective 指令不同,此指令除了主体中的嵌套 reStructuredText 外,还接受一个参数(食谱名称)和一个选项(contains)。

索引类

 1class IngredientIndex(Index):
 2    """A custom index that creates an ingredient matrix."""
 3
 4    name = 'ingredient'
 5    localname = 'Ingredient Index'
 6    shortname = 'Ingredient'
 7
 8    def generate(self, docnames=None):
 9        content = defaultdict(list)
10
11        recipes = {
12            name: (dispname, typ, docname, anchor)
13            for name, dispname, typ, docname, anchor, _ in self.domain.get_objects()
14        }
15        recipe_ingredients = self.domain.data['recipe_ingredients']
16        ingredient_recipes = defaultdict(list)
17
18        # flip from recipe_ingredients to ingredient_recipes
19        for recipe_name, ingredients in recipe_ingredients.items():
20            for ingredient in ingredients:
21                ingredient_recipes[ingredient].append(recipe_name)
22
23        # convert the mapping of ingredient to recipes to produce the expected
24        # output, shown below, using the ingredient name as a key to group
25        #
26        # name, subtype, docname, anchor, extra, qualifier, description
27        for ingredient, recipe_names in ingredient_recipes.items():
28            for recipe_name in recipe_names:
29                dispname, typ, docname, anchor = recipes[recipe_name]
30                content[ingredient].append((
31                    dispname,
32                    0,
33                    docname,
34                    anchor,
35                    docname,
36                    '',
37                    typ,
38                ))
39
40        # convert the dict to the sorted list of tuples expected
41        content = sorted(content.items())
42
43        return content, True
 1class RecipeIndex(Index):
 2    """A custom index that creates an recipe matrix."""
 3
 4    name = 'recipe'
 5    localname = 'Recipe Index'
 6    shortname = 'Recipe'
 7
 8    def generate(self, docnames=None):
 9        content = defaultdict(list)
10
11        # sort the list of recipes in alphabetical order
12        recipes = self.domain.get_objects()
13        recipes = sorted(recipes, key=lambda recipe: recipe[0])
14
15        # generate the expected output, shown below, from the above using the
16        # first letter of the recipe as a key to group thing
17        #
18        # name, subtype, docname, anchor, extra, qualifier, description
19        for _name, dispname, typ, docname, anchor, _priority in recipes:
20            content[dispname[0].lower()].append((
21                dispname,
22                0,
23                docname,
24                anchor,
25                docname,
26                '',
27                typ,
28            ))
29
30        # convert the dict to the sorted list of tuples expected
31        content = sorted(content.items())
32
33        return content, True

IngredientIndexRecipeIndex 都继承自 Index。它们实现了自定义逻辑以生成一个定义索引的值元组。请注意,RecipeIndex 是一个简单的索引,只有一个条目。扩展它以涵盖更多对象类型尚不属于代码的一部分。

这两个索引都使用 Index.generate() 方法来完成工作。此方法结合了我们域的信息,对它进行排序,并以列表结构返回它,该结构将被 Sphinx 接受。这看起来可能很复杂,但实际上它只是一个元组列表,例如 ('tomato', 'TomatoSoup', 'test', 'rec-TomatoSoup',...)。有关此 API 的更多信息,请参考 域 API 指南

这些索引页面可以通过 ref 角色引用,方法是将域名称和索引 name 值组合在一起。例如,RecipeIndex 可以用 :ref:`recipe-recipe` 引用,而 IngredientIndex 可以用 :ref:`recipe-ingredient` 引用。

Sphinx 域是一个专门的容器,它将角色、指令和索引以及其他内容联系在一起。让我们看一下我们正在创建的这个 recipe 域和域的总体情况。

 1class RecipeDomain(Domain):
 2    name = 'recipe'
 3    label = 'Recipe Sample'
 4    roles = {
 5        'ref': XRefRole(),
 6    }
 7    directives = {
 8        'recipe': RecipeDirective,
 9    }
10    indices = {
11        RecipeIndex,
12        IngredientIndex,
13    }
14    initial_data = {
15        'recipes': [],  # object list
16        'recipe_ingredients': {},  # name -> object
17    }
18    data_version = 0
19
20    def get_full_qualified_name(self, node):
21        return f'recipe.{node.arguments[0]}'
22
23    def get_objects(self):
24        yield from self.data['recipes']
25
26    def resolve_xref(self, env, fromdocname, builder, typ, target, node, contnode):
27        match = [
28            (docname, anchor)
29            for name, sig, typ, docname, anchor, prio in self.get_objects()
30            if sig == target
31        ]
32
33        if len(match) > 0:
34            todocname = match[0][0]
35            targ = match[0][1]
36
37            return make_refnode(builder, fromdocname, todocname, targ, contnode, targ)
38        else:
39            print('Awww, found nothing')
40            return None
41
42    def add_recipe(self, signature, ingredients):
43        """Add a new recipe to the domain."""
44        name = f'recipe.{signature}'
45        anchor = f'recipe-{signature}'
46
47        self.data['recipe_ingredients'][name] = ingredients
48        # name, dispname, type, docname, anchor, priority
49        self.data['recipes'].append((
50            name,
51            signature,
52            'Recipe',
53            self.env.docname,
54            anchor,
55            0,
56        ))

关于此 recipe 域和域的总体情况,有一些有趣的事情需要注意。首先,我们实际上通过 directivesrolesindices 属性在指令本身注册我们的指令、角色和索引,而不是通过以后在 setup 中的调用。我们还可以注意到,我们实际上没有定义自定义角色,而是重复使用了 sphinx.roles.XRefRole 角色,并定义了 sphinx.domains.Domain.resolve_xref 方法。此方法接受两个参数,typtarget,它们分别指的是交叉引用类型及其目标名称。我们将使用 target 从我们域的 recipes 中解析我们的目标,因为我们目前只有一种节点类型。

继续前进,我们可以看到我们已经定义了 initial_data。在 initial_data 中定义的值将被复制到 env.domaindata[domain_name] 作为域的初始数据,并且域实例可以通过 self.data 访问它。我们看到我们在 initial_data 中定义了两个项目:recipesrecipe_ingredients。每个都包含一个所有定义的对象的列表(即所有食谱)和一个哈希,它将规范成分名称映射到对象列表。我们命名对象的方式在我们扩展中很常见,并在 get_full_qualified_name 方法中定义。对于创建的每个对象,规范名称为 recipe.<recipename>,其中 <recipename> 是文档编写者为对象(食谱)提供的名称。这使扩展能够使用具有相同名称的不同对象类型。拥有规范名称和对象的中心位置是一个巨大的优势。我们的索引和交叉引用代码都使用此功能。

setup 函数

像往常一样setup 函数是一个要求,用于将我们扩展的各个部分挂钩到 Sphinx。让我们看一下此扩展的 setup 函数。

1def setup(app: Sphinx) -> ExtensionMetadata:
2    app.add_domain(RecipeDomain)
3
4    return {
5        'version': '0.1',
6        'parallel_read_safe': True,
7        'parallel_write_safe': True,
8    }

这看起来与我们习惯看到的有点不同。没有对 add_directive() 甚至 add_role() 的调用。相反,我们有一个对 add_domain() 的调用,然后是对 标准域 的一些初始化。这是因为我们已经在指令本身注册了我们的指令、角色和索引。

使用扩展

您现在可以在整个项目中使用扩展。例如

index.rst
Joe's Recipes
=============

Below are a collection of my favourite recipes. I highly recommend the
:recipe:ref:`TomatoSoup` recipe in particular!

.. toctree::

   tomato-soup
tomato-soup.rst
The recipe contains `tomato` and `cilantro`.

.. recipe:recipe:: TomatoSoup
   :contains: tomato, cilantro, salt, pepper

   This recipe is a tasty tomato soup, combine all ingredients
   and cook.

需要注意的是,使用了 :recipe:ref: 角色来交叉引用实际定义在其他地方(使用 :recipe:recipe: 指令)的食谱。

进一步阅读

有关更多信息,请参阅 docutils 文档和 Sphinx API

如果您希望在多个项目中或与他人共享您的扩展,请查看 第三方扩展 部分。