drakmail · February 20, 2016 18:34 · Feb 20, 2016
diff --git a/drink_habit.rb b/drink_habit.rb
@@ -0,0 +1,8 @@
+class DrinkHabit < Habit
+  # DB structure
+  # id:integer
+  # title:text
+  # food_amount:integer
+  # drink_time:datetime
+  # type: string
+end
diff --git a/food_habit.rb b/food_habit.rb
@@ -0,0 +1,8 @@
+class FoodHabit < Habit
+  # DB structure
+  # id:integer
+  # title:text
+  # food_amount:integer
+  # drink_time:datetime
+  # type: string
+end
diff --git a/habit.rb b/habit.rb
@@ -0,0 +1,9 @@
+class Habit < ActiveRecord::Base
+  # DB structure
+  # id:integer
+  # title:text
+  # food_amount:integer
+  # drink_time:datetime
+  # type: string
+  belongs_to :user
+end
diff --git a/habits_controller.rb b/habits_controller.rb
@@ -0,0 +1,16 @@
+class HabitsController < ApplicationController
+  def update
+    habit = Habit.find(params[:id])
+    if habit.update habits_params
+      redirect_to habit_path(habit)
+    else
+      render 'show'
+    end
+  end
+
+  protected
+
+  def habits_params
+    params.require(:habit).permit(:id, :title, :food_amount, :drink_time)
+  end
+end
diff --git a/habits_slash_edit.html.erb b/habits_slash_edit.html.erb
@@ -0,0 +1,10 @@
+<%= form_for @habit do |f| %>
+  <%= f.text_field :title %>
+  <%= if @habit.is_a? FoodHabit %> <!-- better to use rendering of partials and helper -->
+    <%= f.text_field :food_amount %>
+  <% end %>
+  <% if @habit.is_a? DrinkHabit %>
+    <%= f.text_field :drink_time %>
+  <% end %>
+  <%= f.submit %>
+<% end %>