no more withdraw. make negative deposits

llazzaro · Nov 30, 2015 · 4a014db · 4a014db
1 parent a4cd0f1
commit 4a014db
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/pyStock/models/account.py b/pyStock/models/account.py
@@ -116,9 +116,6 @@ def execute(self, order, tick):
     def deposit(self, money):
         money.account = self
 
-    def withdraw(self, money):
-        self.deposit(Money(amount=-1 * money.amount, currency=money.currency))
-
     @hybrid_property
     def total(self):
         """

diff --git a/pyStock/models/events.py b/pyStock/models/events.py
@@ -13,8 +13,8 @@ def validate_buy_order(mapper, connection, target):
     cost = target.share * target.price + target.account.broker.commission(target)
     if cost > target.account.cash[target.security.currency]:
         raise Exception('Transition fails validation: cash {0} is smaller than cost {1}'.format(target.account.cash[target.security.currency], cost))
-    # TODO create the withdraw from account
-    target.account.withdraw(Money(amount=target.share * Decimal(target.price), currency=target.security.exchange.currency))
+
+    target.account.deposit(Money(amount=target.share * -1 * Decimal(target.price), currency=target.security.exchange.currency))
     session = object_session(target)
     position = Position(buy_order=target, stage=OpenPositionStage(), share=target.share, account=target.account)
     session.add(position)