python-docx/features/steps/hyperlink.py at master · python-openxml/python-docx · GitHub

History

116 lines (84 loc) · 4.37 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

"""Step implementations for hyperlink-related features."""

from __future__ import annotations

from typing import Dict, Tuple

from behave import given, then

from behave.runner import Context

from docx import Document

from helpers import test_docx

# given ===================================================

@given("a hyperlink")

def given_a_hyperlink(context: Context):

document = Document(test_docx("par-hyperlinks"))

context.hyperlink = document.paragraphs[1].hyperlinks[0]

@given("a hyperlink having a URI fragment")

def given_a_hyperlink_having_a_uri_fragment(context: Context):

document = Document(test_docx("par-hlink-frags"))

context.hyperlink = document.paragraphs[1].hyperlinks[0]

@given("a hyperlink having address {address} and fragment {fragment}")

def given_a_hyperlink_having_address_and_fragment(context: Context, address: str, fragment: str):

paragraph_idxs: Dict[Tuple[str, str], int] = {

("''", "linkedBookmark"): 1,

("https://foo.com", "''"): 2,

("https://foo.com?q=bar", "''"): 3,

("http://foo.com/", "intro"): 4,

("https://foo.com?q=bar#baz", "''"): 5,

("court-exif.jpg", "''"): 7,

}

paragraph_idx = paragraph_idxs[(address, fragment)]

document = Document(test_docx("par-hlink-frags"))

paragraph = document.paragraphs[paragraph_idx]

context.hyperlink = paragraph.hyperlinks[0]

@given("a hyperlink having {zero_or_more} rendered page breaks")

def given_a_hyperlink_having_rendered_page_breaks(context: Context, zero_or_more: str):

paragraph_idx = {

"no": 1,

"one": 2,

}[zero_or_more]

document = Document(test_docx("par-hyperlinks"))

paragraph = document.paragraphs[paragraph_idx]

context.hyperlink = paragraph.hyperlinks[0]

@given("a hyperlink having {one_or_more} runs")

def given_a_hyperlink_having_one_or_more_runs(context: Context, one_or_more: str):

paragraph_idx, hyperlink_idx = {

"one": (1, 0),

"two": (2, 1),

}[one_or_more]

document = Document(test_docx("par-hyperlinks"))

paragraph = document.paragraphs[paragraph_idx]

context.hyperlink = paragraph.hyperlinks[hyperlink_idx]

# then =====================================================

@then("hyperlink.address is the URL of the hyperlink")

def then_hyperlink_address_is_the_URL_of_the_hyperlink(context: Context):

actual_value = context.hyperlink.address

expected_value = "http://yahoo.com/"