From 0f692554226e77f03f841d8d75f12bd34703a52b Mon Sep 17 00:00:00 2001
From: wyx <414722560@qq.com>
Date: Sat, 25 Jul 2020 12:12:34 +0800
Subject: [PATCH] update model wallet

---
 app/models/wallet.rb | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/app/models/wallet.rb b/app/models/wallet.rb
index 987db9e8..e7e78671 100644
--- a/app/models/wallet.rb
+++ b/app/models/wallet.rb
@@ -1,2 +1,6 @@
 class Wallet < ApplicationRecord
+  belongs_to :user
+  has_many :outcome, class_name: 'CoinChange', foreign_key: 'from_wallet_id', dependent: :destroy
+  has_many :income, class_name: 'CoinChange', foreign_key: 'to_wallet_id', dependent: :destroy
+  validates :balance, presence: true
 end