diff --git a/initializer.py b/initializer.py index 1b9d2a3..2879c99 100644 --- a/initializer.py +++ b/initializer.py @@ -4,9 +4,10 @@ import logging import matplotlib.pyplot as plt import pickle import sys -from categories import Categories -from transaction import Transaction as Tr, TransactionError, Transactions -from parsers import Parser + +from pfbudget.categories import Categories +from pfbudget.transactions import Transaction as Tr, TransactionError, Transactions +from pfbudget.parsers import Parser def get_transactions(data_dir): diff --git a/main.py b/main.py index be66ae0..be46520 100644 --- a/main.py +++ b/main.py @@ -5,9 +5,9 @@ import matplotlib.pyplot as plt import pickle from initializer import initialize -from categories import Categories -from transaction import Transaction as Tr, TransactionError, Transactions -from parsers import Parser +from pfbudget.categories import Categories +from pfbudget.transactions import Transaction as Tr, TransactionError, Transactions +from pfbudget.parsers import Parser p = ".pfbudget.pickle" diff --git a/pfbudget/__init__.py b/pfbudget/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/categories.py b/pfbudget/categories.py similarity index 100% rename from categories.py rename to pfbudget/categories.py diff --git a/parsers.py b/pfbudget/parsers.py similarity index 99% rename from parsers.py rename to pfbudget/parsers.py index ecbb22f..3f9e66c 100644 --- a/parsers.py +++ b/pfbudget/parsers.py @@ -1,7 +1,8 @@ from datetime import datetime from decimal import Decimal, InvalidOperation from pathlib import Path -from transaction import Transaction + +from .transactions import Transaction class Parser: diff --git a/transaction.py b/pfbudget/transactions.py similarity index 99% rename from transaction.py rename to pfbudget/transactions.py index 62eab01..89edb99 100644 --- a/transaction.py +++ b/pfbudget/transactions.py @@ -1,8 +1,8 @@ -from categories import Categories from csv import reader, writer from datetime import date from decimal import Decimal, InvalidOperation +from .categories import Categories class TransactionError(Exception): pass