jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

switch aws client libraries

Browse Source
This commit is contained in:
r888888888
2015-07-13 18:17:31 -07:00
parent 63f4ecf0f1
commit 2c8cacd50e
4 changed files with 26 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
script/donmai/backup_db_to_s3
View File

@@ -1,7 +1,7 @@
#!/usr/bin/env ruby
require 'rubygems'
require 'aws/s3'
require 'aws-sdk/s3'
MAX_BACKUPS = 30
@@ -19,14 +19,15 @@ end
access_key = open("#{home}/.s3/access_key").read.strip
secret_access_key = open("#{home}/.s3/secret_access_key").read.strip
credentials = Aws::Credentials.new(access_key, secret_access_key)
client = Aws::S3::Client.new(region: "us-west-2", credentials: credentials)
bucket = "danbooru-backup"
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
current_backups = client.list_objects(buckets: buckets).contents.map {|x| x.key}.select {|x| x =~ /^db-/}.sort.reverse
if current_backups.size > MAX_BACKUPS
current_backups[MAX_BACKUPS..-1].each do |old_backup|
AWS::S3::S3Object.delete(old_backup, "danbooru-backup")
client.delete_object(bucket: bucket, key: old_backup)
puts "Deleted old backup #{old_backup}"
end
end
@@ -37,7 +38,7 @@ filename = data.mtime.strftime("db-%Y-%m-%d-%H-%M")
tries = 0
begin
AWS::S3::S3Object.store(filename, data, "danbooru-backup", :access => :private)
client.put_object(bucket: bucket, key: filename, body: data, :acl => "private")
rescue Errno::EPIPE
tries += 1
if tries > 3