diff --git a/app/models/amazon_backup.rb b/app/models/amazon_backup.rb index 19919a25d..f6bb7447e 100644 --- a/app/models/amazon_backup.rb +++ b/app/models/amazon_backup.rb @@ -16,7 +16,8 @@ class AmazonBackup < ActiveRecord::Base Post.where("id > ?", last_id).limit(200).order("id").each do |post| AWS::S3::Base.establish_connection!( :access_key_id => Danbooru.config.amazon_s3_access_key_id, - :secret_access_key => Danbooru.config.amazon_s3_secret_access_key + :secret_access_key => Danbooru.config.amazon_s3_secret_access_key, + :server => "s3.amazonaws.com" ) if File.exists?(post.file_path) @@ -24,7 +25,7 @@ class AmazonBackup < ActiveRecord::Base AWS::S3::S3Object.store(File.basename(post.file_path), open(post.file_path, "rb"), Danbooru.config.amazon_s3_bucket_name, "Content-MD5" => base64_md5) end - if post.image? && File.exists?(post.preview_file_path) + if post.is_image? && File.exists?(post.preview_file_path) AWS::S3::S3Object.store("preview/#{post.md5}.jpg", open(post.preview_file_path, "rb"), Danbooru.config.amazon_s3_bucket_name) end @@ -32,7 +33,7 @@ class AmazonBackup < ActiveRecord::Base AWS::S3::S3Object.store("large/#{post.md5}.jpg", open(post.large_file_path, "rb"), Danbooru.config.amazon_s3_bucket_name) end - AmazonBackup.update_id(last_id) + AmazonBackup.update_id(post.id) end rescue Exception => x # probably some network error, retry next time diff --git a/config/danbooru_default_config.rb b/config/danbooru_default_config.rb index e51b83179..44277d123 100644 --- a/config/danbooru_default_config.rb +++ b/config/danbooru_default_config.rb @@ -4,7 +4,7 @@ module Danbooru class Configuration # The version of this Danbooru. def version - "2.39.0" + "2.40.0" end # The name of this Danbooru. @@ -298,6 +298,10 @@ module Danbooru nil end + def amazon_s3_bucket_name + "danbooru" + end + def enable_dimension_autotagging true end diff --git a/script/donmai/backup_db_to_s3 b/script/donmai/backup_db_to_s3 index a9f2cb60c..f3de65c29 100755 --- a/script/donmai/backup_db_to_s3 +++ b/script/donmai/backup_db_to_s3 @@ -20,7 +20,7 @@ end access_key = open("#{home}/.s3/access_key").read.strip secret_access_key = open("#{home}/.s3/secret_access_key").read.strip -AWS::S3::Base.establish_connection!(:access_key_id => access_key, :secret_access_key => secret_access_key) +AWS::S3::Base.establish_connection!(:access_key_id => access_key, :secret_access_key => secret_access_key, :server => "s3.amazonaws.com") current_backups = AWS::S3::Bucket.find("danbooru-backup").objects.map {|x| x.key}.select {|x| x =~ /^db-/}.sort.reverse @@ -34,5 +34,17 @@ end backup = Dir["/var/www/danbooru2/shared/backup/db-*.dump"].sort.last data = File.open(backup, "rb") filename = data.mtime.strftime("db-%Y-%m-%d-%H-%M") -AWS::S3::S3Object.store(filename, data, "danbooru-backup", :access => :private) +tries = 0 + +begin + AWS::S3::S3Object.store(filename, data, "danbooru-backup", :access => :private) +rescue Errno::EPIPE + tries += 1 + if tries > 3 + raise + else + retry + end +end + puts "Uploaded #{backup} as #{filename}"