edshot99/eBooru
1
0
Fork 0
forked from e621ng/e621ng
Code Issues 11 Releases 1 Activity

Fix failure in rake db:migrate on fresh install (#3443).

Browse Source 
Create the DanbooruBot account in db/seeds.rb instead of in an
initializer. Move the old db/seeds.rb to db/populate.rb.

Fixes a bug introduced in e2eb45a. Creating DanbooruBot in an
initializer was wrong because in a fresh install the users table doesn't
exist yet. This caused `rake db:migrate` to fail.
This commit is contained in:
evazion 2017-12-21 11:56:12 -06:00
parent b48d5e59ad
commit fcec99405c
5 changed files with 230 additions and 227 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
config/danbooru_default_config.rb
View File

@ -36,9 +36,10 @@ module Danbooru
"webmaster@#{server_host}"
end
# System actions, such as sending automated dmails, will be performed with this account.
# This account will be created automatically if it doesn't exist. It will
# be promoted to Moderator if it isn't already a Moderator.
# System actions, such as sending automated dmails, will be performed with
# this account. This account must have Moderator privileges.
#
# Run `rake db:seed` to create this account if it doesn't already exist in your install.
def system_user
"DanbooruBot"
end

10
config/initializers/system_user.rb
View File

@ -1,10 +0,0 @@
require "securerandom"
system = User.find_or_create_by!(name: Danbooru.config.system_user) do |user|
user.password = SecureRandom.base64(32)
end
unless system.is_moderator?
system.level = User::Levels::MODERATOR
system.save
end

218
db/populate.rb Normal file
View File

@ -0,0 +1,218 @@
require 'set'
require 'timecop'
CurrentUser.ip_addr = "127.0.0.1"
Delayed::Worker.delay_jobs = false
$used_names = Set.new([""])
def rand_string(n, unique = false)
string = ""
n = rand(n) + 1
while $used_names.include?(string)
consonants = "bcdfghjklmnpqrstvwxz".scan(/./)
vowels = "aeiouy".scan(/./)
string = ""
n.times do
string << consonants[rand(consonants.size)]
string << vowels[rand(vowels.size)]
end
return string unless unique
end
$used_names.add(string)
string
end
def rand_sentence(n, unique = false)
(0..n).map {rand_string(n, unique)}.join(" ") + "."
end
def rand_paragraph(n, unique = false)
(0..n).map {rand_sentence(n, unique)}.join(" ")
end
def rand_document(n, unique = false)
(0..n).map {rand_paragraph(n, unique)}.join("\n\n")
end
if User.count == 0
puts "Creating users"
Timecop.travel(1.month.ago) do
user = User.create(
:name => "admin",
:password => "password1",
:password_confirmation => "password1"
)
CurrentUser.user = user
User::Levels.constants.reject{|x| [:ADMIN, :BLOCKED].include?(x)}.each do |level|
newuser = User.create(
:name => level.to_s.downcase,
:password => "password1",
:password_confirmation => "password1"
)
newuser.promote_to!(User::Levels.const_get(level), {:skip_feedback => true, :skip_dmail => true})
end
newuser = User.create(
:name => "banned",
:password => "password1",
:password_confirmation => "password1"
)
Ban.create(:user_id => newuser.id, :reason => "from the start", :duration => 99999)
newuser = User.create(
:name => "uploader",
:password => "password1",
:password_confirmation => "password1"
)
newuser.promote_to!(User::Levels::BUILDER, {:can_upload_free => true, :skip_feedback => true, :skip_dmail => true})
newuser = User.create(
:name => "approver",
:password => "password1",
:password_confirmation => "password1"
)
newuser.promote_to!(User::Levels::BUILDER, {:can_approve_posts => true, :skip_feedback => true, :skip_dmail => true})
end
0.upto(10) do |i|
User.create(
:name => rand_string(8, true),
:password => "password1",
:password_confirmation => "password1"
)
end
$used_names = Set.new([""])
else
puts "Skipping users"
user = User.find_by_name("albert")
end
CurrentUser.user = User.admins.first
if Upload.count == 0
puts "Creating uploads"
1.upto(100) do |i|
color1 = rand(4096).to_s(16)
color2 = rand(4096).to_s(16)
width = rand(2000) + 100
height = rand(2000) + 100
url = "http://ipsumimage.appspot.com/#{width}x#{height}"
tags = rand_sentence(12).scan(/[a-z]+/).join(" ")
Upload.create!(:source => url, :content_type => "image/gif", :rating => "q", :tag_string => tags, :server => Socket.gethostname)
end
else
puts "Skipping uploads"
end
if Post.count == 0
puts "Creating posts"
Upload.all.each do |upload|
upload.process!
end
if Post.count == 0
raise "Uploads failed conversion"
end
else
puts "Skipping posts"
end
if Comment.count == 0
puts "Creating comments"
Post.all.each do |post|
rand(10).times do
Comment.create(:post_id => post.id, :body => rand_paragraph(6))
end
end
else
puts "Skipping comments"
end
if Note.count == 0
puts "Creating notes"
Post.order("random()").limit(10).each do |post|
rand(5).times do
note = Note.create(:post_id => post.id, :x => rand(post.image_width), :y => rand(post.image_height), :width => 100, :height => 100, :body => Time.now.to_f.to_s)
rand(10).times do |i|
note.update_attributes(:body => rand_sentence(6))
end
end
end
else
puts "Skipping notes"
end
if Artist.count == 0
puts "Creating artists"
20.times do |i|
Artist.create(:name => rand_string(9, true))
end
$used_names = Set.new([""])
else
puts "Skipping artists"
end
if TagAlias.count == 0
puts "Creating tag aliases"
20.times do |i|
TagAlias.create(:antecedent_name => rand_string(9, true), :consequent_name => rand_string(9, true))
end
$used_names = Set.new([""])
else
puts "Skipping tag aliases"
end
if TagImplication.count == 0
puts "Creating tag implictions"
20.times do |i|
TagImplication.create(:antecedent_name => rand_string(9, true), :consequent_name => rand_string(9, true))
end
$used_names = Set.new([""])
else
puts "Skipping tag implications"
end
if Pool.count == 0
puts "Creating pools"
1.upto(20) do |i|
pool = Pool.create(:name => rand_string(9, true))
rand(33).times do |j|
pool.add!(Post.order("random()").first)
end
end
$used_names = Set.new([""])
end
if Favorite.count == 0
puts "Creating favorites"
Post.order("random()").limit(50).each do |post|
user = User.order("random()").first
post.add_favorite!(user)
post.add_favorite!(CurrentUser.user)
end
else
puts "Skipping favorites"
end
if ForumTopic.count == 0
puts "Creating forum posts"
20.times do |i|
topic = ForumTopic.create(:title => rand_sentence(6))
rand(100).times do |j|
post = ForumPost.create(:topic_id => topic.id, :body => rand_document(6))
end
end
end

220
db/seeds.rb
View File

@ -1,218 +1,10 @@
require 'set'
require 'timecop'
require "securerandom"
CurrentUser.ip_addr = "127.0.0.1"
Delayed::Worker.delay_jobs = false
$used_names = Set.new([""])
def rand_string(n, unique = false)
string = ""
n = rand(n) + 1
while $used_names.include?(string)
consonants = "bcdfghjklmnpqrstvwxz".scan(/./)
vowels = "aeiouy".scan(/./)
string = ""
n.times do
string << consonants[rand(consonants.size)]
string << vowels[rand(vowels.size)]
end
return string unless unique
end
$used_names.add(string)
string
system = User.find_or_create_by!(name: Danbooru.config.system_user) do |user|
user.password = SecureRandom.base64(32)
end
def rand_sentence(n, unique = false)
(0..n).map {rand_string(n, unique)}.join(" ") + "."
end
def rand_paragraph(n, unique = false)
(0..n).map {rand_sentence(n, unique)}.join(" ")
end
def rand_document(n, unique = false)
(0..n).map {rand_paragraph(n, unique)}.join("\n\n")
end
if User.count == 0
puts "Creating users"
Timecop.travel(1.month.ago) do
user = User.create(
:name => "admin",
:password => "password1",
:password_confirmation => "password1"
)
CurrentUser.user = user
User::Levels.constants.reject{|x| [:ADMIN, :BLOCKED].include?(x)}.each do |level|
newuser = User.create(
:name => level.to_s.downcase,
:password => "password1",
:password_confirmation => "password1"
)
newuser.promote_to!(User::Levels.const_get(level), {:skip_feedback => true, :skip_dmail => true})
end
newuser = User.create(
:name => "banned",
:password => "password1",
:password_confirmation => "password1"
)
Ban.create(:user_id => newuser.id, :reason => "from the start", :duration => 99999)
newuser = User.create(
:name => "uploader",
:password => "password1",
:password_confirmation => "password1"
)
newuser.promote_to!(User::Levels::BUILDER, {:can_upload_free => true, :skip_feedback => true, :skip_dmail => true})
newuser = User.create(
:name => "approver",
:password => "password1",
:password_confirmation => "password1"
)
newuser.promote_to!(User::Levels::BUILDER, {:can_approve_posts => true, :skip_feedback => true, :skip_dmail => true})
end
0.upto(10) do |i|
User.create(
:name => rand_string(8, true),
:password => "password1",
:password_confirmation => "password1"
)
end
$used_names = Set.new([""])
else
puts "Skipping users"
user = User.find_by_name("albert")
end
CurrentUser.user = User.admins.first
if Upload.count == 0
puts "Creating uploads"
1.upto(100) do |i|
color1 = rand(4096).to_s(16)
color2 = rand(4096).to_s(16)
width = rand(2000) + 100
height = rand(2000) + 100
url = "http://ipsumimage.appspot.com/#{width}x#{height}"
tags = rand_sentence(12).scan(/[a-z]+/).join(" ")
Upload.create!(:source => url, :content_type => "image/gif", :rating => "q", :tag_string => tags, :server => Socket.gethostname)
end
else
puts "Skipping uploads"
end
if Post.count == 0
puts "Creating posts"
Upload.all.each do |upload|
upload.process!
end
if Post.count == 0
raise "Uploads failed conversion"
end
else
puts "Skipping posts"
end
if Comment.count == 0
puts "Creating comments"
Post.all.each do |post|
rand(10).times do
Comment.create(:post_id => post.id, :body => rand_paragraph(6))
end
end
else
puts "Skipping comments"
end
if Note.count == 0
puts "Creating notes"
Post.order("random()").limit(10).each do |post|
rand(5).times do
note = Note.create(:post_id => post.id, :x => rand(post.image_width), :y => rand(post.image_height), :width => 100, :height => 100, :body => Time.now.to_f.to_s)
rand(10).times do |i|
note.update_attributes(:body => rand_sentence(6))
end
end
end
else
puts "Skipping notes"
end
if Artist.count == 0
puts "Creating artists"
20.times do |i|
Artist.create(:name => rand_string(9, true))
end
$used_names = Set.new([""])
else
puts "Skipping artists"
end
if TagAlias.count == 0
puts "Creating tag aliases"
20.times do |i|
TagAlias.create(:antecedent_name => rand_string(9, true), :consequent_name => rand_string(9, true))
end
$used_names = Set.new([""])
else
puts "Skipping tag aliases"
end
if TagImplication.count == 0
puts "Creating tag implictions"
20.times do |i|
TagImplication.create(:antecedent_name => rand_string(9, true), :consequent_name => rand_string(9, true))
end
$used_names = Set.new([""])
else
puts "Skipping tag implications"
end
if Pool.count == 0
puts "Creating pools"
1.upto(20) do |i|
pool = Pool.create(:name => rand_string(9, true))
rand(33).times do |j|
pool.add!(Post.order("random()").first)
end
end
$used_names = Set.new([""])
end
if Favorite.count == 0
puts "Creating favorites"
Post.order("random()").limit(50).each do |post|
user = User.order("random()").first
post.add_favorite!(user)
post.add_favorite!(CurrentUser.user)
end
else
puts "Skipping favorites"
end
if ForumTopic.count == 0
puts "Creating forum posts"
20.times do |i|
topic = ForumTopic.create(:title => rand_sentence(6))
rand(100).times do |j|
post = ForumPost.create(:topic_id => topic.id, :body => rand_document(6))
end
end
unless system.is_moderator?
system.level = User::Levels::MODERATOR
system.save
end

2
test/test_helper.rb
View File

@ -64,3 +64,5 @@ end
Delayed::Worker.delay_jobs = false
TestAfterCommit.enabled = false
Rails.application.load_seed