You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
65 lines
1.7 KiB
65 lines
1.7 KiB
class Person
|
|
attr_reader :id, :name, :age, :home_id
|
|
# connect to postgres
|
|
DB = PG.connect(host: "localhost", port: 5432, dbname: 'contacts')
|
|
|
|
def initialize(opts = {})
|
|
@id = opts["id"].to_i
|
|
@name = opts["name"]
|
|
@age = opts["age"].to_i
|
|
if opts["home_id"]
|
|
@home_id = opts["home_id"].to_i
|
|
end
|
|
end
|
|
|
|
def self.all
|
|
results = DB.exec("SELECT * FROM people;")
|
|
return results.map { |result| Person.new(result) }
|
|
end
|
|
|
|
def self.find(id)
|
|
results = DB.exec("SELECT * FROM people WHERE id=#{id};")
|
|
return Person.new(results.first)
|
|
end
|
|
|
|
def self.create(opts={})
|
|
results = DB.exec(
|
|
<<-SQL
|
|
INSERT INTO people (name, age, home_id)
|
|
VALUES ( '#{opts["name"]}', #{opts["age"]}, #{opts["home_id"] ? opts["home_id"] : "NULL"})
|
|
RETURNING id, name, age, home_id;
|
|
SQL
|
|
)
|
|
return Person.new(results.first)
|
|
end
|
|
|
|
def self.delete(id)
|
|
results = DB.exec("DELETE FROM people WHERE id=#{id};")
|
|
return { deleted: true }
|
|
end
|
|
|
|
def self.update(id, opts={})
|
|
results = DB.exec(
|
|
<<-SQL
|
|
UPDATE people
|
|
SET name='#{opts["name"]}', age=#{opts["age"]}
|
|
WHERE id=#{id}
|
|
RETURNING id, name, age;
|
|
SQL
|
|
)
|
|
return Person.new(results.first)
|
|
end
|
|
|
|
def self.addHome(person_id, home_id)
|
|
results = DB.exec(
|
|
<<-SQL
|
|
UPDATE people
|
|
SET home_id = #{home_id}
|
|
WHERE id = #{person_id}
|
|
RETURNING id, name, age, home_id;
|
|
SQL
|
|
)
|
|
return Person.new(results.first)
|
|
end
|
|
end
|