def create_wallet(user, network, address, balance=None): wallet = Wallet(user_id=user.id, network=network, address=address) if balance: wallet.balance = balance db.session.add(wallet) db.session.commit() return wallet
def get_or_create_wallet(user): try: wallet = Wallet.objects.get(user=user) except Wallet.DoesNotExist: # creating new wallet wallet = Wallet() wallet.user = user wallet.local_id = random.randint(1000000, 9999999) wallet.title = 'DUCK.{}'.format(wallet.local_id) wallet.display_name = None wallet.balance = 0 wallet.save() # print("New wallet id =", wallet.pk) return wallet