stockfolio 0.1.1

data/bin/stockfolio

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+$LOAD_PATH << './lib'
+
+require 'stockfolio'
+StockFolio::Runner.start

data/lib/stockfolio.rb

@@ -0,0 +1,30 @@
+require 'data_mapper' # requires all the gems listed above
+require  'dm-migrations'
+
+require_relative 'stockfolio/portfolio'
+require_relative 'stockfolio/transaction'
+require_relative 'stockfolio/watchlist'
+
+module StockFolio
+    autoload :Runner, 'stockfolio/runner'
+    autoload :Web, 'stockfolio/web'
+end
+
+#DataMapper::Logger.new($stdout, :debug)
+
+rcfile = ENV['STOCKFOLIO_YML'] || Dir.home + '/.stockfolio.yml'
+
+if File.exists?(rcfile)
+    config = YAML::load(File.open(rcfile))
+    ENV['STOCKFOLIO_DB'] = config['db'] || nil
+end
+
+dbfile = ENV['STOCKFOLIO_DB'] || Dir.home + '/.stockfolio.db'
+dbfile = File.expand_path(dbfile)
+
+# A Sqlite3 connection to a persistent database
+DataMapper.setup(:default, "sqlite://#{dbfile}")
+
+DataMapper.finalize
+DataMapper.auto_upgrade!
+

data/lib/stockfolio/portfolio.rb

@@ -0,0 +1,10 @@
+class Portfolio
+    include DataMapper::Resource
+
+    property :id,         Serial 
+    property :name,       String
+    property :created_at, DateTime
+
+    has n, :transactions
+end
+

data/lib/stockfolio/runner.rb

@@ -0,0 +1,194 @@
+require 'boson/runner'
+require 'hirb'
+
+class StockFolio::Runner < Boson::Runner
+
+    def self.common_transaction_options
+        option :portfolio, :type => :string, :desc => 'Portfolio name', :required => true
+        option :symbol, :type=>:string, :desc => 'Symbol of stock', :required => true
+        option :price, :type => :numeric, :desc => 'Price paid', :required => true
+        option :quantity, :type => :numeric, :desc => 'Quantity', :required => true
+        option :fee, :type => :numeric, :desc => 'Transaction fee'
+        option :date, :type => :string, :desc => 'Transaction date'
+    end
+
+    def quote(symbol)
+        quote = StockFolio::Web.quote(symbol)
+        print_quotes(quote)
+    end
+
+    desc 'List porfolios'
+    def portfolios
+        q = Portfolio.all
+        portfolios = []
+        q.each do |p|
+            portfolios << {
+                :name => p.name,
+                :created_at => p.created_at
+            }
+        end
+
+        puts Hirb::Helpers::Table.render(portfolios, :fields => [:name, :created_at])
+    end
+
+    desc 'Current positions'
+    def positions(name = nil)
+        portfolios = []
+        if nil == name
+            portfolios = Portfolio.all
+        else
+            portfolios = Portfolio.all(:name => name)
+        end
+
+        positions = {}
+        portfolios.each do |portfolio|
+            #puts "Positions for #{portfolio.name}"
+            portfolio.transactions.each do |transaction|
+                #puts "#{transaction.order_name} #{transaction.quantity} #{transaction.symbol}"
+                if positions[transaction.symbol]
+                    positions[transaction.symbol][:quantity] = positions[transaction.symbol][:quantity] + transaction.quantity
+                    #if transaction.quantity > 0
+                        positions[transaction.symbol][:cost] = positions[transaction.symbol][:cost] + transaction.quantity * transaction.price + transaction.real_fee
+                    #end
+                    positions[transaction.symbol][:balance] = positions[transaction.symbol][:balance] + transaction.quantity * transaction.price + transaction.real_fee
+                else
+                    positions[transaction.symbol] = {
+                        :symbol => transaction.symbol,
+                        :cost => transaction.quantity * transaction.price + transaction.real_fee,
+                        :balance => transaction.quantity * transaction.price,
+                        :quantity => transaction.quantity
+                    }
+                end
+            end
+        end
+
+        # Get current prices
+        quotes = StockFolio::Web.quote(positions.keys.join(","))
+        quotes.each do |q|
+            symbol = "#{q["e"]}:#{q["t"]}"
+            positions[symbol][:l] = q["l"]
+            positions[symbol][:c] = q["c"]
+            positions[symbol][:cp] = q["cp"]
+            positions[symbol][:value] = positions[symbol][:quantity] * q["l"].to_f
+        end
+
+        pos = []
+        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]}%)"
+
+            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)}%"
+            else
+                p["Gain"] = "$#{(0 - position[:balance]).round(2)}"
+                p["Gain %"] = "#{(100.0 * (0 - position[:balance]) / position[:cost]).round(1)}%"
+            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 %"])
+        
+    end
+
+    common_transaction_options
+    desc 'Add buy transaction'
+    def buy(options={})
+        transaction = transaction_from_options(options)
+    end
+
+    common_transaction_options
+    desc 'Add sell transaction'
+    def sell(options={})
+        options[:quantity] = 0 - options[:quantity]
+        transaction = transaction_from_options(options)
+    end
+
+    desc 'Create Portfolio'
+    def create(name)
+        Portfolio.create(:name => name, :created_at => DateTime.now)
+    end
+
+    desc 'Add to watchlist'
+    def watch(symbol)
+        # Validate symbol
+        quote = StockFolio::Web.quote(symbol)
+        if nil != quote
+            quote.each do |q|
+                WatchList.create(
+                    :symbol => "#{q["e"]}:#{q["t"]}",
+                    :created_at => DateTime.now
+                )
+                puts "Added #{q["t"]}"
+            end
+        end
+    end
+
+    desc 'Show Watchlist'
+    def watchlist
+        items = WatchList.all(:order => [:symbol.asc])
+        symbols = []
+        items.each do |item|
+            symbols << item.symbol
+        end
+        quote = StockFolio::Web.quote(symbols.join(",")) 
+        print_quotes(quote)
+    end
+
+    private
+    def transaction_from_options(options={})
+        # Do we have this portfolio?!?
+        portfolios = Portfolio.all(:name => options[:portfolio])
+        if portfolios.size == 0
+            puts "Portfolio #{options[:portfolio]} does not exist"
+            return
+        end
+
+        portfolio = portfolios[0]
+        puts "Portfolio #{portfolio.id} - #{portfolio.name}"
+
+        # Validate the symbol
+        quote = StockFolio::Web.quote(options[:symbol])
+        if nil == quote
+            puts "Symbol #{options[:symbol]} not found"
+            return
+        end
+
+        transaction = Transaction.new(
+                :symbol => options[:symbol],
+                :quantity => options[:quantity],
+                :price => options[:price],
+
+
+                :fee => 0,
+                :portfolio => portfolio,
+                :executed_at => DateTime.now,
+                :created_at => DateTime.now
+        )
+        transaction.fee = options[:fee] if options[:fee]
+        transaction.executed_at = DateTime.parse(options[:date]) if options[:date]
+
+
+        transaction.save
+
+    end
+
+    def print_quotes(quote)
+        if nil != quote
+            # Beautify those
+            quote.each do |q|
+                q["Symbol"] = q["t"]
+                q["Last Price"] = "$#{q["l"]}"
+                q["Change"] = "#{q["c"]} (#{q["cp"]}%)"
+            end
+            puts Hirb::Helpers::Table.render(quote, fields: ["Symbol", "Last Price", "Change"])
+        end
+ 
+    end
+end

data/lib/stockfolio/transaction.rb

@@ -0,0 +1,31 @@
+class Transaction
+    include DataMapper::Resource
+
+    property :id,          Serial 
+    property :symbol,      String
+    property :quantity,    Integer
+    property :price,       Float
+    property :executed_at, DateTime
+    property :fee,         Float
+    property :created_at,  DateTime
+
+    belongs_to :portfolio
+
+    def order_name
+        if quantity > 0
+            return "BUY "
+        else
+            return "SELL"
+        end
+    end
+
+    def real_fee
+        if fee == nil
+            return 0
+        else
+            return fee
+        end
+    end
+end
+
+

data/lib/stockfolio/version.rb

@@ -0,0 +1,3 @@
+module StockFolio
+    VERSION = '0.1.1'
+end

data/lib/stockfolio/watchlist.rb

@@ -0,0 +1,8 @@
+class WatchList
+    include DataMapper::Resource
+
+    property :id,         Serial 
+    property :symbol,     String
+    property :created_at, DateTime
+end
+

data/lib/stockfolio/web.rb

@@ -0,0 +1,16 @@
+require 'json'
+require 'net/http'
+
+class StockFolio::Web
+    def self.quote(symbol)
+        url = "http://finance.google.com/finance/info?client=ig&q=#{symbol}"
+
+        resp = Net::HTTP.get_response(URI.parse(url))
+        data = resp.body
+
+        if data.empty?
+            return nil
+        end
+        JSON.parse(data.slice(3, data.length).strip)
+    end
+end

metadata

@@ -0,0 +1,76 @@
+--- !ruby/object:Gem::Specification
+name: stockfolio
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+  prerelease: 
+platform: ruby
+authors:
+- Jerome Poichet
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-03-22 00:00:00.000000000Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: boson
+  requirement: &2165393860 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2165393860
+- !ruby/object:Gem::Dependency
+  name: hirb
+  requirement: &2165393380 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *2165393380
+description: 
+email: poitch@gmail.com
+executables:
+- stockfolio
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/stockfolio/portfolio.rb
+- lib/stockfolio/runner.rb
+- lib/stockfolio/transaction.rb
+- lib/stockfolio/version.rb
+- lib/stockfolio/watchlist.rb
+- lib/stockfolio/web.rb
+- lib/stockfolio.rb
+- bin/stockfolio
+homepage: http://github.com/poitch/stockfolio
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.10
+signing_key: 
+specification_version: 3
+summary: Track stock portfolio from the command line
+test_files: []