56 lines
1.7 KiB
Python
56 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
|
|
# shelved dictionary cannot be read and written to concurrently (so no '+=' or
|
|
# 'append' operations).
|
|
#
|
|
# The original implementation opened the database with 'writeback' set and
|
|
# called 'sync()' after each write operation. The current implementation does
|
|
# not set 'writeback' and instead requires 'close()' to be called before data
|
|
# is saved. This is the method recommended by the Python reference manual.
|
|
|
|
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()
|