poj_org 0.1.2 → 0.1.3

data/lib/poj_org/problem.rb

@@ -1,18 +1,14 @@
 class PojOrg::Problem
-  def initialize(id)
-    @id = id
+  attr_accessor :id, :title, :time_limit_in_ms, :memory_limit_in_kb
+  def initialize(new_id)
+    self.id = new_id
+    html = open(url).read
+    self.title              = html[/(?<=<div class="ptt" lang="en-US">).+?(?=<\/div>)/]
+    self.time_limit_in_ms   = html[/(?<=<b>Time Limit:<\/b>)\s*\d+/].to_i
+    self.memory_limit_in_kb = html[/(?<=<b>Memory Limit:<\/b>)\s*\d+/].to_i
   end
 
   def url
-    "http://poj.org/problem?id=#{@id}"
-  end
-
-  def details
-    html = open(url).read
-    details = {
-      :title => html[/(?<=<div class="ptt" lang="en-US">).+?(?=<\/div>)/],
-      :time_limit_in_ms => html[/(?<=<b>Time Limit:<\/b>)\s*\d+/].to_i,
-      :memory_limit_in_kb => html[/(?<=<b>Memory Limit:<\/b>)\s*\d+/].to_i
-    }
+    "http://poj.org/problem?id=#{id}"
   end
 end

data/lib/poj_org/source.rb

@@ -2,24 +2,22 @@ require 'open-uri'
 require 'cgi'
 
 class PojOrg::Source
-  def initialize(id)
-    @id = id
-  end
+  attr_accessor :id, :user, :time_in_ms, :memory_in_kb, :language, :source
 
-  def url
-    "http://poj.org/showsource?solution_id=#{@id}"
+  def initialize(new_id)
+    self.id = new_id
+    if PojOrg::authenticated?
+      cookie = PojOrg::login
+      html = open(url, 'Cookie' => cookie).read
+      self.user         = (/<td><b>User:.+?<a.+?>(.+?)<\/a><\/td>/).match(html)[1]
+      self.time_in_ms   = (/<td><b>Time:.+?(\d+)MS<\/td>/).match(html)[1].to_i
+      self.memory_in_kb = (/<td><b>Memory:.+?(\d+)K<\/td>/).match(html)[1].to_i
+      self.language     = (/<td><b>Language:<\/b>\s*(.+?)<\/td>/).match(html)[1]
+      self.source       = CGI.unescapeHTML((/<pre.+?>(.+)<\/pre>/m).match(html)[1])
+    end
   end
 
-  def details
-    return nil unless PojOrg::authenticated?
-    cookie = PojOrg::login
-    html = open(url, 'Cookie' => cookie).read
-    details = {
-      :user         => (/<td><b>User:.+?<a.+?>(.+?)<\/a><\/td>/).match(html)[1],
-      :time_in_ms   => (/<td><b>Time:.+?(\d+)MS<\/td>/).match(html)[1].to_i,
-      :memory_in_kb => (/<td><b>Memory:.+?(\d+)K<\/td>/).match(html)[1].to_i,
-      :language     => (/<td><b>Language:<\/b>\s*(.+?)<\/td>/).match(html)[1],
-      :source       => CGI.unescapeHTML((/<pre.+?>(.+)<\/pre>/m).match(html)[1])
-    }
+  def url
+    "http://poj.org/showsource?solution_id=#{id}"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: poj_org
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
   prerelease: 
 platform: ruby
 authors: