Record treatments

6 files changed, 39 insertions, 1 deletions

diff --git a/rvh/app/models/employee.rb b/rvh/app/models/employee.rb
index d5e0233..b9571d1 100644
--- a/rvh/app/models/employee.rb
+++ b/rvh/app/models/employee.rb
@@ -1,2 +1,4 @@
 class Employee < ApplicationRecord
+  belongs_to :account
+  has_many :treatments
 end
diff --git a/rvh/app/models/patient.rb b/rvh/app/models/patient.rb
index 7e2676f..05bf8e9 100644
--- a/rvh/app/models/patient.rb
+++ b/rvh/app/models/patient.rb
@@ -1,4 +1,5 @@
 class Patient < ApplicationRecord
   belongs_to :account
   has_one :bed
+  has_many :treatments
 end
diff --git a/rvh/app/models/treatment.rb b/rvh/app/models/treatment.rb
new file mode 100644
index 0000000..8c28da1
--- /dev/null
+++ b/rvh/app/models/treatment.rb
@@ -0,0 +1,4 @@
+class Treatment < ApplicationRecord
+  belongs_to :employee
+  belongs_to :patient
+end
diff --git a/rvh/db/migrate/20200315181211_create_employees.rb b/rvh/db/migrate/20200315181211_create_employees.rb
index 4fcbd76..8e96348 100644
--- a/rvh/db/migrate/20200315181211_create_employees.rb
+++ b/rvh/db/migrate/20200315181211_create_employees.rb
@@ -1,6 +1,7 @@
 class CreateEmployees < ActiveRecord::Migration[6.0]
   def change
     create_table :employees do |t|
+      t.references :account, null: false, foreign_key: true
       t.string :type
       t.datetime :hired_at
 
diff --git a/rvh/db/migrate/20200315183345_create_treatments.rb b/rvh/db/migrate/20200315183345_create_treatments.rb
new file mode 100644
index 0000000..860f0d3
--- /dev/null
+++ b/rvh/db/migrate/20200315183345_create_treatments.rb
@@ -0,0 +1,13 @@
+class CreateTreatments < ActiveRecord::Migration[6.0]
+  def change
+    create_table :treatments do |t|
+      t.references :employee, null: false, foreign_key: true
+      t.references :patient, null: false, foreign_key: true
+      t.string :name, null: false
+      t.datetime :occurred_at, null: false
+      t.text :results
+
+      t.timestamps
+    end
+  end
+end
diff --git a/rvh/db/schema.rb b/rvh/db/schema.rb
index 450c169..55cd78a 100644
--- a/rvh/db/schema.rb
+++ b/rvh/db/schema.rb
@@ -10,7 +10,7 @@
 #
 # It's strongly recommended that you check this file into your version control system.
 
-ActiveRecord::Schema.define(version: 2020_03_15_182810) do
+ActiveRecord::Schema.define(version: 2020_03_15_183345) do
 
   create_table "accounts", force: :cascade do |t|
     t.string "name", null: false
@@ -42,10 +42,12 @@ ActiveRecord::Schema.define(version: 2020_03_15_182810) do
   end
 
   create_table "employees", force: :cascade do |t|
+    t.integer "account_id", null: false
     t.string "type"
     t.datetime "hired_at"
     t.datetime "created_at", precision: 6, null: false
     t.datetime "updated_at", precision: 6, null: false
+    t.index ["account_id"], name: "index_employees_on_account_id"
     t.index ["type"], name: "index_employees_on_type"
   end
 
@@ -68,6 +70,18 @@ ActiveRecord::Schema.define(version: 2020_03_15_182810) do
     t.index ["employee_id"], name: "index_shifts_on_employee_id"
   end
 
+  create_table "treatments", force: :cascade do |t|
+    t.integer "employee_id", null: false
+    t.integer "patient_id", null: false
+    t.string "name", null: false
+    t.datetime "occurred_at", null: false
+    t.text "results"
+    t.datetime "created_at", precision: 6, null: false
+    t.datetime "updated_at", precision: 6, null: false
+    t.index ["employee_id"], name: "index_treatments_on_employee_id"
+    t.index ["patient_id"], name: "index_treatments_on_patient_id"
+  end
+
   create_table "visits", force: :cascade do |t|
     t.integer "patient_id", null: false
     t.datetime "scheduled_at", null: false
@@ -79,8 +93,11 @@ ActiveRecord::Schema.define(version: 2020_03_15_182810) do
 
   add_foreign_key "beds", "care_centres"
   add_foreign_key "beds", "patients"
+  add_foreign_key "employees", "accounts"
   add_foreign_key "patients", "accounts"
   add_foreign_key "shifts", "care_centres"
   add_foreign_key "shifts", "employees"
+  add_foreign_key "treatments", "employees"
+  add_foreign_key "treatments", "patients"
   add_foreign_key "visits", "patients"
 end