robinhood-api 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: cf571e1d096eb33585450426648dbb611defd49e
-  data.tar.gz: d3f5ead7ec9e3a78cd4193a163219664f9b5c595
+  metadata.gz: 4dfe1dce8dda9168979bd651c80b877f0ab9e0a0
+  data.tar.gz: 684af5059539659160b16aafeec5048bc185e54a
 SHA512:
-  metadata.gz: 58518aea355a1c4345094807000ef4be7110b0146f315c3bce29a4318832be2b4b5a4f7779fa2fce4feab861d179111ddcfb61e86bd8a5429859650129d7d880
-  data.tar.gz: cbb117fe801eccd261d373b71f7b341d9d5784db1f2d9480955cbb509794a35b00753aac01312e0f1b21593e8b85a57f4ee6c2427fe431f06814872a79e5b740
+  metadata.gz: 8a93b1179979a58f8f41892d6735a23b6ca63b13b27cdb84128b8d8d22d280d82a542f9ef1b28779c2950cc2bb1c0510156a852dc210dc7aa66bd898dfdde44f
+  data.tar.gz: b31cd8904edb399e9b062f9801d648de7b09ecf5ff81076238a1eb53f833f97084eff1039ce5ce040668962005ea0bdf69c802c801e2b4ea281df165ad394e39

data/lib/robinhood.rb

@@ -4,11 +4,16 @@ module RobinhoodModule
       m = instance_method(name)
       define_method(name) do |*args, &block|  
         # yield
-        if token_compliant(name)
-          m.bind(self).(*args, &block)
+        if prerequisites?(name)
+          if token_compliant?(name)
+            m.bind(self).(*args, &block)
+          else
+            puts "You have to run the login(<username>, <password>) method or a Robinhood.new instance before running the #{name} method"
+            exit 1
+          end
         else
-          puts "You have to run the login(<username>, <password>) method or a Robinhood.new instance before running the #{name} method"
-          exit 1
+          puts "You must define ROBINHOOD_USERNAME and ROBINHOOD_PASSWORD environment variables to run the #{name} method"
+          exit 2
         end
       end
     end
@@ -200,14 +205,25 @@ module RobinhoodModule
     }
   end
 
+  def methodlist
+    [:investment_profile, :accounts, :ach_iav_auth, :ach_relationships, :ach_transfers, :applications, :dividends, :edocuments, :margin_upgrade, :notifications, :orders, :password_reset, :document_requests, :user, :watchlists]
+  end
+
   def headers
     @headers ||= {
       'Accept' => 'application/json',
     }
   end
 
-  def token_compliant(method)
-    methodlist = [:investment_profile, :accounts, :ach_iav_auth, :ach_relationships, :ach_transfers, :applications, :dividends, :edocuments, :margin_upgrade, :notifications, :orders, :password_reset, :document_requests, :user, :watchlists]
+  def prerequisites?(method)
+    if methodlist.include?(method)
+      return ENV['ROBINHOOD_USERNAME'] && ENV['ROBINHOOD_PASSWORD']
+    else
+      return true
+    end
+  end
+
+  def token_compliant?(method)
 
     if methodlist.include?(method)
       if headers.key?('Authorization')

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: robinhood-api
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Thomas Alcala