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.
36 lines
1.0 KiB
36 lines
1.0 KiB
class Company
|
|
# 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["industry"]
|
|
end
|
|
|
|
def self.all
|
|
results = DB.exec("SELECT * FROM companies;")
|
|
return results.map { |result| Company.new(result) }
|
|
end
|
|
|
|
def self.find(id)
|
|
results = DB.exec("SELECT * FROM companies WHERE id=#{id};")
|
|
return Company.new(results.first)
|
|
end
|
|
|
|
def self.create(opts={})
|
|
results = DB.exec("INSERT INTO companies (name, industry) VALUES ( '#{opts["name"]}', '#{opts["industry"]}' );")
|
|
return { created:true }
|
|
end
|
|
|
|
def self.delete(id)
|
|
results = DB.exec("DELETE FROM companies WHERE id=#{id};")
|
|
return { deleted: true }
|
|
end
|
|
|
|
def self.update(id, opts={})
|
|
results = DB.exec("UPDATE companies SET name='#{opts["name"]}', industry='#{opts["industry"]}' WHERE id=#{id} ;")
|
|
return { updated: true }
|
|
end
|
|
end
|