🎉 Exercism Research is now launched. Help Exercism, help science and have some fun at research.exercism.io 🎉
Avatar of nicolemon

nicolemon's solution

to Ledger in the Python Track

Published at Jul 27 2018 · 0 comments
Instructions
Test suite
Solution

Refactor a ledger printer.

The ledger exercise is a refactoring exercise. There is code that prints a nicely formatted ledger, given a locale (American or Dutch) and a currency (US dollar or euro). The code however is rather badly written, though (somewhat surprisingly) it consistently passes the test suite.

Rewrite this code. Remember that in refactoring the trick is to make small steps that keep the tests passing. That way you can always quickly go back to a working version. Version control tools like git can help here as well.

Please keep a log of what changes you've made and make a comment on the exercise containing that log, this will help reviewers.

Exception messages

Sometimes it is necessary to raise an exception. When you do this, you should include a meaningful error message to indicate what the source of the error is. This makes your code more readable and helps significantly with debugging. Not every exercise will require you to raise an exception, but for those that do, the tests will only pass if you include a message.

To raise a message with an exception, just write it as an argument to the exception type. For example, instead of raise Exception, you should write:

raise Exception("Meaningful message indicating the source of the error")

Running the tests

To run the tests, run the appropriate command below (why they are different):

  • Python 2.7: py.test ledger_test.py
  • Python 3.4+: pytest ledger_test.py

Alternatively, you can tell Python to run the pytest module (allowing the same command to be used regardless of Python version): python -m pytest ledger_test.py

Common pytest options

  • -v : enable verbose output
  • -x : stop running tests on first failure
  • --ff : run failures from previous test before running other test cases

For other options, see python -m pytest -h

Submitting Exercises

Note that, when trying to submit an exercise, make sure the solution is in the $EXERCISM_WORKSPACE/python/ledger directory.

You can find your Exercism workspace by running exercism debug and looking for the line that starts with Workspace.

For more detailed information about running tests, code style and linting, please see Running the Tests.

Submitting Incomplete Solutions

It's possible to submit an incomplete solution so you can see how others have completed the exercise.

ledger_test.py

# -*- coding: utf-8 -*-
import unittest

from ledger import format_entries, create_entry


class LedgerTest(unittest.TestCase):
    maxDiff = 5000

    def test_empty_ledger(self):
        currency = 'USD'
        locale = 'en_US'
        entries = []
        expected = 'Date       | Description               | Change       '
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_one_entry(self):
        currency = 'USD'
        locale = 'en_US'
        entries = [
            create_entry('2015-01-01', 'Buy present', -1000),
        ]
        expected = '\n'.join([
            'Date       | Description               | Change       ',
            '01/01/2015 | Buy present               |      ($10.00)',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_credit_and_debit(self):
        currency = 'USD'
        locale = 'en_US'
        entries = [
            create_entry('2015-01-02', 'Get present', 1000),
            create_entry('2015-01-01', 'Buy present', -1000),
        ]
        expected = '\n'.join([
            'Date       | Description               | Change       ',
            '01/01/2015 | Buy present               |      ($10.00)',
            '01/02/2015 | Get present               |       $10.00 ',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_multiple_entries_on_same_date_ordered_by_description(self):
        currency = 'USD'
        locale = 'en_US'
        entries = [
            create_entry('2015-01-02', 'Get present', 1000),
            create_entry('2015-01-01', 'Buy present', -1000),
        ]
        expected = '\n'.join([
            'Date       | Description               | Change       ',
            '01/01/2015 | Buy present               |      ($10.00)',
            '01/02/2015 | Get present               |       $10.00 ',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_final_order_tie_breaker_is_change(self):
        currency = 'USD'
        locale = 'en_US'
        entries = [
            create_entry('2015-01-01', 'Something', 0),
            create_entry('2015-01-01', 'Something', -1),
            create_entry('2015-01-01', 'Something', 1),
        ]
        expected = '\n'.join([
            'Date       | Description               | Change       ',
            '01/01/2015 | Something                 |       ($0.01)',
            '01/01/2015 | Something                 |        $0.00 ',
            '01/01/2015 | Something                 |        $0.01 ',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_overlong_description(self):
        currency = 'USD'
        locale = 'en_US'
        entries = [
            create_entry('2015-01-01', 'Freude schoner Gotterfunken', -123456),
        ]
        expected = '\n'.join([
            'Date       | Description               | Change       ',
            '01/01/2015 | Freude schoner Gotterf... |   ($1,234.56)',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_euros(self):
        currency = 'EUR'
        locale = 'en_US'
        entries = [
            create_entry('2015-01-01', 'Buy present', -1000),
        ]
        expected = '\n'.join([
            'Date       | Description               | Change       ',
            u'01/01/2015 | Buy present               |      (€10.00)',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_dutch_locale(self):
        currency = 'USD'
        locale = 'nl_NL'
        entries = [
            create_entry('2015-03-12', 'Buy present', 123456),
        ]
        expected = '\n'.join([
            'Datum      | Omschrijving              | Verandering  ',
            '12-03-2015 | Buy present               |   $ 1.234,56 ',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_dutch_locale_and_euros(self):
        currency = 'EUR'
        locale = 'nl_NL'
        entries = [
            create_entry('2015-03-12', 'Buy present', 123456),
        ]
        expected = '\n'.join([
            'Datum      | Omschrijving              | Verandering  ',
            u'12-03-2015 | Buy present               |   € 1.234,56 ',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_dutch_negative_number_with_3_digits_before_decimal_point(self):
        currency = 'USD'
        locale = 'nl_NL'
        entries = [
            create_entry('2015-03-12', 'Buy present', -12345),
        ]
        expected = '\n'.join([
            'Datum      | Omschrijving              | Verandering  ',
            '12-03-2015 | Buy present               |    $ -123,45 ',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)

    def test_american_negative_number_with_3_digits_before_decimal_point(self):
        currency = 'USD'
        locale = 'en_US'
        entries = [
            create_entry('2015-03-12', 'Buy present', -12345),
        ]
        expected = '\n'.join([
            'Date       | Description               | Change       ',
            '03/12/2015 | Buy present               |     ($123.45)',
        ])
        self.assertEqual(format_entries(currency, locale, entries), expected)


if __name__ == '__main__':
    unittest.main()
# -*- coding: utf-8 -*-
from datetime import datetime


class en_US:
    DATE = 'Date'
    DESCRIPTION = 'Description'
    CHANGE = 'Change'
    DATE_FORMAT = '%m/%d/%Y'
    AMOUNT = '{symbol}{amount} '
    NEGATIVE_AMOUNT = '({symbol}{amount})'
    TSEP = ','
    DSEP = '.'


class USD:
    SYMBOL = '$'


class nl_NL:
    DATE = 'Datum'
    DESCRIPTION = 'Omschrijving'
    CHANGE = 'Verandering'
    DATE_FORMAT = '%d-%m-%Y'
    AMOUNT = '{symbol} {amount} '
    NEGATIVE_AMOUNT = '{symbol} -{amount} '
    TSEP = '.'
    DSEP = ','


class EUR:
    SYMBOL = u'€'


class LedgerEntry(object):
    def __init__(self, date=None, description=None, change=None):
        self.date = date
        self.description = description
        self.change = change

    def __lt__(a, b):
        if a.date < b.date:
            return True
        elif a.date == b.date:
            if a.change < b.change:
                return True
            elif a.change == b.change:
                if a.description < b.description:
                    return True
            else:
                return False
        else:
            return False


    def format_date(self, locale):
        return datetime.strftime(self.date, locale.DATE_FORMAT)

    def format_description(self):
        if len(self.description) > 25:
            return '{}...'.format(self.description[:22])
        else:
            return self.description

    def format_change(self, currency, locale):
        if self.change < 0:
            change_fmt = locale.NEGATIVE_AMOUNT
        else:
            change_fmt = locale.AMOUNT

        amount = abs(int(self.change / 100.0))
        if amount != 0:
            amount_parts = []
            while amount > 0:
                leading = amount // 1000
                trailing = amount % 1000
                amount_parts.append(str(trailing))
                amount = leading
            amount = locale.TSEP.join(amount_parts[::-1])
        else:
            amount = '0'

        cents = '{:02d}'.format(abs(self.change) % 100)

        change = locale.DSEP.join([amount, cents])

        return change_fmt.format(symbol=currency.SYMBOL, amount=change)


def create_entry(date, description, change):
    date = datetime.strptime(date, '%Y-%m-%d')
    return LedgerEntry(date=date, description=description, change=change)


def format_entries(currency, locale, entries):
    currency = eval(currency)
    locale = eval(locale)

    table = []
    header_row = '{:<10} | {:<25} | {:<13}'.format(locale.DATE, locale.DESCRIPTION, locale.CHANGE)
    table.append(header_row)

    entries.sort()

    for entry in entries:
        date = entry.format_date(locale)
        description = '{:<25}'.format(entry.format_description())
        change = '{:>13}'.format(entry.format_change(currency, locale))
        table_entry = ' | '.join([date, description, change])

        table.append(table_entry)

    return '\n'.join(table)

Community comments

Find this solution interesting? Ask the author a question to learn more.

What can you learn from this solution?

A huge amount can be learned from reading other people’s code. This is why we wanted to give exercism users the option of making their solutions public.

Here are some questions to help you reflect on this solution and learn the most from it.

  • What compromises have been made?
  • Are there new concepts here that you could read more about to improve your understanding?