decimal-or-float, conversion

1 files changed, 9 insertions, 6 deletions

diff --git a/trade_queue.py b/trade_queue.py
index 1048b8a..ed9902e 100644
--- a/trade_queue.py
+++ b/trade_queue.py
@@ -1,4 +1,5 @@
 from collections import deque
+from decimal import Decimal
 from typing import Deque, List
 
 from trade import Trade
@@ -13,14 +14,14 @@ class FIFOQueue:
     def __init__(self) -> None:
         self.queue: Deque[Trade] = deque()
 
-    def add(self, amount: float, total_cost: float, date: str) -> None:
+    def add(self, amount: float|Decimal, total_cost: float|Decimal, date: str) -> None:
         """
         Add a trade to the queue.
         """
         trade = Trade(amount, total_cost, date)
         self.queue.append(trade)
 
-    def remove(self, amount: float) -> List[Trade]:
+    def remove(self, amount: float|Decimal) -> List[Trade]:
         """
         Remove a specified amount from the queue, returning the
         trades used to buy.
@@ -28,7 +29,9 @@ class FIFOQueue:
         if amount <= 0:
             raise ValueError("The amount to remove must be positive.")
 
-        remaining: float = amount
+        amount = Decimal(amount)
+
+        remaining: Decimal = amount
         entries: List[Trade] = []
 
         while remaining > 0:
@@ -39,7 +42,7 @@ class FIFOQueue:
             if trade.amount > remaining:
                 trade.remove_coins(remaining)
                 entries.append(Trade(remaining, trade.total_cost, trade.date))
-                remaining = 0
+                remaining = Decimal(0)
             else:
                 remaining -= trade.amount
                 entries.append(trade)
@@ -47,8 +50,8 @@ class FIFOQueue:
 
         return entries
 
-    def get_remaining_amount(self) -> float:
+    def get_remaining_amount(self) -> Decimal:
         """
         Calculate the total remaining amount in the queue.
         """
-        return sum(trade.amount for trade in self.queue)
+        return sum((trade.amount for trade in self.queue), Decimal(0))