58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
# CFBank.py -- CFBank class
|
|
# Created by: Joris Bontje <jbontje@suespammers.org>
|
|
|
|
# CFBank uses the 'shelve' Python library to store persistent data. The shelve
|
|
# is opened for R/W operations by default, but cannot be read from and written
|
|
# to simultaneously (so no 'append' or '+=' operations).
|
|
#
|
|
# In a past implementation, the bank database was opened with 'writeback' set
|
|
# to 'True' and called sync() after each write operation, but still suffered
|
|
# from strange bank problems.
|
|
#
|
|
# In its current implementation, close() MUST be called to preserve data
|
|
# across runs. This fixed the aforementioned bug for me.
|
|
|
|
import os.path
|
|
import shelve
|
|
|
|
import Crossfire
|
|
|
|
class CFBank:
|
|
def __init__(self, bankfile):
|
|
self.bankdb_file = os.path.join(Crossfire.LocalDirectory(),
|
|
bankfile)
|
|
self.bankdb = shelve.open(self.bankdb_file)
|
|
|
|
def deposit(self, user, amount):
|
|
if not user in self.bankdb:
|
|
self.bankdb[user] = amount
|
|
else:
|
|
balance = self.bankdb[user]
|
|
self.bankdb[user] = balance + amount
|
|
|
|
def withdraw(self, user, amount):
|
|
if user in self.bankdb:
|
|
if self.bankdb[user] >= amount:
|
|
balance = self.bankdb[user]
|
|
self.bankdb[user] = balance - amount
|
|
return 1
|
|
return 0
|
|
|
|
def getbalance(self, user):
|
|
if user in self.bankdb:
|
|
return self.bankdb[user]
|
|
else:
|
|
return 0
|
|
|
|
def remove_account(self, user):
|
|
if user in self.bankdb:
|
|
del self.bankdb[user]
|
|
Crossfire.Log(Crossfire.LogDebug,
|
|
"%s's bank account removed." % user)
|
|
return 1
|
|
else:
|
|
return 0
|
|
|
|
def close(self):
|
|
self.bankdb.close()
|