bigquery 0.1.0

data/lib/bigquery.rb

@@ -0,0 +1,62 @@
+require 'google/api_client'
+
+class BigQuery
+
+  attr_accessor :dataset, :project_id
+
+  def initialize(opts = {})
+    @client = Google::APIClient.new
+
+    @client.authorization.client_id = opts['client_id']
+    @client.authorization.client_secret = opts['client_secret']
+    @client.authorization.refresh_token = opts['refresh_token']
+
+    @client.authorization.fetch_access_token!
+    @bq = @client.discovered_api("bigquery", "v2")
+
+    @project_id = opts['project_id']
+    @dataset = opts['dataset']
+  end
+
+  def query(q)
+    api({
+      :api_method => @bq.jobs.query,
+      :body_object => { "query" => q, 'timeoutMs' => 90 * 1000}
+    })
+  end
+
+  def load(opts)
+    api({
+      :api_method => @bq.jobs.insert,
+      :body_object => {
+        "configuration" => {
+          "load" => opts
+        }
+      }
+    })
+  end
+
+  def tables(dataset = @dataset)
+    api({
+      :api_method => @bq.tables.list,
+      :parameters => {"datasetId" => dataset}
+    })['tables']
+  end
+
+  def tables_formatted(dataset = @dataset)
+    tables(dataset).map {|t| "[#{dataset}.#{t['tableReference']['tableId']}]"}
+  end
+
+  private
+
+  def api(opts)
+    if opts[:parameters]
+      opts[:parameters] = opts[:parameters].merge({"projectId" => @project_id})
+    else
+      opts[:parameters] = {"projectId" => @project_id}
+    end
+
+    resp = @client.execute(opts)
+    JSON.parse(resp.body)
+  end
+end

metadata

@@ -0,0 +1,61 @@
+--- !ruby/object:Gem::Specification
+name: bigquery
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+  prerelease: 
+platform: ruby
+authors:
+- Adam Bronte
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-09-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: google-api-client
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: A nice wrapper for Google Big Query
+email: adam@brontesaurus.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/bigquery.rb
+homepage: https://github.com/abronte/BigQuery
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- 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.24
+signing_key: 
+specification_version: 3
+summary: A nice wrapper for Google Big Query
+test_files: []