Sequence,

    Union,

)

from .beancount_types import (

    MetaKey,

    MetaValue,

    Posting as BasePosting,

    Transaction,

)

DecimalCompat = Union[decimal.Decimal, int]

LINK_METADATA = frozenset([

    'approval',

    'check',

    'contract',

    'invoice',

    'purchase-order',

    'receipt',

    'statement',

])

class Account(str):

    """Account name string

    This is a string that names an account, like Accrued:AccountsPayable

    or Income:Donations. This class provides additional methods for common

    account name parsing and queries.

    """

    SEP = bc_account.sep

    def is_income(self) -> bool:

        return self.is_under('Income:', 'UnearnedIncome:') is not None

    def is_real_asset(self) -> bool:

        return bool(

            self.is_under('Assets:')

            and not self.is_under('Assets:PrepaidExpenses', 'Assets:PrepaidVacation')

        )

    def is_under(self, *acct_seq: str) -> Optional[str]:

        """Return a match if this account is "under" a part of the hierarchy

        Pass in any number of account name strings as arguments. If this

        account is under one of those strings in the account hierarchy, the

        first matching string will be returned. Otherwise, None is returned.

        You can use the return value of this method as a boolean if you don't

        care which account string is matched.

        An account is considered to be under itself:

          Account('Expenses:Tax').is_under('Expenses:Tax') # returns 'Expenses:Tax'

        To do a "strictly under" search, end your search strings with colons:

          Account('Expenses:Tax').is_under('Expenses:Tax:') # returns None

          Account('Expenses:Tax').is_under('Expenses:') # returns 'Expenses:'

        This method does check that all the account boundaries match:

          Account('Expenses:Tax').is_under('Exp') # returns None

        """

def test_is_under_multi_arg(acct_name, expected):

    assert data.Account(acct_name).is_under('Income', 'UnearnedIncome') == expected

    if expected:

        expected += ':'

    assert data.Account(acct_name).is_under('Income:', 'UnearnedIncome:') == expected

@pytest.mark.parametrize('acct_name,expected', [

    ('Accrued:AccountsReceivable', False),

    ('Assets:Cash', False),

    ('Expenses:General', False),

    ('Income:Donations', True),

    ('Income:Sales', True),

    ('Income:Other', True),

    ('Liabilities:CreditCard', False),

    ('UnearnedIncome:MatchPledges', True),

])

def test_is_income(acct_name, expected):

    assert data.Account(acct_name).is_income() == expected

@pytest.mark.parametrize('acct_name,expected', [

    ('Accrued:AccountsPayable', False),

    ('Accrued:AccountsReceivable', False),

    ('Assets:Cash', True),

    ('Assets:Cash:EUR', True),

    ('Assets:PrepaidExpenses', False),

    ('Assets:PrepaidVacation', False),

    ('Assets:Bank:Checking', True),

    ('Expenses:General', False),

    ('Income:Donations', False),

])

def test_is_real_asset(acct_name, expected):

    assert data.Account(acct_name).is_real_asset() == expected