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()