from dataclasses import dataclass, field
from typing import TYPE_CHECKING, Iterable, List, Union
from sphinxcontrib.bibtex.style.template import (
entry_label,
join,
join2,
post_text,
pre_text,
reference,
)
from . import BaseReferenceStyle, BracketStyle, PersonStyle
if TYPE_CHECKING:
from pybtex.richtext import BaseText
from pybtex.style.template import Node
[docs]@dataclass
class BasicLabelParentheticalReferenceStyle(BaseReferenceStyle):
"""Reference by label if parenthetical,
and by author and label if textual.
"""
#: Bracket style.
bracket: BracketStyle = field(default_factory=BracketStyle)
#: Separator between pre-text and citation.
pre_text_sep: Union["BaseText", str] = " "
#: Separator between citation and post-text.
post_text_sep: Union["BaseText", str] = ", "
[docs] def role_names(self) -> Iterable[str]:
return [
f"{alt}p{full_author}" for alt in ["", "al"] for full_author in ["", "s"]
]
[docs] def outer(self, role_name: str, children: List["BaseText"]) -> "Node":
return self.bracket.outer(
children, brackets="al" not in role_name, capfirst=False
)
[docs] def inner(self, role_name: str) -> "Node":
return join2(sep1=self.pre_text_sep, sep2=self.post_text_sep)[
pre_text,
reference[entry_label],
post_text,
]
[docs]@dataclass
class BasicLabelTextualReferenceStyle(BaseReferenceStyle):
"""Reference by label if parenthetical,
and by author and label if textual.
"""
#: Bracket style.
bracket: BracketStyle = field(default_factory=BracketStyle)
#: Person style.
person: PersonStyle = field(default_factory=PersonStyle)
#: Separator between text and reference.
text_reference_sep: Union["BaseText", str] = " "
#: Separator between pre-text and citation.
pre_text_sep: Union["BaseText", str] = " "
#: Separator between citation and post-text.
post_text_sep: Union["BaseText", str] = ", "
[docs] def role_names(self) -> Iterable[str]:
return [
f"{capfirst}t{full_author}"
for capfirst in ["", "c"]
for full_author in ["", "s"]
]
[docs] def outer(self, role_name: str, children: List["BaseText"]) -> "Node":
return self.bracket.outer(children, brackets=False, capfirst="c" in role_name)
[docs] def inner(self, role_name: str) -> "Node":
return join(sep=self.text_reference_sep)[
self.person.author_or_editor_or_title(full="s" in role_name),
join[
self.bracket.left,
join2(sep1=self.pre_text_sep, sep2=self.post_text_sep)[
pre_text,
reference[entry_label],
post_text,
],
self.bracket.right,
],
]