stockfolio 0.1.4 → 0.1.5

data/bin/stockfolio

@@ -3,4 +3,9 @@
 $LOAD_PATH << './lib'
 
 require 'stockfolio'
-StockFolio::Runner.start
+begin
+    StockFolio::Runner.start
+rescue Boson::OptionParser::Error => e
+    puts e.message
+    exit 1
+end

data/lib/stockfolio/check.rb

@@ -0,0 +1,15 @@
+require 'stockfolio/version'
+require 'gems'
+
+class StockFolio::Check
+
+    def self.has_newer
+        @info = Gems.info 'stockfolio'
+        Gem::Version.new(@info["version"]) > Gem::Version.new(StockFolio::VERSION)
+    end
+
+    def self.version_available
+        return @info["version"]
+    end
+
+end

data/lib/stockfolio/runner.rb

@@ -12,11 +12,46 @@ class StockFolio::Runner < Boson::Runner
         option :date, :type => :string, :desc => 'Transaction date'
     end
 
+    def version
+        require 'stockfolio/version'
+        require 'stockfolio/check'
+        puts StockFolio::VERSION
+        if StockFolio::Check.has_newer
+            puts "There is a more recent version #{StockFolio::Check.version_available}"
+        end
+    end
+
+    desc "Current quote for a stock"
     def quote(symbol)
         quote = StockFolio::Web.quote(symbol)
         print_quotes(quote)
     end
 
+    desc "Historical prices for a stock"
+    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])
+    end
+
+    desc "Search symbol"
+    def search(term)
+        matches = StockFolio::Web.search(term)
+
+        lines = []
+        matches.each do |match|
+            if match['e'].length > 0 && match['t'].length > 0
+            line = {
+                :Symbol => "#{match['e']}:#{match['t']}",
+                :Name => match['n']
+            }
+            lines << line
+            end
+        end
+
+        puts Hirb::Helpers::Table.render(lines, :fields => [:Symbol, :Name])
+
+    end
+
     desc 'List porfolios'
     def portfolios
         q = Portfolio.all
@@ -122,6 +157,72 @@ class StockFolio::Runner < Boson::Runner
         puts "Created portfolio \"#{name}\""
     end
 
+    desc 'List transactions for a portfolio'
+    def transactions(name)
+        p = Portfolio.first(:name => name)
+        if p == nil
+            puts "Portfolio #{name} not found"
+            exit
+        end
+
+        transactions = []
+        p.transactions.each do |transaction|
+            transactions << {
+                :id => transaction.id,
+                :symbol => transaction.symbol,
+                :quantity => transaction.quantity,
+                :price => transaction.price,
+                :executed_at => transaction.executed_at,
+                :fee => transaction.fee,
+                :created_at => transaction.created_at
+            }
+        end
+        puts Hirb::Helpers::Table.render(transactions, fields: [:id, :symbol, :quantity, :price, :fee, :executed_at, :created_at])
+
+    end
+
+    def delete(type,alpha,beta = nil)
+        if type == "transaction"
+            name = alpha
+            id = beta
+            puts "Deleting transaction #{id} from #{name}"
+
+            portfolio = Portfolio.first(:name => name)
+            if portfolio == nil
+                puts "Portfolio #{name} not found"
+                exit
+            end
+
+            transaction = Transaction.get(id)
+
+            if transaction != nil && transaction.portfolio == portfolio
+                transaction.destroy!
+            else
+                puts "Could not find transaction #{id} in portfolio #{name}"
+            end
+
+        elsif type == "portfolio"
+            name = alpha
+            puts "Deleting portfolio #{name}"
+
+            portfolio = Portfolio.first(:name => name)
+            if portfolio == nil
+                puts "Portfolio #{name} not found"
+                exit
+            end
+
+            portfolio.transactions.each do |transaction|
+                transaction.destroy!
+            end
+            portfolio.destroy!
+
+        end
+    end
+
+    #
+    # WATCHLIST
+    # 
+
     desc 'Add to watchlist'
     def watch(symbol)
         # Validate symbol
@@ -137,6 +238,15 @@ class StockFolio::Runner < Boson::Runner
         end
     end
 
+    desc 'Remove from watchlist'
+    def unwatch(symbol)
+        item = WatchList.first(:symbol => symbol)
+        if item != nil
+            item.destroy
+            puts "Removed #{symbol} from watchlist"
+        end
+    end
+
     desc 'Show Watchlist'
     def watchlist
         items = WatchList.all(:order => [:symbol.asc])

data/lib/stockfolio/version.rb

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

data/lib/stockfolio/web.rb

@@ -1,3 +1,4 @@
+require 'uri'
 require 'json'
 require 'net/http'
 
@@ -13,4 +14,49 @@ class StockFolio::Web
         end
         JSON.parse(data.slice(3, data.length).strip)
     end
+
+    def self.historical(symbol, startDate, endDate)
+        s = startDate.strftime("%b+%d,+%Y")
+        e = endDate.strftime("%b+%d,+%Y")
+        url = "http://finance.google.com/finance/historical?q=#{symbol}&startdate=#{s}&enddate=#{e}&output=csv"
+
+        resp = Net::HTTP.get_response(URI.parse(url))
+        data = resp.body
+
+        if data.empty?
+            return nil
+        end
+
+        lines = data.split("\n")
+        lines.shift
+        items = []
+        lines.each do |line|
+            parts = line.split(",")
+            i = 0
+            item = {}
+            item[:date]   = Date.parse(parts[0])
+            item[:open]   = parts[1]
+            item[:high]   = parts[2]
+            item[:low]    = parts[3]
+            item[:close]  = parts[4]
+            item[:volume] = parts[5]
+            items << item
+        end
+
+        items
+    end
+
+    def self.search(term)
+        q = URI.escape(term, Regexp.new("[^#{URI::PATTERN::UNRESERVED}]"))
+        url = "http://www.google.com/finance/match?matchtype=matchall&q=#{q}"
+        resp = Net::HTTP.get_response(URI.parse(url))
+        data = resp.body
+
+        if data.empty?
+            return nil
+        end
+        result = JSON.parse(data)
+        result["matches"]
+    end
+
 end

metadata

@@ -1,89 +1,136 @@
---- !ruby/object:Gem::Specification
+--- !ruby/object:Gem::Specification 
 name: stockfolio
-version: !ruby/object:Gem::Version
-  version: 0.1.4
-  prerelease: 
+version: !ruby/object:Gem::Version 
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 5
+  version: 0.1.5
 platform: ruby
-authors:
+authors: 
 - Jerome Poichet
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-03-22 00:00:00.000000000Z
-dependencies:
-- !ruby/object:Gem::Dependency
+
+date: 2012-03-22 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
   name: boson
-  requirement: &2169047320 !ruby/object:Gem::Requirement
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
     none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: '0'
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
   type: :runtime
-  prerelease: false
-  version_requirements: *2169047320
-- !ruby/object:Gem::Dependency
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
   name: hirb
-  requirement: &2169046660 !ruby/object:Gem::Requirement
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
     none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: '0'
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
   type: :runtime
-  prerelease: false
-  version_requirements: *2169046660
-- !ruby/object:Gem::Dependency
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
   name: data_mapper
-  requirement: &2169045840 !ruby/object:Gem::Requirement
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
     none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: '0'
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
   type: :runtime
-  prerelease: false
-  version_requirements: *2169045840
-- !ruby/object:Gem::Dependency
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
   name: dm-migrations
-  requirement: &2169045220 !ruby/object:Gem::Requirement
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
     none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: '0'
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
   type: :runtime
-  prerelease: false
-  version_requirements: *2169045220
-- !ruby/object:Gem::Dependency
+  version_requirements: *id004
+- !ruby/object:Gem::Dependency 
   name: dm-sqlite-adapter
-  requirement: &2169044620 !ruby/object:Gem::Requirement
+  prerelease: false
+  requirement: &id005 !ruby/object:Gem::Requirement 
     none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: '0'
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
   type: :runtime
-  prerelease: false
-  version_requirements: *2169044620
-- !ruby/object:Gem::Dependency
+  version_requirements: *id005
+- !ruby/object:Gem::Dependency 
   name: json
-  requirement: &2169043820 !ruby/object:Gem::Requirement
+  prerelease: false
+  requirement: &id006 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id006
+- !ruby/object:Gem::Dependency 
+  name: uri
+  prerelease: false
+  requirement: &id007 !ruby/object:Gem::Requirement 
     none: false
-    requirements:
-    - - ! '>='
-      - !ruby/object:Gem::Version
-        version: '0'
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
   type: :runtime
+  version_requirements: *id007
+- !ruby/object:Gem::Dependency 
+  name: gems
   prerelease: false
-  version_requirements: *2169043820
+  requirement: &id008 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id008
 description: 
 email: poitch@gmail.com
-executables:
+executables: 
 - stockfolio
 extensions: []
+
 extra_rdoc_files: []
-files:
+
+files: 
+- lib/stockfolio/check.rb
 - lib/stockfolio/portfolio.rb
 - lib/stockfolio/runner.rb
 - lib/stockfolio/transaction.rb
@@ -92,29 +139,38 @@ 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
-      version: '0'
-required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
   none: false
-  requirements:
-  - - ! '>='
-    - !ruby/object:Gem::Version
-      version: '0'
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      segments: 
+      - 0
+      version: "0"
 requirements: []
+
 rubyforge_project: 
-rubygems_version: 1.8.10
+rubygems_version: 1.3.7
 signing_key: 
 specification_version: 3
 summary: Track stock portfolio from the command line
 test_files: []
+