pyradise 0.2.1 → 0.3.0

data/VERSION

@@ -1 +1 @@
-0.2.1
+0.3.0

data/bin/pyradise

@@ -15,7 +15,6 @@ $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
 require 'pyradise'
 include Pyradise
 
-OPTIONS = {}
 MANDATORY_OPTIONS = %w(  )
 
 parser = OptionParser.new do |opts|
@@ -26,11 +25,21 @@ Usage: #{File.basename($0)} command [args]
 
 Options are:
 
-   fetch        - Get fresh info
-   list/search  - List and filter items
-   info/show    - More info about an item
+   fetch         -  Start/refresh database.
+   list/search   -  List and filter items
+     -o --order  -  Order by: name, price, store.
+   info/show     -  More info about an item
+   clear         -  Wipe out the database
+
+Common usage:
+
+   pyradise fetch
+    " list mb i7        - Searches for Intel i7 mobos.
+    " list mem -o name  - Searches for 'mem' ordered by name.
+    " info <sid>        - Show price history of the item.
 
 BANNER
+  opts.on("-o", "--order ORDER", String, "Table order" ) { |order| Options[:order] = order }
   opts.separator ""
   opts.parse!(ARGV)
 end
@@ -39,5 +48,5 @@ if ARGV.empty?
   puts parser.banner
   exit
 else
-  Pyradise.run! ARGV, OPTIONS
+  Pyradise::Cli.run! ARGV
 end

data/lib/pyradise/cli.rb

@@ -0,0 +1,164 @@
+module Pyradise
+  class Cli
+
+    class << self
+      include I18n
+
+      def run! comm
+        if respond_to? comm[0]
+          send(*comm)
+        else
+          puts "Can't do that..."
+          exit
+        end
+      end
+
+      def fetch
+        create from_stores
+      end
+
+      def list(*query)
+        p query
+        t = Time.now
+        w = terminal_size[0]
+        s = w - 36
+        puts "\n#{t(:search)} #{'"' + green(query.join(" ")) + '"' if query[0]}... #{t(:order)}: #{green(Options[:order])}"
+#        puts "_" * w
+        puts
+        prods = Product.filter(:name.like("%#{query.join("%")}%")).order(Options[:order])
+        prods.each_with_index do |prod, i|
+          name = prod.name.length > s ? prod.name[0..s] + ".. " : prod.name + " "
+          for q in query
+            b = i % 2 == 0 ? "\e[0m\e[2m" : "\e[0m\e[0m"
+            replaces = name.scan(/#{q}/i).length
+            name.gsub!(/#{q}/i, "\e[32m\\0#{b}")
+            name += "." until name.length > w - 23
+          end
+          total_price = prod.price * Options[:rate] * Options[:tax]
+          out = sprintf("%-6s | %-6s | %-#{s}s %-3d |  R$ %s", prod.store, prod.sid,  name,  prod.price, yellow(total_price.to_i))
+
+          puts i % 2 == 0 ? bold(out) : out
+        end
+        puts "_" * w
+        puts red("Total: #{prods.all.length} (#{Time.now - t}s)")
+      end
+      alias :search :list
+
+      def info(sid=nil)
+        if !sid
+          puts red("Use: pyradise view <ID>")
+        elsif !prod = Product.filter(:sid => sid.to_i).first
+          puts yellow("Product not found.")
+        else
+          w = terminal_size[0] - 20
+          prices = prod.prices
+          max = prices.values.max.to_f
+          prices.keys.sort.each do |k|
+            rel = "=" * (prices[k]  / max * w)
+            puts "#{Time.at(k).strftime('%M/%d')} #{rel}| #{prices[k]}"
+          end
+        end
+      end
+      alias :show :info
+
+      #
+      # Wipe all data
+      def clear
+        print t(:delete)
+        return puts unless read_char =~ /y/
+        Product.dataset.delete
+        puts "\nDB cleared."
+      end
+
+      private
+
+      def read_char
+        system "stty raw -echo"
+        out = STDIN.getc
+        RUBY_PLATFORM =~ /1.9/ ? out : out.chr
+      ensure
+        system "stty -raw echo"
+      end
+
+      #
+      # Create records from txt files
+      def create txts
+        products = []
+        for txt in txts
+          print "Parsing #{txt[:store]}..."
+          c = Product.all.length
+          parse(txt[:txt], txt[:delimiter]).each do |t|
+            next if t[:price] == 0
+            create_product(t, txt[:store].to_s)
+          end
+          puts "#{Product.all.length - c} #{t(:created)}."
+        end
+      end
+
+      def create_product(t, store)
+        if prod = Product.filter(:name => t[:name], :store => store).first
+          prod.new_price!(t[:price]) if t[:price] != prod.price
+        else
+          Product.create(t.merge(:store => store))
+        end
+      rescue  => e
+        puts "SQLITE Err => #{e}"
+        p t[:name]
+        puts t.inspect
+      end
+
+      def from_stores
+        stores = []
+        for store in YAML.load(File.new(File.dirname(__FILE__) + '/../stores.yml'))[:stores]
+          data = {}
+          data[:store] = store[0]
+          data[:delimiter] = store[1][:delimiter]
+
+          # Open URL
+          next unless data[:txt] = fetch_store(store)
+
+          # Open local file to write
+          open("#{HOME}#{store[0]}-#{Time.now.to_i}.txt", "wb") do |dump|
+            dump.write(data[:txt].gsub(/\t/, ""))
+          end
+          puts "Store #{store[0]} dumped."
+          stores << data
+        end
+        stores
+      end
+
+      def fetch_store(store)
+        open(store[1][:txt]).read
+      rescue => e
+        puts "#{store[0]} offline: #{e}.".capitalize
+        nil
+      end
+
+      def parse(txt, del)
+        products = []
+        txt.each_line do |l|
+          sid, *info = l.split(del)
+          next if info.length < 2
+          price = info.delete_at(-1).strip.to_i
+          next if price.nil? || price.zero?
+          products << { :sid => sid.strip, :name => info.join("").strip.gsub(/\.{2,}/, ""), :price => price }
+        end
+        products
+      end
+
+      # Get terminal size to fit the table nicely.
+      # From highliner.
+      def terminal_size
+        `stty size`.split.map { |x| x.to_i }.reverse
+      end
+
+      def red(txt);      "\e[31m#{txt}\e[0m";    end
+      def green(txt);    "\e[32m#{txt}\e[0m";    end
+      def yellow(txt);   "\e[33m#{txt}\e[0m";    end
+      def bold(txt);      "\e[2m#{txt}\e[0m";    end
+    end
+
+
+
+  end
+end

data/lib/pyradise/i18n.rb

@@ -0,0 +1,28 @@
+module Pyradise
+  module I18n
+    I = {
+    :en_us =>
+      {
+        :search => "Searching for",
+        :order  => "Ordered by",
+        :delete => "About to delete all the records. Sure? [y/N] ",
+        :created => "products created"
+      },
+    :pt_br =>
+      {
+        :search => "Procurando por",
+        :order  => "Ordenado por",
+        :delete => "Deletar todo o banco? [y/N] ",
+        :created => "produtos criados"
+      }
+
+
+
+    }
+
+    def t(text)
+      I[Options[:lang].to_sym][text.to_sym]
+    end
+
+  end
+end

data/lib/pyradise/product.rb

@@ -1,4 +1,5 @@
-class Product < Sequel::Model
+module Pyradise
+  class Product < Sequel::Model
 
   def before_save
     phist = { Time.now.to_i => price }
@@ -13,4 +14,5 @@ class Product < Sequel::Model
     self.update(:prices => Marshal.dump(prices ? prices.merge({Time.now.to_i => np}) : np))
   end
 
+  end
 end

data/lib/pyradise.rb

@@ -1,145 +1,34 @@
-require 'rubygems'
 require 'open-uri'
 require 'sequel'
 
-#TODO: def init
-HOME = ENV['HOME'] + "/.pyradise"
-unless File.exists? HOME
-  FileUtils.mkdir_p HOME
-  conf = open(HOME + "/conf.yml", "wb")
-  conf.write(":rate: 2.0\n:tax: 1.3\n")
-  conf.close
-end
-
-DB = Sequel.connect("sqlite://#{HOME}/py.sqlite3")
-require 'pyradise/product'
-# require 'pyradise/stat'
-
-unless DB.table_exists? :products
-  require 'pyradise/migrate'
-  CreatePyradise.apply DB, :up
-end
-
 module Pyradise
 
-  CONF = YAML.load(File.new(HOME + "/conf.yml"))
-  RATE = CONF[:rate] || 2.0
-  TAX = CONF[:tax] || 1.3
-
-  class << self
-
-    def run! comm, opts
-      if respond_to? comm[0]
-        send(*comm)
-      else
-        puts "Can't do that..."
-        exit
-      end
-    end
-
-    def fetch
-      create from_stores
-    end
-
-    def list(*query)
-      t = Time.now
-      w = terminal_size[0]
-      s = w - 35
-      puts "Searching #{'"' + query[0] + '"' if query[0]}... Order by: #{query[1] || 'name'}"
-      puts "_" * w
-      prods = Product.filter(:name.like("%#{query[0]}%")).order(query[1] ? query[1].to_sym : :name)
-      prods.each_with_index do |prod, i|
-        name = prod.name.length > s ? prod.name[0..s] + ".." : prod.name
-        out = sprintf("%-6s | %-5s | %-#{w-38}s %-3d |  R$ %d", prod.store, prod.sid,  name,  prod.price, prod.price * RATE * TAX)
-        puts i % 2 == 0 ? bold(out) : out
-      end
-      puts "_" * w
-      puts green("Total: #{prods.all.length} (#{Time.now - t}s)")
-    end
-    alias :search :list
-
-    def info(sid=nil)
-      if !sid
-        puts red("Use: pyradise view <ID>")
-      elsif !prod = Product.filter(:sid => sid.to_i).first
-        puts yellow("Product not found.")
-      else
-        w = terminal_size[0] - 20
-        prices = prod.prices
-        max = prices.values.max.to_f
-        prices.keys.sort.each do |k|
-          rel = "=" * (prices[k]  / max * w)
-          puts "#{Time.at(k).strftime('%M/%d')} #{rel}| #{prices[k]}"
-        end
-      end
-    end
-    alias :show :info
-
-    def clear
-      Product.dataset.delete
-    end
-
-    private
-
-    def create txts
-      products = []
-      for txt in txts
-        print "Parsing #{txt[:store]}..."
-        c = Product.all.length
-        parse(txt[:txt], txt[:delimiter]).each do |t|
-          next if t[:price] == 0
-          create_product(t, txt[:store].to_s)
-        end
-        puts "#{Product.all.length - c} products created."
-      end
-    end
-
-    def create_product(t, store)
-      if prod = Product.filter(:name => t[:name], :store => store).first
-        prod.new_price!(t[:price]) if t[:price] != prod.price
-      else
-        Product.create(t.merge(:store => store))
-      end
-      rescue  => e
-      puts "SQLITE Err => #{e}, #{t.inspect} - #{store} #{prod}"
-    end
+  HOME = ENV['HOME'] + "/.pyradise"
+  #TODO: def init
+  DB = Sequel.connect("sqlite://#{HOME}/py.sqlite3")
+  def locale
+    `locale | grep LANG`.scan(/LANG=(.*)\./)[0][0].downcase rescue "en_us"
+  end
 
-    def from_stores
-      stores = []
-      for store in YAML.load(File.new(File.dirname(__FILE__) + '/stores.yml'))[:stores]
-        data = {}
-        data[:store] = store[0]
-        data[:delimiter] = store[1][:delimiter]
-        dump = open("#{HOME}#{store[0]}-#{Time.now.to_i}.txt", "wb")
-        data[:txt] = open(store[1][:txt]).read
-        dump.write(data[:txt])
-        dump.close
-        puts "Store #{store[0]} dumped... "
-        stores << data
-      end
-      stores
-    end
+  unless File.exists? HOME
+    FileUtils.mkdir_p HOME
+    conf = open(HOME + "/conf.yml", "wb")
+    conf.write(":rate: 1.8  # dollar conversion rate\n")
+    conf.write(":tax:  1.3  # transport tax\n")
+    conf.write(":lang: #{locale}  # program language\n")
+    conf.close
+  end
 
-    def parse(txt, del)
-      products = []
-      txt.each_line do |l|
-        sid, *info = l.split(del)
-        next if info.length < 2
-        price = info.delete_at(-1).strip.to_i
-        next if price.nil? || price.zero?
-        products << { :sid => sid.strip, :name => info.join("").strip.gsub(/\.{2,}/, ""), :price => price }
-      end
-      products
-    end
+ # require 'pyradise/stat'
+  unless DB.table_exists? :products
+    require 'pyradise/migrate'
+    CreatePyradise.apply DB, :up
+  end
+  Defaults = {:rate => 1, :tax => 1.3, :order => :price, :lang => :en_us}
+  Options = Defaults.merge(YAML.load(File.new(HOME + "/conf.yml")))
 
-    #from highliner
-    def terminal_size
-      `stty size`.split.map { |x| x.to_i }.reverse
-    end
+  require 'pyradise/product'
+  require 'pyradise/i18n'
+  require 'pyradise/cli'
 
-    def red(txt);      "\e[31m#{txt}\e[0m";    end
-    def green(txt);    "\e[32m#{txt}\e[0m";    end
-    def yellow(txt);   "\e[33m#{txt}\e[0m";    end
-    def bold(txt);      "\e[2m#{txt}\e[0m";    end
-  end
 end

data/pyradise.gemspec

@@ -1,15 +1,15 @@
 # Generated by jeweler
-# DO NOT EDIT THIS FILE
-# Instead, edit Jeweler::Tasks in Rakefile, and run `rake gemspec`
+# DO NOT EDIT THIS FILE DIRECTLY
+# Instead, edit Jeweler::Tasks in Rakefile, and run the gemspec command
 # -*- encoding: utf-8 -*-
 
 Gem::Specification.new do |s|
   s.name = %q{pyradise}
-  s.version = "0.2.1"
+  s.version = "0.3.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Marcos Piccinini"]
-  s.date = %q{2009-10-10}
+  s.date = %q{2009-12-28}
   s.default_executable = %q{pyradise}
   s.email = %q{x@nofxx.com}
   s.executables = ["pyradise"]
@@ -26,6 +26,8 @@ Gem::Specification.new do |s|
      "VERSION",
      "bin/pyradise",
      "lib/pyradise.rb",
+     "lib/pyradise/cli.rb",
+     "lib/pyradise/i18n.rb",
      "lib/pyradise/migrate.rb",
      "lib/pyradise/product.rb",
      "lib/stores.yml",
@@ -59,3 +61,4 @@ Gem::Specification.new do |s|
     s.add_dependency(%q<sequel>, [">= 0"])
   end
 end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: pyradise
 version: !ruby/object:Gem::Version 
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors: 
 - Marcos Piccinini
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-10-10 00:00:00 -03:00
+date: 2009-12-28 00:00:00 -02:00
 default_executable: pyradise
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -40,6 +40,8 @@ files:
 - VERSION
 - bin/pyradise
 - lib/pyradise.rb
+- lib/pyradise/cli.rb
+- lib/pyradise/i18n.rb
 - lib/pyradise/migrate.rb
 - lib/pyradise/product.rb
 - lib/stores.yml