Enable automatic formatting for `sphinx/domains/math.py`

This commit is contained in:
Adam Turner 2025-01-01 11:54:41 +00:00
parent 4a81818bb3
commit f42f3676d5
2 changed files with 44 additions and 21 deletions

View File

@ -394,7 +394,6 @@ preview = true
quote-style = "single"
exclude = [
"sphinx/builders/latex/constants.py",
"sphinx/domains/math.py",
"sphinx/domains/python/_annotations.py",
"sphinx/domains/python/__init__.py",
"sphinx/domains/python/_object.py",

View File

@ -27,8 +27,13 @@ logger = logging.getLogger(__name__)
class MathReferenceRole(XRefRole):
def result_nodes(self, document: nodes.document, env: BuildEnvironment, node: Element,
is_ref: bool) -> tuple[list[Node], list[system_message]]:
def result_nodes(
self,
document: nodes.document,
env: BuildEnvironment,
node: Element,
is_ref: bool,
) -> tuple[list[Node], list[system_message]]:
node['refdomain'] = 'math'
return [node], []
@ -60,8 +65,12 @@ class MathDomain(Domain):
def note_equation(self, docname: str, labelid: str, location: Any = None) -> None:
if labelid in self.equations:
other = self.equations[labelid][0]
logger.warning(__('duplicate label of equation %s, other instance in %s'),
labelid, other, location=location)
logger.warning(
__('duplicate label of equation %s, other instance in %s'),
labelid,
other,
location=location,
)
self.equations[labelid] = (docname, self.env.new_serialno('eqno') + 1)
@ -71,8 +80,9 @@ class MathDomain(Domain):
else:
return None
def process_doc(self, env: BuildEnvironment, docname: str,
document: nodes.document) -> None:
def process_doc(
self, env: BuildEnvironment, docname: str, document: nodes.document
) -> None:
def math_node(node: Node) -> bool:
return isinstance(node, nodes.math | nodes.math_block)
@ -93,9 +103,16 @@ class MathDomain(Domain):
for docname in docnames:
self.data['has_equations'][docname] = otherdata['has_equations'][docname]
def resolve_xref(self, env: BuildEnvironment, fromdocname: str, builder: Builder,
typ: str, target: str, node: pending_xref, contnode: Element,
) -> nodes.reference | None:
def resolve_xref(
self,
env: BuildEnvironment,
fromdocname: str,
builder: Builder,
typ: str,
target: str,
node: pending_xref,
contnode: Element,
) -> nodes.reference | None:
assert typ in {'eq', 'numref'}
result = self.equations.get(target)
if result:
@ -104,7 +121,8 @@ class MathDomain(Domain):
node_id = make_id('equation-%s' % target)
if env.config.math_numfig and env.config.numfig:
if docname in env.toc_fignumbers:
numbers = env.toc_fignumbers[docname]['displaymath'].get(node_id, ())
toc_dm = env.toc_fignumbers[docname]['displaymath']
numbers = toc_dm.get(node_id, ())
eqno = '.'.join(map(str, numbers))
eqno = env.config.math_numsep.join(eqno.rsplit('.', 1))
else:
@ -113,21 +131,27 @@ class MathDomain(Domain):
eqno = str(number)
try:
eqref_format = env.config.math_eqref_format or "({number})"
eqref_format = env.config.math_eqref_format or '({number})'
title = nodes.Text(eqref_format.format(number=eqno))
except KeyError as exc:
logger.warning(__('Invalid math_eqref_format: %r'), exc,
location=node)
title = nodes.Text("(%d)" % number)
title = nodes.Text("(%d)" % number)
logger.warning(__('Invalid math_eqref_format: %r'), exc, location=node)
title = nodes.Text('(%d)' % number)
return make_refnode(builder, fromdocname, docname, node_id, title)
else:
return None
def resolve_any_xref(self, env: BuildEnvironment, fromdocname: str, builder: Builder,
target: str, node: pending_xref, contnode: Element,
) -> list[tuple[str, nodes.reference]]:
refnode = self.resolve_xref(env, fromdocname, builder, 'eq', target, node, contnode)
def resolve_any_xref(
self,
env: BuildEnvironment,
fromdocname: str,
builder: Builder,
target: str,
node: pending_xref,
contnode: Element,
) -> list[tuple[str, nodes.reference]]:
refnode = self.resolve_xref(
env, fromdocname, builder, 'eq', target, node, contnode
)
if refnode is None:
return []
else:
@ -143,7 +167,7 @@ class MathDomain(Domain):
return (
self.data['has_equations'].get(docname, False)
or any(map(self.has_equations, self.env.toctree_includes.get(docname, ())))
)
) # fmt: skip
def setup(app: Sphinx) -> ExtensionMetadata: