def factory(name, description, percentage, flat):
u = User()
u.username = name
u.password = name
db.session.add(u)
db.session.flush()
b = Balance()
b.type = 'user'
b.user_id = u.id
b.amount = 0
b.currency_id = 1
db.session.add(b)
f = Fee()
f.name = name
f.description = description
f.fractional_pct = percentage
f.flat = flat
f.user_id = u.id
db.session.add(f)
def create_balance(self,
user=None,
campaign=None,
currency_code=None,
amount=None):
amount = amount if amount is not None else Decimal('50.0000')
if not currency_code:
currency_code = 'USD'
currency = Currency.get(currency_code)
b = Balance()
b.enabled = True
b.amount = amount
b.currency = currency
if user is not None:
b.user_id = user.id
b.type = 'user'
elif campaign is not None:
b.campaign_id = campaign.id
b.type = 'campaign'
self.commit_model(b)
return b
