mirror of
https://github.com/discourse/discourse.git
synced 2024-11-24 00:37:06 +08:00
49 lines
2.0 KiB
Ruby
49 lines
2.0 KiB
Ruby
|
desc 'export the database'
|
||
|
|
task 'export', [:output_filename] => :environment do |t, args|
|
||
|
|
puts 'Starting export...'
|
||
|
|
output_filename = Jobs::Exporter.new.execute( format: :json, filename: args.output_filename )
|
||
|
|
puts 'Export done.'
|
||
|
|
puts "Output file is in: #{output_filename}", ''
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'import from an export file and replace the contents of the current database'
|
||
|
|
task 'import', [:input_filename] => :environment do |t, args|
|
||
|
|
puts 'Starting import...'
|
||
|
|
begin
|
||
|
|
Jobs::Importer.new.execute( format: :json, filename: args.input_filename )
|
||
|
|
puts 'Import done.'
|
||
|
|
rescue Import::FilenameMissingError
|
||
|
|
puts '', 'The filename argument was missing.', '', 'Usage:', ''
|
||
|
|
puts ' rake import[/path/to/export.json.gz]', ''
|
||
|
|
rescue Import::ImportDisabledError
|
||
|
|
puts '', 'Imports are not allowed.', 'An admin needs to set allow_import to true in the site settings before imports can be run.', ''
|
||
|
|
puts 'Import cancelled.', ''
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'After a successful import, restore the backup tables'
|
||
|
|
task 'import:rollback' => :environment do |t|
|
||
|
|
num_backup_tables = User.exec_sql("select count(*) as count from information_schema.tables where table_schema = 'backup'")[0]['count'].to_i
|
||
|
|
|
||
|
|
if User.exec_sql("select count(*) as count from information_schema.schemata where schema_name = 'backup'")[0]['count'].to_i <= 0
|
||
|
|
puts "Backup tables don't exist! An import was never performed or the backup tables were dropped.", "Rollback cancelled."
|
||
|
|
elsif num_backup_tables != Export.models_included_in_export.size
|
||
|
|
puts "Expected #{Export.models_included_in_export.size} backup tables, but there are #{num_backup_tables}!", "Rollback cancelled."
|
||
|
|
else
|
||
|
|
puts 'Starting rollback..'
|
||
|
|
Jobs::Importer.new.rollback
|
||
|
|
puts 'Rollback done.'
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'Allow imports'
|
||
|
|
task 'import:enable' => :environment do |t|
|
||
|
|
SiteSetting.allow_import = true
|
||
|
|
puts 'Imports are now permitted. Disable them with rake import:disable'
|
||
|
|
end
|
||
|
|
|
||
|
|
desc 'Forbid imports'
|
||
|
|
task 'import:disable' => :environment do |t|
|
||
|
|
SiteSetting.allow_import = false
|
||
|
|
puts 'Imports are now forbidden.'
|
||
|
|
end
|