Changeset 1522 for pykota/trunk/pykota/storage.py

Timestamp:

06/06/04 00:03:50 (20 years ago)

Author:

jalet

Message:

Payments history is now stored in database

Files:

• pykota/trunk/pykota/storage.py (modified) (3 diffs)

pykota/trunk/pykota/storage.py

@@ -22 +22 @@
 #
 # $Log$
+# Revision 1.54  2004/06/05 22:03:49  jalet
+# Payments history is now stored in database
+#
 # Revision 1.53  2004/06/03 23:14:10  jalet
 # Now stores the job's size in bytes in the database.
@@ -230 +233 @@
         self.LifeTimePaid = None
         self.Email = None
+        self.Payments = [] # TODO : maybe handle this smartly for SQL, for now just don't retrieve them
         
     def consumeAccountBalance(self, amount) :     
@@ -238 +242 @@
     def setAccountBalance(self, balance, lifetimepaid) :    
         """Sets the user's account balance in case he pays more money."""
-        self.parent.writeUserAccountBalance(self, balance, lifetimepaid)
-        self.AccountBalance = balance
-        self.LifeTimePaid = lifetimepaid
+        diff = float(lifetimepaid or 0.0) - float(self.LifeTimePaid or 0.0)
+        self.parent.beginTransaction()
+        try :
+            self.parent.writeUserAccountBalance(self, balance, lifetimepaid)
+            self.parent.writeNewPayment(self, diff)
+        except PyKotaStorageError, msg :    
+            self.parent.rollbackTransaction()
+            raise PyKotaStorageError, msg
+        else :    
+            self.parent.commitTransaction()
+            self.AccountBalance = balance
+            self.LifeTimePaid = lifetimepaid
         
     def setLimitBy(self, limitby) :