def test_create(self):
payment_id = payments.create(str(self.amount),
self.email,
self.method,
self.type,
self.notes,
self.reference)
payment = Payment.get(Payment.id == payment_id)
self.assertNotEqual(payment.id, self.id)
self.assertEqual(payment.amount, self.amount)
self.assertEqual(payment.entity.id, self.entity.id)
self.assertEqual(payment.source, self.method)
self.assertEqual(payment.is_donation, self.is_donation)
self.assertEqual(payment.notes, self.notes)
self.assertEqual(payment.bank_reference, self.reference)
payment.delete_instance()
def transactions():
transactions = Payment.select()
return render_template('admin/transactions.html', transactions=transactions)
database = SqliteDatabase(app.config['DATABASE'], threadlocals=True)
from mhvdb2.admin import admin
app.register_blueprint(admin, url_prefix='/admin')
@app.before_request
def before_request():
g.db = database
g.db.connect()
@app.after_request
def after_request(response):
g.db.close()
return response
import mhvdb2.routes # noqa
from mhvdb2.models import Entity, Payment, User
database.connect()
if not Entity.table_exists():
Entity.create_table()
if not Payment.table_exists():
Payment.create_table()
if not User.table_exists():
User.create_table()
database.close()
def create(amount, email, method, type, notes, reference):
# Create payment
payment = Payment()
payment.time = datetime.now()
payment.entity = Entity.get(Entity.email == email)
payment.amount = amount
payment.source = method
payment.is_donation = type != 0
payment.notes = notes
if method == 0: # Bank transfer
payment.bank_reference = reference
payment.pending = True
payment.save()
