toychooser 1.0.0

data/bin/toychooser

@@ -0,0 +1,53 @@
+#!/usr/bin/env ruby
+
+=begin
+  Copyright 2014 Maxine Red <maxine@furfind.net>
+
+  This file is part of toy-chooser.
+
+  toy-chooser is free software: you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  toy-chooser is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with toy-chooser.  If not, see <http://www.gnu.org/licenses/>.
+=end
+
+$:.unshift File.dirname(__FILE__)+"/../lib"
+require "toychooser"
+
+commands = ["add","remove","update","list","suggest", "show"]
+
+args = ARGV
+options = args.grep(/^-\w$/)
+options.each do |opt|
+  case opt
+  when "-h" then
+    puts "usage:",
+      "toychooser -[h|v|V] {#{commands.join("|")}} [TOYDATA]",
+      "-h\t\t\tDisplay this help message",
+      "-v\t\t\tIncrease verbosity. Can be used more than once.",
+      "-V\t\t\tDisplay version information."
+    exit
+  when "-v" then 
+  when "-V" then 
+    puts "#{Toychooser::Name} v#{Toychooser::Version}"
+    exit
+  else
+    $stderr.puts "Option #{opt} unkown."
+    abort
+  end
+end
+
+command = args.grep(/^(#{commands.join("|")})$/).first.to_sym
+
+args.reject!{|x|x.match(/^-\w|(#{commands.join("|")})$/)}
+
+chooser = Chooser.new
+chooser.__send__(command,args)

data/lib/chooser.rb

@@ -0,0 +1,191 @@
+=begin
+  Copyright 2014 Maxine Red <maxine@furfind.net>
+
+  This file is part of toy-chooser.
+
+  toy-chooser is free software: you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  toy-chooser is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with toy-chooser.  If not, see <http://www.gnu.org/licenses/>.
+=end
+
+require "sqlite3"
+require "net/http"
+require "json"
+require "nokogiri"
+
+require "manufacturer/bad-dragon"
+
+class Chooser
+  def initialize
+    @db_file = File.expand_path("~/.toys.sqlite")
+    if !File.exist?(@db_file) then
+      database_create
+    end
+  end
+  # Update database for given manufacturer.
+  def update(manu)
+    case manu.first
+    when "bad-dragon" then fetch_bd
+    else
+      $stderr.puts "Manufacturer unkown.","Known manufacturers are: bad-dragon"
+      abort
+    end
+  end
+  # Add a toy to the 'owned' list. Toys are in the format sku/size.
+  def add(toys)
+    toys.each do |toy|
+      sku, size = toy.split("/")
+      sku, size = SQLite3::Database.quote(sku), SQLite3::Database.quote(size.upcase)
+      SQLite3::Database.new(@db_file) do |db|
+        query   = "SELECT id, name FROM toys WHERE sku = '#{sku}'"
+        query  += " AND size = '#{size}';"
+        id, name = db.execute(query).flatten
+        if !id then
+          $stderr.puts "Unknown toy or size \"#{toy}\"."
+          next
+        end
+        query  = "INSERT INTO owned (sku, size, toy_id) VALUES"
+        query += " ('#{sku}','#{size}',#{id});"
+        db.execute(query)
+        puts "Added \"#{name}\" (#{size}) to owned list."
+      end
+    end
+  end
+  # Show measurements of toys.
+  def show(toys)
+    toys.each do |toy|
+      sku, size = toy.split("/")
+      sku, size = SQLite3::Database.quote(sku), SQLite3::Database.quote(size.to_s.upcase)
+      SQLite3::Database.new(@db_file) do |db|
+        query  = "SELECT sku, size, price, dia_head, dia_shaft, dia_knot, "
+        query += "circ_head, circ_shaft, circ_knot, usable_length, "
+        query += "total_length FROM toys WHERE sku = '#{sku}';"
+        draw_show(db.execute(query))
+      end
+    end
+  end
+  # Suggests toys for future purchases guessed from what is owned already.
+  def suggest(noargs=[])
+    i, sizes = 0, Array.new(7) {0}
+    ranges = [0.5, 0.4, 0.3, 0.6, 0.5, 0.4, 0.9]
+    cols = ["dia_head","dia_shaft","dia_knot","circ_head","circ_shaft",
+            "circ_knot","usable_length"]
+    SQLite3::Database.new(@db_file) do |db|
+      query  = "SELECT #{cols.join(", ")} FROM toys, owned "
+      query += "WHERE toys.id = owned.toy_id;"
+      db.execute(query) do |row|
+        row.each_with_index do |c,id|
+          sizes[id] += c.to_i
+        end
+        i += 1
+      end
+      sizes = sizes.map.with_index do |x,id|
+        avg = (x/i.to_f).round.to_i
+        line  = "((#{cols[id]} >= #{(avg*(1-ranges[id])).round.to_i} "
+        line += "AND #{cols[id]} <= #{(avg*(1+ranges[id])).round.to_i}) "
+        line += "OR #{cols[id]} IS NULL)"
+        line
+      end
+      query  = "SELECT sku, size FROM toys WHERE #{sizes.join(" AND ")};"
+      draw_list(db.execute(query))
+    end
+  end
+  # List existing toys.
+  def list(toy_list="toys")
+    toy_list = toy_list.first
+    toy_list = "toys" unless toy_list
+    if !["toys","owned"].include?(toy_list) then
+      $stderr.puts "No such list."
+      abort
+    end
+    SQLite3::Database.new(@db_file) do |db|
+      draw_list(db.execute("SELECT sku, size FROM #{toy_list};"))
+    end
+  end
+  private
+  def draw_list(sql_list)
+    toys = Hash.new
+    sizes = ["XS","S ","M ","L ","XL","O "]
+    deli = "+-"+"-"*15+"-+"+"----+"*sizes.count+""+"\n"
+    print deli
+    puts "| #{" "*15} | #{sizes.join(" | ")} |"
+    print deli
+    sql_list.each do |t|
+      if !toys.has_key?(t[0]) then
+        toys.store(t[0],[t[1]])
+      else
+        toys[t[0]] << t[1]
+      end
+    end
+    toys.each do |k,v|
+      cols = sizes.map do |c|
+        v.include?(c.sub(" ","")) ? c : "  "
+      end
+      puts "| #{k.pad(15)} | #{cols.join(" | ")} |"
+    end
+    print deli
+  end
+  def draw_show(sql_list)
+    toys = Hash.new
+    sizes = ["  XS   ","   S   ","   M   ","   L   ","  XL   ","   O   "]
+    measurements = ["Price", "Dia. of head", "Dia. of shaft",
+                    "Dia. of speccial", "Circ. of head",
+                    "Circ. of shaft", "Circ. of special",
+                    "Usable length", "Total length"]
+    deli = "+-"+"-"*15+"-+"+"---------+"*sizes.count+""+"\n"
+    print deli
+    puts "| #{sql_list.first.first.pad(15)} | #{sizes.join(" | ")} |"
+    print deli
+    sql_list.map{|c|c.shift;c}.each do |t|
+      toys.store(t.shift,t)
+    end
+    measurements.each_with_index do |m,id|
+      cols = sizes.map do |c|
+        c.gsub!(" ","")
+        toys[c] ? toys[c][id].to_i : 0
+      end
+      if m == "Price" then
+        cols.map!{|c|c.pad(4," ").sub(/\s(\d)/,"$\\1")+".00"}
+      else
+        cols.map!{|c|(c/100.0).pad(7," ")}
+      end
+      puts "| #{m.pad(15)} | #{cols.join(" | ")} |"
+      print deli if m == "Price"
+    end
+    print deli
+  end
+  def database_create
+    SQLite3::Database.new(@db_file) do |db|
+      db.execute("CREATE TABLE toys (
+                 id INTEGER PRIMARY KEY,
+                 sku VARCHAR(127),
+                 name VARCHAR(255),
+                 size VARCHAR(3),
+                 price INTEGER,
+                 dia_head INTEGER,
+                 dia_shaft INTEGER,
+                 dia_knot INTEGER,
+                 circ_head INTEGER,
+                 circ_shaft INTEGER,
+                 circ_knot INTEGER,
+                 usable_length INTEGER,
+                 total_length INTEGER
+                );")
+      db.execute("CREATE TABLE owned (
+                 id INTEGER PRIMARY KEY,
+                 sku VARCHAR(127),
+                 size VARCHAR(3),
+                 toy_id INTEGER REFERENCES toys(id)
+                 );")
+    end
+  end
+end

data/lib/manufacturer/bad-dragon.rb

@@ -0,0 +1,88 @@
+=begin
+  Copyright 2014 Maxine Red <maxine@furfind.net>
+
+  This file is part of toy-chooser.
+
+  toy-chooser is free software: you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  toy-chooser is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with toy-chooser.  If not, see <http://www.gnu.org/licenses/>.
+=end
+
+class Chooser
+  def fetch_bd
+    net = Net::HTTP.new("bad-dragon.com",80)
+    toys = JSON.parser.new(net.get("/products/getproducts").body).parse
+    toys.reject!{|c|c["cat_id"].to_i!=1}
+    toys.map!{|c|c["link_url"].sub(/.+bad-dragon.com/,"")}
+    toys.each do |t|
+      body = net.get(t).body
+      File.open("cole.html","w"){|f|f.print body}
+      html = Nokogiri::HTML.parse(body)
+      id = html.css("span[itemprop=productID]").first.text
+      name = html.css("span[itemprop=model]").first.text
+      sizes = html.css("table.divided th").map{|th|th.text}
+      sizes.shift # Just drop the first item.
+      table = html.css("table.divided tr").map do |tr|
+        tr.css("td").map{|td|td.text}
+      end
+      table.shift
+      sql_query = "INSERT INTO toys";
+      sizes.count.times do |i|
+        size = sizes[i][0,1].sub("E","XL")
+        size = "XS" if sizes[i] == "Mini"
+        query = Hash.new
+        query.store("sku",id)
+        query.store("name",SQLite3::Database.quote(name))
+        query.store("size",size)
+        price = html.css("#size_inputs label").map do |x|
+          if x.text.match(/^#{sizes[i]}/i) then
+            x.css("i").text.sub("$","").to_i
+          end
+        end.compact.first
+        price = html.css("span[itemprop=price]").first.text.to_i if !price
+        query.store("price",price)
+        table.each do |t|
+          next if t[0].match(/thinnest|smallest/i)
+          # We look only for biggest measures.
+          col = t[0].downcase.sub(/meter of /,"_").sub(/umference of /,"_")
+          col.gsub!(" ","_")
+          col.gsub!(/_\(.+\)/,"")
+          inches = (t[i+1].to_f*100).round.to_i
+          query.store(col,inches)
+        end
+        if i == 0 then
+          sql_query += " (#{query.keys.join(",")}) VALUES\n"
+        end
+        sql_query += " ('#{query.values.join("','")}'),\n"
+      end
+      sql_query[-2,2] = ";"
+      SQLite3::Database.new(@db_file) do |db|
+        begin
+          db.execute(sql_query)
+        rescue
+          p sql_query
+          raise
+        end
+      end
+      puts "Updated \"#{name}\" successfully."
+    end
+  end
+  def remove_aliases(column)
+    column = column.sub("usuable","usable").sub("avg._","")
+    column = column.sub(/(thickest|largest)_part_of_/,"")
+    column = column.sub(/_(base)/,"")
+    # Other common aliases
+    column = column.sub("tip","head").sub("middle","shaft","bottom","knot")
+    column = column.sub("shaft_ridges","knot").sub("medial_ring","knot")
+    column = column.sub("bulb","knot")
+  end
+end

data/lib/standard/int.rb

@@ -0,0 +1,31 @@
+=begin
+  Copyright 2014 Maxine Red <maxine@furfind.net>
+
+  This file is part of toy-chooser.
+
+  toy-chooser is free software: you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  toy-chooser is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with toy-chooser.  If not, see <http://www.gnu.org/licenses/>.
+=end
+
+class Float
+  def pad(c,s=0)
+    float = self.to_s
+    float.sub!(/\.\d$/,"\\00")
+    "#{s}"*(c-float.length)+float
+  end
+end
+class Fixnum
+  def pad(c,s=0)
+    "#{s}"*(c-self.to_s.length)+self.to_s
+  end
+end

data/lib/standard/string.rb

@@ -0,0 +1,87 @@
+=begin
+  Copyright 2014 Maxine Red <maxine_red1@yahoo.com>
+
+  This file is part of rubyhexagon.
+
+  rubyhexagon is free software: you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  rubyhexagon is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with rubyhexagon.  If not, see <http://www.gnu.org/licenses/>.
+=end
+
+require "json"
+
+class String
+  # Make json parsing a lot easier and less to write.
+  def parse
+    JSON.parser.new(self).parse
+  end
+  def to_ascii
+    self.gsub("_"," ").encode("us-ascii", :invalid => :replace, :undef => :replace, :replace => "")
+  end
+  # Remove all tags inside of a string.
+  def clean
+    self.gsub(/<.+?>/,"")
+  end
+  # Trim or expand a string to a certain length.
+  def pad(c,s=" ")
+    if self.length > c then
+      self[0,c-3]+"..."
+    elsif self.length < c then
+      self+s.to_s*(c-self.length)
+    else
+      self
+    end
+  end
+  # Make a string bold and colorful, with colors a user can understand. Not
+  # those cryptic numbers.
+  def bold(color=nil)
+    c = case color
+    when "black"  then  "\e[30;1m"
+    when "red"    then  "\e[31;1m"
+    when "green"  then  "\e[32;1m"
+    when "yellow" then  "\e[33;1m"
+    when "blue"   then  "\e[34;1m"
+    when "purple" then  "\e[35;1m"
+    when "cyan"   then  "\e[36;1m"
+    else                "\e[37;1m"
+    end
+    c+self+"\e[0m"
+  end
+  # Same as bold, but just make a string colorful.
+  def color(color=nil)
+    c = case color
+    when "black"  then  "\e[30;0m"
+    when "red"    then  "\e[31;0m"
+    when "green"  then  "\e[32;0m"
+    when "yellow" then  "\e[33;0m"
+    when "blue"   then  "\e[34;0m"
+    when "purple" then  "\e[35;0m"
+    when "cyan"   then  "\e[36;0m"
+    else                "\e[37;0m"
+    end
+    c+self+"\e[0m"
+  end
+  # indent a string and all lines that belong to it
+  def indent(indent)
+    s = self.split("#$/")
+    " "*indent+s.join("#$/"+" "*indent)
+  end
+  # It is just convenient to have a custom to_bool function. Only "true" gets
+  # turned into true, anything else is false.
+  def to_bool
+    if self.downcase == "true" then
+      true
+    else
+      false
+    end
+  end
+end

data/lib/toychooser.rb

@@ -0,0 +1,27 @@
+=begin
+  Copyright 2014 Maxine Red <maxine@furfind.net>
+
+  This file is part of toy-chooser.
+
+  toy-chooser is free software: you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation, either version 3 of the License, or
+  (at your option) any later version.
+
+  toy-chooser is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with toy-chooser.  If not, see <http://www.gnu.org/licenses/>.
+=end
+
+require "standard/string"
+require "standard/int"
+require "chooser.rb"
+
+module Toychooser
+  Name = "toychooser"
+  Version = "1.0.0"
+end

metadata

@@ -0,0 +1,69 @@
+--- !ruby/object:Gem::Specification
+name: toychooser
+version: !ruby/object:Gem::Version
+  version: 1.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Maxine Red
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-06-11 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: pg
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.18'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '0.18'
+description: A simple little script that takes in data from toy manufacturers, your
+  current toy collection and suggests new toys for you.
+email: maxine@furfind.net
+executables:
+- toychooser
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/toychooser.rb
+- lib/standard/int.rb
+- lib/standard/string.rb
+- lib/chooser.rb
+- lib/manufacturer/bad-dragon.rb
+- bin/toychooser
+homepage: https://github.com/maxine-red/toy-chooser
+licenses:
+- GPL-3.0
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.2
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.23
+signing_key: 
+specification_version: 3
+summary: A toy chooser for adult toy manufacturers
+test_files: []