self.logger = logger

    def section_key(self, row: AccrualPostings) -> Optional[data.Account]:

        if isinstance(row.account, str):

            return row.account

        else:

            return None

    def start_spreadsheet(self) -> None:

        for accrual_type in AccrualAccount:

            self.use_sheet(accrual_type.name.title())

            for index in range(self.COL_COUNT):

            self.add_row(*(

                self.string_cell(name, stylename=self.style_bold)

                for name in self.COLUMNS

            ))

            self.lock_first_row()

    def start_section(self, key: Optional[data.Account]) -> None:

        if key is None:

            return

        self.age_thresholds = list(AccrualAccount.by_account(key).value.aging_thresholds)

        self.age_balances = [core.MutableBalance() for _ in self.age_thresholds]

        accrual_date = self.date - datetime.timedelta(days=self.age_thresholds[-1])

class BaseODS(BaseSpreadsheet[RT, ST], metaclass=abc.ABCMeta):

    """Abstract base class to help write OpenDocument spreadsheets

    This class provides the very core logic to write an arbitrary set of data

    rows to an OpenDocument spreadsheet. It provides helper methods for

    building sheets, rows, and cells.

    See also the BaseSpreadsheet base class for additional documentation about

    methods you must and can define, the definition of RT and ST, etc.

    """

    def __init__(self, rt_wrapper: Optional[rtutil.RT]=None) -> None:

        self.rt_wrapper = rt_wrapper

        self.locale = babel.core.Locale.default('LC_MONETARY')

        self.currency_fmt_key = 'accounting'

        self._name_counter = itertools.count(1)

        self._currency_style_cache: MutableMapping[str, odf.style.Style] = {}

        self.document = odf.opendocument.OpenDocumentSpreadsheet()

        self.init_settings()

        self.init_styles()

        self.sheet = self.use_sheet("Report")

    ### Low-level document tree manipulation

                config_type = default_type

            except NameError:

                raise ValueError(

                    f"need config_type for {type(value).__name__} value",

                ) from None

        item = self.replace_child(

            root, odf.config.ConfigItem, name=name, type=config_type,

        )

        item.addText(value_s)

    ### Styles

    def _build_currency_style(

            self,

            root: odf.element.Element,

            locale: babel.core.Locale,

            code: str,

            fmt_index: int,

            properties: Optional[odf.style.TextProperties]=None,

            *,

            fmt_key: Optional[str]=None,

            volatile: bool=False,

            minintegerdigits: int=1,

    ) -> odf.element.Element:

        self.style_starttext: odf.style.Style

        self.style_centertext: odf.style.Style

        self.style_endtext: odf.style.Style

        for textalign in ['start', 'center', 'end']:

            aligned_style = self.replace_child(

                styles, odf.style.Style, name=f'{textalign.title()}Text',

            )

            aligned_style.setAttribute('family', 'table-cell')

            aligned_style.addElement(odf.style.ParagraphProperties(textalign=textalign))

            setattr(self, f'style_{textalign}text', aligned_style)

    ### Rows and cells

    def add_row(self, *cells: odf.table.TableCell, **attrs: Any) -> odf.table.TableRow:

        row = odf.table.TableRow(**attrs)

        for cell in cells:

            row.addElement(cell)

        self.sheet.addElement(row)

        return row

    def balance_cell(self, balance: Balance, **attrs: Any) -> odf.table.TableCell:

        if balance.is_zero():

            return self.float_cell(0, **attrs)

from typing import (

    Callable,

    Dict,

    Iterable,

    Iterator,

    List,

    Mapping,

    Optional,

    Sequence,

    TextIO,

    Tuple,

)

from pathlib import Path

import odf.table  # type:ignore[import]

from beancount.parser import printer as bc_printer

from . import core

from .. import books

from .. import cliutil

from .. import config as configmod

        'Booked Amount',

    ]

    ACCOUNT_COLUMNS: Dict[str, Sequence[str]] = collections.OrderedDict([

        ('Income', ['project', 'rt-id', 'receipt', 'income-type']),

        ('Expenses', ['project', 'rt-id', 'receipt', 'approval', 'expense-allocation']),

        ('Equity', ['rt-id']),

        ('Assets:Receivable', ['project', 'rt-id', 'invoice', 'approval', 'contract', 'purchase-order']),

        ('Liabilities:Payable', ['project', 'rt-id', 'invoice', 'approval', 'contract', 'purchase-order']),

        ('Assets:PayPal', ['rt-id', 'paypal-id', 'receipt', 'approval']),

        ('Assets', ['rt-id', 'receipt', 'approval', 'bank-statement']),

        ('Liabilities', ['rt-id', 'receipt', 'approval', 'bank-statement']),

    ])

    # Excel 2003 was limited to 65,536 rows per worksheet.

    # While we can probably count on all our users supporting more modern

    # formats (Excel 2007 supports over 1 million rows per worksheet),

    # keeping the default limit conservative seems good to avoid running into

    # other limits (like the number of hyperlinks per worksheet), plus just

    # better for human organization and readability.

    SHEET_SIZE = 65000

    def __init__(self,

                 start_date: datetime.date,

                 stop_date: datetime.date,

                 sheet_names: Optional[Sequence[str]]=None,

                 rt_wrapper: Optional[rtutil.RT]=None,

                 sheet_size: Optional[int]=None,

    ) -> None:

        if sheet_names is None:

            sheet_names = list(self.ACCOUNT_COLUMNS)

        if sheet_size is None:

            sheet_size = self.SHEET_SIZE

        super().__init__(rt_wrapper)

        self.date_range = ranges.DateRange(start_date, stop_date)

        self.required_sheet_names = sheet_names

        self.sheet_size = sheet_size

    @classmethod

    def _group_tally(

            cls,

            tally_by_account: PostTally,

            key: Callable[[data.Account], Optional[str]],

    ) -> Dict[str, PostTally]:

        retval: Dict[str, PostTally] = collections.defaultdict(list)

        for count, account in tally_by_account:

            item_key = key(account)

            if item_key is not None:

                retval[item_key].append((count, account))

        return retval

        self.use_sheet(sheet_name.replace(':', ' '))

        columns_key = data.Account(sheet_name).is_under(*self.ACCOUNT_COLUMNS)

        # columns_key must not be None because ACCOUNT_COLUMNS has an entry

        # for all five root accounts.

        assert columns_key is not None

        self.metadata_columns = self.ACCOUNT_COLUMNS[columns_key]

        self.sheet_columns: Sequence[str] = [

            *self.CORE_COLUMNS,

            *(data.Metadata.human_name(meta_key) for meta_key in self.metadata_columns),

        ]

        for col_name in self.sheet_columns:

            self.sheet.addElement(odf.table.TableColumn(

            ))

        self.add_row(*(

            self.string_cell(col_name, stylename=self.style_bold)

            for col_name in self.sheet_columns

        ))

        self.lock_first_row()

    def _report_section_balance(self, key: data.Account, date_key: str) -> None:

        uses_opening = key.is_under('Assets', 'Equity', 'Liabilities')

        if date_key == 'start':

            if not uses_opening:

                return

                stylename=self.merge_styles(self.style_bold, self.style_endtext),

            ),

            self.balance_cell(balance, stylename=self.style_bold),

        )

    def write_balance_sheet(self) -> None:

        balance_accounts = ['Equity', 'Income', 'Expenses']

        # FIXME: This is a hack to exclude non-project Equity accounts from

        # project reports.

        if balance_accounts[0] not in self.required_sheet_names:

            balance_accounts[0] = 'Equity:Funds'

        self.use_sheet("Balance")

        self.add_row(

            self.string_cell("Account", stylename=self.style_bold),

            self.string_cell("Balance", stylename=self.style_bold),

        )

        self.lock_first_row()

        self.add_row()

        self.add_row(self.string_cell(

            f"Ledger From {self.date_range.start.isoformat()}"

            f" To {self.date_range.stop.isoformat()}",

            stylename=self.merge_styles(self.style_centertext, self.style_bold),

            numbercolumnsspanned=2,

        ))

#!/usr/bin/env python3

from setuptools import setup

setup(

    name='conservancy_beancount',

    description="Plugin, library, and reports for reading Conservancy's books",

    author='Software Freedom Conservancy',

    author_email='info@sfconservancy.org',

    license='GNU AGPLv3+',

    install_requires=[

        'babel>=2.6',  # Debian:python3-babel

        'beancount>=2.2',  # Debian:beancount

        # 1.4.1 crashes when trying to save some documents.

        'odfpy>=1.4.0,!=1.4.1',  # Debian:python3-odf

        'PyYAML>=3.0',  # Debian:python3-yaml

        'regex',  # Debian:python3-regex

        'rt>=2.0',

        texts = [cell.text for row in testutil.ODSCell.from_sheet(actual)

                 for cell in row]

        assert texts == [name]

def test_ods_writer_use_sheet_discards_unused_sheets(ods_writer):

    ods_writer.use_sheet('Three')

    ods_writer.use_sheet('Two')

    ods_writer.use_sheet('One')

    sheets = ods_writer.document.getElementsByType(odf.table.Table)

    assert len(sheets) == 1

    assert sheets[0].getAttribute('name') == 'One'

@pytest.mark.parametrize('currency_code', [

    'USD',

    'EUR',

    'BRL',

])

def test_ods_currency_style(ods_writer, currency_code):

    style = ods_writer.currency_style(currency_code)

    assert style.getAttribute('family') == 'table-cell'

    curr_style = get_child(

        ods_writer.document.styles,

        odf.number.CurrencyStyle,

        name=style.getAttribute('datastylename'),

    actual = ods_writer.currency_style('USD')

    assert actual is expected

def test_ods_currency_style_cache_considers_properties(ods_writer):

    bold_text = odf.style.TextProperties(fontweight='bold')

    plain = ods_writer.currency_style('USD')

    bold = ods_writer.currency_style('USD', positive_properties=bold_text)

    assert plain is not bold

    assert plain.getAttribute('name') != bold.getAttribute('name')

    assert plain.getAttribute('datastylename') != bold.getAttribute('datastylename')

@pytest.mark.parametrize('attr_name,child_type,checked_attr', [

    ('style_bold', odf.style.TextProperties, 'fontweight'),

    ('style_centertext', odf.style.ParagraphProperties, 'textalign'),

    ('style_dividerline', odf.style.TableCellProperties, 'borderbottom'),

    ('style_endtext', odf.style.ParagraphProperties, 'textalign'),

    ('style_starttext', odf.style.ParagraphProperties, 'textalign'),

])

def test_ods_writer_style(ods_writer, attr_name, child_type, checked_attr):

    style = getattr(ods_writer, attr_name)

    actual = get_child(root, odf.style.Style, name=style.getAttribute('name'))

    assert actual is style

    child = get_child(actual, child_type)

    assert child.getAttribute(checked_attr)

def test_ods_writer_merge_styles(ods_writer):

    style = ods_writer.merge_styles(ods_writer.style_bold, ods_writer.style_dividerline)

    actual = get_child(

        ods_writer.document.styles,

        odf.style.Style,

        name=style.getAttribute('name'),