import bigquery classes from reportbooru
This commit is contained in:
32
app/logical/big_query/base.rb
Normal file
32
app/logical/big_query/base.rb
Normal file
@@ -0,0 +1,32 @@
|
||||
require "big_query"
|
||||
|
||||
module BigQuery
|
||||
class Base
|
||||
def query(q)
|
||||
client.query(q)
|
||||
end
|
||||
|
||||
def escape(s)
|
||||
Regexp.escape(s).gsub(/\\/, '\0\0').gsub(/['"]/, '\\\\\0')
|
||||
end
|
||||
|
||||
def client
|
||||
@_client ||= BigQuery::Client.new(
|
||||
"json_key" => client_options[:google_key_path],
|
||||
"project_id" => google_config["project_id"],
|
||||
"dataset" => client_options[:google_data_set]
|
||||
)
|
||||
end
|
||||
|
||||
def client_options
|
||||
@_client_options ||= {
|
||||
google_key_path: "/var/www/danbooru/shared/google-key.json",
|
||||
google_data_set: "danbooru_#{Rails.env}"
|
||||
}
|
||||
end
|
||||
|
||||
def google_config
|
||||
@_google_config ||= JSON.parse(File.read(client_options[:google_key_path]))
|
||||
end
|
||||
end
|
||||
end
|
||||
Reference in New Issue
Block a user