stockfolio 0.1.5 → 0.1.6

data/bin/stockfolio

@@ -1,8 +1,10 @@
 #!/usr/bin/env ruby
 
-$LOAD_PATH << './lib'
+$: << File.dirname($0) + "/../lib"
 
+require 'rubygems'
 require 'stockfolio'
+
 begin
     StockFolio::Runner.start
 rescue Boson::OptionParser::Error => e

data/lib/stockfolio.rb

@@ -5,9 +5,12 @@ require_relative 'stockfolio/portfolio'
 require_relative 'stockfolio/transaction'
 require_relative 'stockfolio/watchlist'
 
+require_relative 'stockfolio/formatters'
+
 module StockFolio
     autoload :Runner, 'stockfolio/runner'
     autoload :Web, 'stockfolio/web'
+    autoload :ConsoleGraph, 'stockfolio/consolegraph'
 end
 
 #DataMapper::Logger.new($stdout, :debug)
@@ -17,6 +20,9 @@ rcfile = ENV['STOCKFOLIO_YML'] || Dir.home + '/.stockfolio.yml'
 if File.exists?(rcfile)
     config = YAML::load(File.open(rcfile))
     ENV['STOCKFOLIO_DB'] = config['db'] || nil
+    if (config[:database]) 
+        ENV['STOCKFOLIO_DB'] = config[:database][:location] || nil
+    end
 end
 
 dbfile = ENV['STOCKFOLIO_DB'] || Dir.home + '/.stockfolio.db'

data/lib/stockfolio/consolegraph.rb

@@ -0,0 +1,52 @@
+module StockFolio
+    class ConsoleGraph
+
+        def initialize(values)
+            @height = 25
+            @width = 120
+
+            min = nil
+            max = nil
+            values.each do |value|
+                value = value.to_f
+                min = nil == min ? value : min
+                max = nil == max ? value : max
+
+                min = value < min ? value : min
+                max = value > max ? value : max
+            end
+
+            heights = []
+            values.each do |value|
+                heights << (value.to_f * @height / max)
+            end
+
+            matrix = []
+            0.upto(@height) do |i|
+                j = 0
+                matrix[i] = []
+                heights.each do |height|
+                    if height >= i
+                        matrix[i][j] = true
+                    else
+                        matrix[i][j] = false
+                    end
+                    j = j + 1
+                end
+            end
+
+
+            # Draw now
+            puts ' ^'
+            matrix.reverse!
+            matrix.each do |row|
+                line = ' |' + row.map { |f| f ? "*" : " " }.join('')
+                puts line
+            end
+            puts ' +' + '-'  * values.length + '>'
+        end
+    end
+end
+
+
+#StockFolio::ConsoleGraph.new([1,2,3,4,5,6,7,8,9,10,9,8,7,7,6,5,4,3,2,1])

data/lib/stockfolio/formatters.rb

@@ -0,0 +1,21 @@
+require 'hirb'
+
+module Hirb::Helpers::Table::Filters
+    def to_dollars(amount)
+        if amount
+            amount = amount.to_f
+            if amount < 0
+                sprintf('($%0.2f)',0-amount).gsub(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1,")
+            else
+                sprintf('$%0.2f',amount).gsub(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1,")
+            end
+        end
+    end
+
+    def to_percent(value)
+        if value
+            "#{(100.0 * value).round(1)}%"
+        end
+    end
+end
+

data/lib/stockfolio/runner.rb

@@ -31,6 +31,22 @@ class StockFolio::Runner < Boson::Runner
     def historical(symbol,start_date,end_date)
         history = StockFolio::Web.historical(symbol, DateTime.parse(start_date), DateTime.parse(end_date))
         puts Hirb::Helpers::Table.render(history, :fields => [:date, :open, :close, :low, :high, :volume])
+        # http://tagaholic.me/hirb/doc/classes/Hirb/Helpers/Table.html#M000008
+        # :headers => { :date => "Date", ... }
+        # :description => false
+        # :filters => { :date => lambda { |f} }}
+        # https://github.com/change/method_profiler/blob/master/lib/method_profiler/hirb.rb
+    end
+
+    def graph_historical(symbol,start_date,end_date)
+        history = StockFolio::Web.historical(symbol, DateTime.parse(start_date), DateTime.parse(end_date))
+
+        values = []
+        history.each do |row|
+            values << row[:close]
+        end
+        values.reverse!
+        StockFolio::ConsoleGraph.new(values)
     end
 
     desc "Search symbol"
@@ -71,8 +87,16 @@ class StockFolio::Runner < Boson::Runner
         portfolios = []
         if nil == name
             portfolios = Portfolio.all
+            if portfolios.size == 0
+                puts "No portfolio found"
+                exit
+            end
         else
             portfolios = Portfolio.all(:name => name)
+            if portfolios.size == 0
+                puts "Portfolio #{name} not found"
+                exit
+            end
         end
 
         positions = {}
@@ -108,28 +132,54 @@ class StockFolio::Runner < Boson::Runner
         end
 
         pos = []
+
+        total = {}
+        total[:symbol] = "Total"
+        total[:daygain] = 0
+        total[:cost] = 0
+        total[:value] = 0
+        total[:gain] = 0
+
         positions.each do |symbol,position|
             p = {}
-            p["Symbol"] = symbol.split(":")[1]
-            p["Last Price"] = "$#{position[:l].to_f.round(2)}"
-            p["Change"] = "#{position[:c]} (#{position[:cp]}%)"
+            p[:symbol] = symbol.split(":")[1]
+            p[:last_price] = position[:l].to_f
+            p[:change] = "#{position[:c]} (#{position[:cp]}%)"
 
             if position[:quantity] > 0
-                p["Day's Gain"] = (position[:quantity] * position[:c].to_f).round(2)
-                p["Shares"] = position[:quantity]
-                p["Cost Basis"] = "$#{position[:cost].round(2)}"
-                p["Market Value"] = "$#{position[:value].round(2)}"
-                p["Gain"] = "$#{(position[:value] - position[:cost]).round(2)}"
-                p["Gain %"] = "#{(100.0 * (position[:value] - position[:cost]) / position[:cost]).round(1)}%"
+                p[:daygain] = position[:quantity] * position[:c].to_f
+                p[:quantity] = position[:quantity]
+                p[:cost] = position[:cost].to_f
+                p[:value] = position[:value].to_f
+                p[:gain] = position[:value] - position[:cost]
+                p[:gain_p] = (position[:value] - position[:cost]) / position[:cost]
+
+                total[:daygain] = total[:daygain] + p[:daygain]
+                total[:cost] = total[:cost] + position[:cost]
+                total[:value] = total[:value] + position[:value]
+                total[:gain] = total[:gain] + position[:value] - position[:cost]
             else
-                p["Gain"] = "$#{(0 - position[:balance]).round(2)}"
-                p["Gain %"] = "#{(100.0 * (0 - position[:balance]) / position[:cost]).round(1)}%"
+                p[:gain] = (0 - position[:balance])
+                p[:gain_p] = (0 - position[:balance]) / position[:cost]
+
             end
             pos << p
         end
 
-        puts Hirb::Helpers::Table.render(pos, :fields => ["Symbol", "Last Price", "Change", "Day's Gain", "Shares", "Cost Basis", "Market Value", "Gain", "Gain %"])
-        
+        pos.sort! { |a,b| a[:symbol] <=> b[:symbol] }
+
+        total[:gain_p] = (total[:value] - total[:cost]) / total[:cost]
+
+        pos << total
+
+        puts Hirb::Helpers::Table.render(pos, 
+            :fields => [:symbol, :last_price, :change, :daygain, :quantity, :cost, :value, :gain, :gain_p],
+            :headers => {:symbol => "Symbol", :last_price => "Last Price", :change => "Change", :daygain => "Day's Gain", :quantity => "Shares", :cost => "Cost Basis", :value => "Market Value", :gain => "Gain", :gain_p => "Gain %"},
+            :filters => { :last_price => :to_dollars, :daygain => :to_dollars, :cost => :to_dollars, :value => :to_dollars, :gain => :to_dollars, :gain_p => :to_percent },
+            :description => false
+
+                                        )
+        puts "Market is #{StockFolio::Web.market_status}"
     end
 
     common_transaction_options
@@ -298,12 +348,38 @@ class StockFolio::Runner < Boson::Runner
     def print_quotes(quote)
         if nil != quote
             # Beautify those
+            extended = false
+            quote.sort! { |a,b| a["t"] <=> b["t"] }
             quote.each do |q|
-                q["Symbol"] = q["t"]
-                q["Last Price"] = "$#{q["l"]}"
-                q["Change"] = "#{q["c"]} (#{q["cp"]}%)"
+                #q["Symbol"] = q["t"]
+                #q["Last Price"] = "$#{q["l"]}"
+
+                #q["t"] = "#{q["e"]}:#{q["t"]}"
+                q[:change] = "#{q["c"]} (#{q["cp"]}%)"
+                if q["el"]
+                    q[:echange] = "#{q["ec"]} (#{q["ecp"]}%)"
+                    extended = true
+                else
+                    q["elt"] = q["lt"]
+                end
             end
-            puts Hirb::Helpers::Table.render(quote, fields: ["Symbol", "Last Price", "Change"])
+            headers = { "t" => "Symbol", "l" => "Last Price", :change => "Change", "lt" => "Last Updated" }
+            
+            fields = ["t", "l", :change, "lt"]
+            filters = { "l" => :to_dollars }
+
+            if extended
+                fields = ["t", "l", :change, "el", :echange, "elt"]
+                headers = { "t" => "Symbol", "l" => "Close Price", :change => "Change", "el" => "After Hours Price", :echange => "After Hours Change", "elt" => "Last Updated" }
+                filters = { "l" => :to_dollars, "el" => :to_dollars }
+            end
+
+            puts Hirb::Helpers::Table.render(quote, 
+                     :fields => fields,
+                     :headers => headers,
+                     :filters => filters,
+                     :description => false
+            )
         end
  
     end

data/lib/stockfolio/version.rb

@@ -1,3 +1,3 @@
 module StockFolio
-    VERSION = '0.1.5'
+    VERSION = '0.1.6'
 end

data/lib/stockfolio/web.rb

@@ -12,7 +12,9 @@ class StockFolio::Web
         if data.empty?
             return nil
         end
-        JSON.parse(data.slice(3, data.length).strip)
+        r = JSON.parse(data.slice(3, data.length).strip)
+        #puts r
+        r
     end
 
     def self.historical(symbol, startDate, endDate)
@@ -59,4 +61,27 @@ class StockFolio::Web
         result["matches"]
     end
 
+    def self.market_status()
+        url = "http://www.nasdaq.com/dynamic_includes/marketstatus.js"
+        resp = Net::HTTP.get_response(URI.parse(url))
+        data = resp.body
+        if data.empty?
+            return false
+        end
+
+        parts = data.split('=')
+        status = parts[1].gsub('"', '').gsub(';', '').strip
+        if status == "O"
+            return "Opened"
+        elsif status == "A"
+            return "After hours"
+        elsif status == "C"
+            return "Closed"
+        else
+            return nil
+        end
+ 
+    end
+
 end
+

metadata

@@ -1,136 +1,138 @@
---- !ruby/object:Gem::Specification 
+--- !ruby/object:Gem::Specification
 name: stockfolio
-version: !ruby/object:Gem::Version 
-  prerelease: false
-  segments: 
-  - 0
-  - 1
-  - 5
-  version: 0.1.5
+version: !ruby/object:Gem::Version
+  version: 0.1.6
+  prerelease: 
 platform: ruby
-authors: 
+authors:
 - Jerome Poichet
 autorequire: 
 bindir: bin
 cert_chain: []
-
-date: 2012-03-22 00:00:00 -07:00
-default_executable: 
-dependencies: 
-- !ruby/object:Gem::Dependency 
+date: 2012-04-09 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
   name: boson
-  prerelease: false
-  requirement: &id001 !ruby/object:Gem::Requirement 
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id001
-- !ruby/object:Gem::Dependency 
-  name: hirb
   prerelease: false
-  requirement: &id002 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: hirb
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id002
-- !ruby/object:Gem::Dependency 
-  name: data_mapper
   prerelease: false
-  requirement: &id003 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: data_mapper
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id003
-- !ruby/object:Gem::Dependency 
-  name: dm-migrations
   prerelease: false
-  requirement: &id004 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: dm-migrations
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id004
-- !ruby/object:Gem::Dependency 
-  name: dm-sqlite-adapter
   prerelease: false
-  requirement: &id005 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: dm-sqlite-adapter
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id005
-- !ruby/object:Gem::Dependency 
-  name: json
   prerelease: false
-  requirement: &id006 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: json
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id006
-- !ruby/object:Gem::Dependency 
-  name: uri
   prerelease: false
-  requirement: &id007 !ruby/object:Gem::Requirement 
+  version_requirements: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
-  type: :runtime
-  version_requirements: *id007
-- !ruby/object:Gem::Dependency 
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
   name: gems
-  prerelease: false
-  requirement: &id008 !ruby/object:Gem::Requirement 
+  requirement: !ruby/object:Gem::Requirement
     none: false
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        segments: 
-        - 0
-        version: "0"
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
   type: :runtime
-  version_requirements: *id008
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 description: 
 email: poitch@gmail.com
-executables: 
+executables:
 - stockfolio
 extensions: []
-
 extra_rdoc_files: []
-
-files: 
+files:
 - lib/stockfolio/check.rb
+- lib/stockfolio/consolegraph.rb
+- lib/stockfolio/formatters.rb
 - lib/stockfolio/portfolio.rb
 - lib/stockfolio/runner.rb
 - lib/stockfolio/transaction.rb
@@ -139,38 +141,29 @@ files:
 - lib/stockfolio/web.rb
 - lib/stockfolio.rb
 - bin/stockfolio
-has_rdoc: true
 homepage: http://github.com/poitch/stockfolio
 licenses: []
-
 post_install_message: 
 rdoc_options: []
-
-require_paths: 
+require_paths:
 - lib
 - lib
-required_ruby_version: !ruby/object:Gem::Requirement 
+required_ruby_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      segments: 
-      - 0
-      version: "0"
-required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
-  requirements: 
-  - - ">="
-    - !ruby/object:Gem::Version 
-      segments: 
-      - 0
-      version: "0"
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
 requirements: []
-
 rubyforge_project: 
-rubygems_version: 1.3.7
+rubygems_version: 1.8.21
 signing_key: 
 specification_version: 3
 summary: Track stock portfolio from the command line
 test_files: []
-