diff --git a/app/comment_serializer.rb b/app/comment_serializer.rb index d27c604..353061e 100644 --- a/app/comment_serializer.rb +++ b/app/comment_serializer.rb @@ -1,6 +1,6 @@ require_relative "serializer" class CommentSerializer < Serializer - attribute :id - attribute :body + attribute :id + attribute :body end diff --git a/app/serializer.rb b/app/serializer.rb index 258be50..7549506 100644 --- a/app/serializer.rb +++ b/app/serializer.rb @@ -1,2 +1,17 @@ + class Serializer -end + + attr_accessor :values + def initialize(attr) + self.values = attr + end + def self.attribute(*args) + @attr = args unless args.empty? + end + def serialize + values.date = values.date.strftime("%d-%m-%Y") unless values.instance_of?(Comment) + hash_values = values.to_h + hash_values.delete(:title) unless values.instance_of?(Post) + hash_values + end +end \ No newline at end of file