diff --git a/lib/myob/api/client.rb b/lib/myob/api/client.rb index 6ee3aa7..f12b2d2 100644 --- a/lib/myob/api/client.rb +++ b/lib/myob/api/client.rb @@ -8,7 +8,7 @@ class Client attr_reader :current_company_file, :client - def initialize(options) + def initialize(options, &block) Myob::Api::Model::Base.subclasses.each {|c| model(c.name.split("::").last)} @redirect_uri = options[:redirect_uri] @@ -20,7 +20,7 @@ def initialize(options) :site => 'https://secure.myob.com', :authorize_url => '/oauth2/account/authorize', :token_url => '/oauth2/v1/authorize', - }) + }, &block) if options[:company_file] @current_company_file = select_company_file(options[:company_file])