diff options
| -rwxr-xr-x | demo | 3 | ||||
| -rw-r--r-- | finance/flow.py | 14 | ||||
| -rw-r--r-- | finance/visualize.py | 16 |
3 files changed, 18 insertions, 15 deletions
@@ -6,7 +6,8 @@ import argparse from datetime import datetime from decimal import Decimal -from finance.flow import Flow, simulate, display +from finance.flow import Flow, simulate +from finance.visualize import display DESCRIPTION = "Demo application plotting financial data with finance-py" diff --git a/finance/flow.py b/finance/flow.py index bc2ee0a..4cf68db 100644 --- a/finance/flow.py +++ b/finance/flow.py @@ -3,8 +3,6 @@ from datetime import datetime from decimal import Decimal from typing import Generator -import matplotlib.pyplot - @dataclasses.dataclass(kw_only=True, frozen=True) class Flow: @@ -66,15 +64,3 @@ def simulate( values.append(value) return (dates, values) - - -def display(data: tuple[list[datetime], list[Decimal]]) -> None: - matplotlib.pyplot.plot( - list(data[0]), # type: ignore - [float(i) for i in data[1]], - label="Money", - ) - matplotlib.pyplot.xlabel("Time") - matplotlib.pyplot.legend() - matplotlib.pyplot.xticks(rotation=45) - matplotlib.pyplot.show() diff --git a/finance/visualize.py b/finance/visualize.py new file mode 100644 index 0000000..e6d609a --- /dev/null +++ b/finance/visualize.py @@ -0,0 +1,16 @@ +from datetime import datetime +from decimal import Decimal + +import matplotlib.pyplot + + +def display(data: tuple[list[datetime], list[Decimal]]) -> None: + matplotlib.pyplot.plot( + list(data[0]), # type: ignore + [float(i) for i in data[1]], + label="Money", + ) + matplotlib.pyplot.xlabel("Time") + matplotlib.pyplot.legend() + matplotlib.pyplot.xticks(rotation=45) + matplotlib.pyplot.show() |