Refer to the LICENSE.txt that came with the software for details.

This program is distributed in the hope that it will be useful,

but WITHOUT ANY WARRANTY; without even the implied warranty of

MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE."""

import argparse

import datetime

import enum

import io

import logging

import operator

import os

import pkg_resources

import re

import signal

import sys

import traceback

import types

from pathlib import Path

import rt.exceptions as rt_error

from . import data

from . import filters

from . import rtutil

from typing import (

    cast,

    Any,

    BinaryIO,

    Callable,

    IO,

    Iterable,

    NamedTuple,

    NoReturn,

    Optional,

    Sequence,

    TextIO,

    Type,

    Union,

)

from .beancount_types import (

    MetaKey,

)

OutputFile = Union[int, IO]

        metavar='PATH',

        type=Path,

        help="""Use rewrite rules from the given YAML file. You can specify

this option multiple times to load multiple sets of rewrite rules in order.

""")

def add_version_argument(parser: argparse.ArgumentParser) -> argparse.Action:

    progname = parser.prog or sys.argv[0]

    return parser.add_argument(

        '--version', '--copyright', '--license',

        action=InfoAction,

        nargs=0,

        const=f"{progname} version {VERSION}\n{LICENSE}",

        help="Show program version and license information",

    )

def date_arg(arg: str) -> datetime.date:

    return datetime.datetime.strptime(arg, '%Y-%m-%d').date()

def diff_year(date: datetime.date, diff: int) -> datetime.date:

    new_year = date.year + diff

    try:

        return date.replace(year=new_year)

    except ValueError:

        # The original date is Feb 29, which doesn't exist in the new year.

        if diff < 0:

            return datetime.date(new_year, 2, 28)

        else:

            return datetime.date(new_year, 3, 1)

def year_or_date_arg(arg: str) -> Union[int, datetime.date]:

    """Get either a date or a year (int) from an argument string

import argparse

import datetime

import errno

import io

import inspect

import logging

import os

import re

import sys

import traceback

import pytest

from pathlib import Path

from conservancy_beancount import cliutil

FILE_NAMES = ['-foobar', '-foo.bin']

STREAM_PATHS = [None, Path('-')]

class MockTraceback:

    def __init__(self, stack=None, index=0):

        if stack is None:

            stack = inspect.stack(context=False)

        self._stack = stack

        self._index = index

        frame_record = self._stack[self._index]

        self.tb_frame = frame_record.frame

        self.tb_lineno = frame_record.lineno

    @property

    (datetime.date(2010, 2, 28), 0, datetime.date(2010, 2, 28)),

    (datetime.date(2010, 2, 28), 1, datetime.date(2011, 2, 28)),

    (datetime.date(2010, 2, 28), 2, datetime.date(2012, 2, 28)),

    (datetime.date(2010, 2, 28), -1, datetime.date(2009, 2, 28)),

    (datetime.date(2010, 2, 28), -2, datetime.date(2008, 2, 28)),

    (datetime.date(2012, 2, 29), 2, datetime.date(2014, 3, 1)),

    (datetime.date(2012, 2, 29), 4, datetime.date(2016, 2, 29)),

    (datetime.date(2012, 2, 29), -2, datetime.date(2010, 2, 28)),

    (datetime.date(2012, 2, 29), -4, datetime.date(2008, 2, 29)),

    (datetime.date(2010, 3, 1), 1, datetime.date(2011, 3, 1)),

    (datetime.date(2010, 3, 1), 2, datetime.date(2012, 3, 1)),

    (datetime.date(2010, 3, 1), -1, datetime.date(2009, 3, 1)),

    (datetime.date(2010, 3, 1), -2, datetime.date(2008, 3, 1)),

])

def test_diff_year(date, diff, expected):

    assert cliutil.diff_year(date, diff) == expected