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
