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