implement post change report

This commit is contained in:
r888888888
2016-09-07 14:35:35 -07:00
parent 62773852f7
commit e539fdecff
8 changed files with 21 additions and 12 deletions

View File

@@ -1,6 +1,6 @@
require "big_query"
module BigQuery
module GoogleBigQuery
class Base
def query(q)
client.query(q)
@@ -20,7 +20,7 @@ module BigQuery
def client_options
@_client_options ||= {
google_key_path: "/var/www/danbooru/shared/google-key.json",
google_key_path: Danbooru.config.google_api_json_key_path,
google_data_set: "danbooru_#{Rails.env}"
}
end
@@ -28,5 +28,9 @@ module BigQuery
def google_config
@_google_config ||= JSON.parse(File.read(client_options[:google_key_path]))
end
def data_set
client_options[:google_data_set]
end
end
end

View File

@@ -1,15 +1,15 @@
module BigQuery
module GoogleBigQuery
class PostVersion < Base
def find_removed(tag, limit = 1_000)
tag = escape(tag)
limit = limit.to_i
query("select id, post_id, updated_at, updater_id, updater_ip_addr, tags, added_tags, removed_tags, parent_id, rating, source from [danbooru_#{Rails.env}.post_versions] where regexp_match(removed_tags, \"(?:^| )#{tag}(?:$| )\") order by updated_at desc limit #{limit}")
query("select id, post_id, updated_at, updater_id, updater_ip_addr, tags, added_tags, removed_tags, parent_id, rating, source from [#{data_set}.post_versions] where regexp_match(removed_tags, \"(?:^| )#{tag}(?:$| )\") order by updated_at desc limit #{limit}")
end
def find_added(tag, limit = 1_000)
tag = escape(tag)
limit = limit.to_i
query("select id, post_id, updated_at, updater_id, updater_ip_addr, tags, added_tags, removed_tags, parent_id, rating, source from [danbooru_#{Rails.env}.post_versions] where regexp_match(added_tags, \"(?:^| )#{tag}(?:$| )\") order by updated_at desc limit #{limit}")
query("select id, post_id, updated_at, updater_id, updater_ip_addr, tags, added_tags, removed_tags, parent_id, rating, source from [#{data_set}.post_versions] where regexp_match(added_tags, \"(?:^| )#{tag}(?:$| )\") order by updated_at desc limit #{limit}")
end
end
end

View File

@@ -6,13 +6,13 @@ module Reports
@tag = tag
if query_type == "added"
@query = BigQuery::PostVersion.new.find_added(tag)
@query = GoogleBigQuery::PostVersion.new.find_added(tag)
else
@query = BigQuery::PostVersion.new.find_removed(tag)
@query = GoogleBigQuery::PostVersion.new.find_removed(tag)
end
end
def mock_version(raw)
def mock_version(row)
PostVersion.new.tap do |x|
x.id = row["f"][0]["v"]
x.post_id = row["f"][1]["v"]