From b0228cb901b34ac88c7a404e29164b70511518d2 Mon Sep 17 00:00:00 2001
From: cxt <853663049@qq.com>
Date: Thu, 2 Jan 2020 10:29:22 +0800
Subject: [PATCH] =?UTF-8?q?=E8=B0=83=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/models/hack.rb      | 2 ++
 app/models/item_bank.rb | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/app/models/hack.rb b/app/models/hack.rb
index 618d80d28..fc7a3cff7 100644
--- a/app/models/hack.rb
+++ b/app/models/hack.rb
@@ -21,6 +21,8 @@ class Hack < ApplicationRecord
   belongs_to :user
   belongs_to :sub_discipline
 
+  has_one :item_bank, as: :container, dependent: :destroy
+
   scope :published, -> { where(status: 1) }
   scope :unpublish, -> { where(status: 0) }
   scope :opening, -> {where(open_or_not: 1)}
diff --git a/app/models/item_bank.rb b/app/models/item_bank.rb
index b24d247a9..ff5fb6d42 100644
--- a/app/models/item_bank.rb
+++ b/app/models/item_bank.rb
@@ -12,6 +12,8 @@ class ItemBank < ApplicationRecord
   has_many :tag_discipline_containers, as: :container, dependent: :destroy
   has_many :tag_disciplines, through: :tag_discipline_containers
 
+  belongs_to :container, polymorphic: true, optional: true
+
   def analysis
     item_analysis&.analysis
   end