# encoding: utf-8

"""
Step implementations for section-related features
"""

from __future__ import absolute_import, print_function, unicode_literals

from behave import given, then, when

from docx import Document
from docx.enum.section import WD_ORIENT, WD_SECTION
from docx.section import Section
from docx.shared import Inches

from helpers import test_docx


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

@given("a Section object as section")
def given_a_Section_object_as_section(context):
    context.section = Document(test_docx("sct-section-props")).sections[-1]


@given("a Section object {with_or_without} a distinct first-page header as section")
def given_a_Section_object_with_or_without_first_page_header(context, with_or_without):
    section_idx = {"with": 1, "without": 0}[with_or_without]
    context.section = Document(test_docx("sct-first-page-hdrftr")).sections[section_idx]


@given('a section collection containing 3 sections')
def given_a_section_collection_containing_3_sections(context):
    document = Document(test_docx('doc-access-sections'))
    context.sections = document.sections


@given('a section having known page dimension')
def given_a_section_having_known_page_dimension(context):
    document = Document(test_docx('sct-section-props'))
    context.section = document.sections[-1]


@given('a section having known page margins')
def given_a_section_having_known_page_margins(context):
    document = Document(test_docx('sct-section-props'))
    context.section = document.sections[0]


@given('a section having start type {start_type}')
def given_a_section_having_start_type(context, start_type):
    section_idx = {
        'CONTINUOUS': 0,
        'NEW_PAGE':   1,
        'ODD_PAGE':   2,
        'EVEN_PAGE':  3,
        'NEW_COLUMN': 4,
    }[start_type]
    document = Document(test_docx('sct-section-props'))
    context.section = document.sections[section_idx]


@given('a section known to have {orientation} orientation')
def given_a_section_having_known_orientation(context, orientation):
    section_idx = {
        'landscape': 0,
        'portrait':  1
    }[orientation]
    document = Document(test_docx('sct-section-props'))
    context.section = document.sections[section_idx]


# when =====================================================

@when("I assign {bool_val} to section.different_first_page_header_footer")
def when_I_assign_value_to_section_different_first_page_hdrftr(context, bool_val):
    context.section.different_first_page_header_footer = eval(bool_val)


@when('I set the {margin_side} margin to {inches} inches')
def when_I_set_the_margin_side_length(context, margin_side, inches):
    prop_name = {
        'left':   'left_margin',
        'right':  'right_margin',
        'top':    'top_margin',
        'bottom': 'bottom_margin',
        'gutter': 'gutter',
        'header': 'header_distance',
        'footer': 'footer_distance',
    }[margin_side]
    new_value = Inches(float(inches))
    setattr(context.section, prop_name, new_value)


@when('I set the section orientation to {orientation}')
def when_I_set_the_section_orientation(context, orientation):
    new_orientation = {
        'WD_ORIENT.PORTRAIT':  WD_ORIENT.PORTRAIT,
        'WD_ORIENT.LANDSCAPE': WD_ORIENT.LANDSCAPE,
        'None':                None,
    }[orientation]
    context.section.orientation = new_orientation


@when('I set the section page height to {y} inches')
def when_I_set_the_section_page_height_to_y_inches(context, y):
    context.section.page_height = Inches(float(y))


@when('I set the section page width to {x} inches')
def when_I_set_the_section_page_width_to_x_inches(context, x):
    context.section.page_width = Inches(float(x))


@when('I set the section start type to {start_type}')
def when_I_set_the_section_start_type_to_start_type(context, start_type):
    new_start_type = {
        'None':       None,
        'CONTINUOUS': WD_SECTION.CONTINUOUS,
        'EVEN_PAGE':  WD_SECTION.EVEN_PAGE,
        'NEW_COLUMN': WD_SECTION.NEW_COLUMN,
        'NEW_PAGE':   WD_SECTION.NEW_PAGE,
        'ODD_PAGE':   WD_SECTION.ODD_PAGE,
    }[start_type]
    context.section.start_type = new_start_type


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

@then('I can access a section by index')
def then_I_can_access_a_section_by_index(context):
    sections = context.sections
    for idx in range(3):
        section = sections[idx]
        assert isinstance(section, Section)


@then('I can iterate over the sections')
def then_I_can_iterate_over_the_sections(context):
    sections = context.sections
    actual_count = 0
    for section in sections:
        actual_count += 1
        assert isinstance(section, Section)
    assert actual_count == 3


@then('len(sections) is 3')
def then_len_sections_is_3(context):
    sections = context.sections
    assert len(sections) == 3, (
        'expected len(sections) of 3, got %s' % len(sections)
    )


@then("section.different_first_page_header_footer is {bool_val}")
def then_section_different_first_page_header_footer_is(context, bool_val):
    actual = context.section.different_first_page_header_footer
    expected = eval(bool_val)
    assert actual == expected, (
        "section.different_first_page_header_footer is %s" % actual
    )


@then("section.even_page_footer is a _Footer object")
def then_section_even_page_footer_is_a_Footer_object(context):
    actual = type(context.section.even_page_footer).__name__
    expected = "_Footer"
    assert actual == expected, "section.even_page_footer is a %s object" % actual


@then("section.even_page_header is a _Header object")
def then_section_even_page_header_is_a_Header_object(context):
    actual = type(context.section.even_page_header).__name__
    expected = "_Header"
    assert actual == expected, "section.even_page_header is a %s object" % actual


@then("section.first_page_footer is a _Footer object")
def then_section_first_page_footer_is_a_Footer_object(context):
    actual = type(context.section.first_page_footer).__name__
    expected = "_Footer"
    assert actual == expected, "section.first_page_footer is a %s object" % actual


@then("section.first_page_header is a _Header object")
def then_section_first_page_header_is_a_Header_object(context):
    actual = type(context.section.first_page_header).__name__
    expected = "_Header"
    assert actual == expected, "section.first_page_header is a %s object" % actual


@then("section.footer is a _Footer object")
def then_section_footer_is_a_Footer_object(context):
    actual = type(context.section.footer).__name__
    expected = "_Footer"
    assert actual == expected, "section.footer is a %s object" % actual


@then("section.header is a _Header object")
def then_section_header_is_a_Header_object(context):
    actual = type(context.section.header).__name__
    expected = "_Header"
    assert actual == expected, "section.header is a %s object" % actual


@then("section.{propname}.is_linked_to_previous is True")
def then_section_hdrftr_prop_is_linked_to_previous_is_True(context, propname):
    actual = getattr(context.section, propname).is_linked_to_previous
    expected = True
    assert actual == expected, (
        "section.%s.is_linked_to_previous is %s" % (propname, actual)
    )


@then('the reported {margin_side} margin is {inches} inches')
def then_the_reported_margin_is_inches(context, margin_side, inches):
    prop_name = {
        'left':   'left_margin',
        'right':  'right_margin',
        'top':    'top_margin',
        'bottom': 'bottom_margin',
        'gutter': 'gutter',
        'header': 'header_distance',
        'footer': 'footer_distance',
    }[margin_side]
    expected_value = Inches(float(inches))
    actual_value = getattr(context.section, prop_name)
    assert actual_value == expected_value


@then('the reported page orientation is {orientation}')
def then_the_reported_page_orientation_is_orientation(context, orientation):
    expected_value = {
        'WD_ORIENT.LANDSCAPE': WD_ORIENT.LANDSCAPE,
        'WD_ORIENT.PORTRAIT':  WD_ORIENT.PORTRAIT,
    }[orientation]
    assert context.section.orientation == expected_value


@then('the reported page width is {x} inches')
def then_the_reported_page_width_is_width(context, x):
    assert context.section.page_width == Inches(float(x))


@then('the reported page height is {y} inches')
def then_the_reported_page_height_is_11_inches(context, y):
    assert context.section.page_height == Inches(float(y))


@then('the reported section start type is {start_type}')
def then_the_reported_section_start_type_is_type(context, start_type):
    expected_start_type = {
        'CONTINUOUS': WD_SECTION.CONTINUOUS,
        'EVEN_PAGE':  WD_SECTION.EVEN_PAGE,
        'NEW_COLUMN': WD_SECTION.NEW_COLUMN,
        'NEW_PAGE':   WD_SECTION.NEW_PAGE,
        'ODD_PAGE':   WD_SECTION.ODD_PAGE,
    }[start_type]
    assert context.section.start_type == expected_start_type

[((428, 464), 'behave.given', 'given', (['"""a Section object as section"""'], {}), "('a Section object as section')\n", (433, 464), False, 'from behave import given, then, when\n'), ((592, 685), 'behave.given', 'given', (['"""a Section object {with_or_without} a distinct first-page header as section"""'], {}), "(\n    'a Section object {with_or_without} a distinct first-page header as section'\n    )\n", (597, 685), False, 'from behave import given, then, when\n'), ((917, 968), 'behave.given', 'given', (['"""a section collection containing 3 sections"""'], {}), "('a section collection containing 3 sections')\n", (922, 968), False, 'from behave import given, then, when\n'), ((1134, 1180), 'behave.given', 'given', (['"""a section having known page dimension"""'], {}), "('a section having known page dimension')\n", (1139, 1180), False, 'from behave import given, then, when\n'), ((1342, 1386), 'behave.given', 'given', (['"""a section having known page margins"""'], {}), "('a section having known page margins')\n", (1347, 1386), False, 'from behave import given, then, when\n'), ((1545, 1594), 'behave.given', 'given', (['"""a section having start type {start_type}"""'], {}), "('a section having start type {start_type}')\n", (1550, 1594), False, 'from behave import given, then, when\n'), ((1930, 1988), 'behave.given', 'given', (['"""a section known to have {orientation} orientation"""'], {}), "('a section known to have {orientation} orientation')\n", (1935, 1988), False, 'from behave import given, then, when\n'), ((2317, 2390), 'behave.when', 'when', (['"""I assign {bool_val} to section.different_first_page_header_footer"""'], {}), "('I assign {bool_val} to section.different_first_page_header_footer')\n", (2321, 2390), False, 'from behave import given, then, when\n'), ((2549, 2606), 'behave.when', 'when', (['"""I set the {margin_side} margin to {inches} inches"""'], {}), "('I set the {margin_side} margin to {inches} inches')\n", (2553, 2606), False, 'from behave import given, then, when\n'), ((3041, 3095), 'behave.when', 'when', (['"""I set the section orientation to {orientation}"""'], {}), "('I set the section orientation to {orientation}')\n", (3045, 3095), False, 'from behave import given, then, when\n'), ((3394, 3445), 'behave.when', 'when', (['"""I set the section page height to {y} inches"""'], {}), "('I set the section page height to {y} inches')\n", (3398, 3445), False, 'from behave import given, then, when\n'), ((3564, 3614), 'behave.when', 'when', (['"""I set the section page width to {x} inches"""'], {}), "('I set the section page width to {x} inches')\n", (3568, 3614), False, 'from behave import given, then, when\n'), ((3731, 3783), 'behave.when', 'when', (['"""I set the section start type to {start_type}"""'], {}), "('I set the section start type to {start_type}')\n", (3735, 3783), False, 'from behave import given, then, when\n'), ((4260, 4299), 'behave.then', 'then', (['"""I can access a section by index"""'], {}), "('I can access a section by index')\n", (4264, 4299), False, 'from behave import given, then, when\n'), ((4487, 4526), 'behave.then', 'then', (['"""I can iterate over the sections"""'], {}), "('I can iterate over the sections')\n", (4491, 4526), False, 'from behave import given, then, when\n'), ((4762, 4788), 'behave.then', 'then', (['"""len(sections) is 3"""'], {}), "('len(sections) is 3')\n", (4766, 4788), False, 'from behave import given, then, when\n'), ((4962, 5026), 'behave.then', 'then', (['"""section.different_first_page_header_footer is {bool_val}"""'], {}), "('section.different_first_page_header_footer is {bool_val}')\n", (4966, 5026), False, 'from behave import given, then, when\n'), ((5306, 5358), 'behave.then', 'then', (['"""section.even_page_footer is a _Footer object"""'], {}), "('section.even_page_footer is a _Footer object')\n", (5310, 5358), False, 'from behave import given, then, when\n'), ((5593, 5645), 'behave.then', 'then', (['"""section.even_page_header is a _Header object"""'], {}), "('section.even_page_header is a _Header object')\n", (5597, 5645), False, 'from behave import given, then, when\n'), ((5880, 5933), 'behave.then', 'then', (['"""section.first_page_footer is a _Footer object"""'], {}), "('section.first_page_footer is a _Footer object')\n", (5884, 5933), False, 'from behave import given, then, when\n'), ((6171, 6224), 'behave.then', 'then', (['"""section.first_page_header is a _Header object"""'], {}), "('section.first_page_header is a _Header object')\n", (6175, 6224), False, 'from behave import given, then, when\n'), ((6462, 6504), 'behave.then', 'then', (['"""section.footer is a _Footer object"""'], {}), "('section.footer is a _Footer object')\n", (6466, 6504), False, 'from behave import given, then, when\n'), ((6709, 6751), 'behave.then', 'then', (['"""section.header is a _Header object"""'], {}), "('section.header is a _Header object')\n", (6713, 6751), False, 'from behave import given, then, when\n'), ((6956, 7012), 'behave.then', 'then', (['"""section.{propname}.is_linked_to_previous is True"""'], {}), "('section.{propname}.is_linked_to_previous is True')\n", (6960, 7012), False, 'from behave import given, then, when\n'), ((7294, 7354), 'behave.then', 'then', (['"""the reported {margin_side} margin is {inches} inches"""'], {}), "('the reported {margin_side} margin is {inches} inches')\n", (7298, 7354), False, 'from behave import given, then, when\n'), ((7841, 7895), 'behave.then', 'then', (['"""the reported page orientation is {orientation}"""'], {}), "('the reported page orientation is {orientation}')\n", (7845, 7895), False, 'from behave import given, then, when\n'), ((8178, 8223), 'behave.then', 'then', (['"""the reported page width is {x} inches"""'], {}), "('the reported page width is {x} inches')\n", (8182, 8223), False, 'from behave import given, then, when\n'), ((8340, 8386), 'behave.then', 'then', (['"""the reported page height is {y} inches"""'], {}), "('the reported page height is {y} inches')\n", (8344, 8386), False, 'from behave import given, then, when\n'), ((8509, 8564), 'behave.then', 'then', (['"""the reported section start type is {start_type}"""'], {}), "('the reported section start type is {start_type}')\n", (8513, 8564), False, 'from behave import given, then, when\n'), ((1056, 1088), 'helpers.test_docx', 'test_docx', (['"""doc-access-sections"""'], {}), "('doc-access-sections')\n", (1065, 1088), False, 'from helpers import test_docx\n'), ((1263, 1293), 'helpers.test_docx', 'test_docx', (['"""sct-section-props"""'], {}), "('sct-section-props')\n", (1272, 1293), False, 'from helpers import test_docx\n'), ((1467, 1497), 'helpers.test_docx', 'test_docx', (['"""sct-section-props"""'], {}), "('sct-section-props')\n", (1476, 1497), False, 'from helpers import test_docx\n'), ((1842, 1872), 'helpers.test_docx', 'test_docx', (['"""sct-section-props"""'], {}), "('sct-section-props')\n", (1851, 1872), False, 'from helpers import test_docx\n'), ((2167, 2197), 'helpers.test_docx', 'test_docx', (['"""sct-section-props"""'], {}), "('sct-section-props')\n", (2176, 2197), False, 'from helpers import test_docx\n'), ((544, 574), 'helpers.test_docx', 'test_docx', (['"""sct-section-props"""'], {}), "('sct-section-props')\n", (553, 574), False, 'from helpers import test_docx\n'), ((856, 890), 'helpers.test_docx', 'test_docx', (['"""sct-first-page-hdrftr"""'], {}), "('sct-first-page-hdrftr')\n", (865, 890), False, 'from helpers import test_docx\n')]