Justin Edmund
c0922203a7
* Add table for data version and migrate * Modify migration and re-migrate * Create data_version.rb Adds a model for DataVersion * Add aws-sdk-s3 and create aws_service.rb AwsService handles streaming game image files from the Granblue Fantasy server to our S3 instance. * Add importers The Importer libraries take CSV data and import them into the database for each type. We currently support characters, summons and weapons. * Add downloaders Downloaders take Granblue IDs and download images for those items from the Granblue Fantasy server in all relevant sizes. Downloaders can download to disk or stream the file directly to S3. * Create data_importer.rb * Fetches a list of all CSV files present in the updates folder * Checks which have already been imported * Sends unimported data to the appropriate Importer to handle * Create download_manager.rb Creates an appropriate downloader for each Granblue ID it receives * Update download_images.rake Most of this task has been extracted into the Downloader libraries * Update import_data.rake * Create deploy.rake This task is to be run as a post-deploy script. It checks for new unimported data, imports it, then downloads the relevant images to S3 or local disk depending on the parameters provided. * Update credentials.yml.enc * Began working on a README and added example CSVs * Modify importer to handle updates This way we can also add FLBs and other uncaps easier. * Updates only require values that will change When updating a row, fields that don't have a provided value will not be changed * Rebuild search indices in post deploy * Clean up logs with LoggingHelper * More logging adjustments Trying to get a nice-looking output * Change some ASCII characters * Final ASCII changes * Fix issues with Summon and Weapon importers * Finish README for contributing
110 lines
3 KiB
Ruby
110 lines
3 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Granblue
|
|
class DataImporter
|
|
def initialize(test_mode: false, verbose: false)
|
|
@test_mode = test_mode
|
|
@verbose = verbose
|
|
@import_logs = []
|
|
end
|
|
|
|
def process_all_files(&block)
|
|
files = Dir.glob(Rails.root.join('db', 'seed', 'updates', '*.csv')).sort
|
|
|
|
files.each do |file|
|
|
if (new_records = import_csv(file))
|
|
block.call(new_records) if block_given?
|
|
end
|
|
end
|
|
|
|
print_summary if @test_mode
|
|
end
|
|
|
|
private
|
|
|
|
def import_csv(file_path)
|
|
filename = File.basename(file_path)
|
|
return if already_imported?(filename)
|
|
|
|
importer = create_importer(filename, file_path)
|
|
return unless importer
|
|
|
|
log_info "Processing #{filename} in #{@test_mode ? 'test' : 'live'} mode..."
|
|
result = importer.import
|
|
log_import(filename, result)
|
|
log_info "Successfully processed #{filename}"
|
|
result
|
|
end
|
|
|
|
def log_import_results(result)
|
|
return unless @verbose
|
|
|
|
result[:new].each do |type, ids|
|
|
log_info "Created #{ids.size} new #{type.pluralize}" if ids.any?
|
|
end
|
|
result[:updated].each do |type, ids|
|
|
log_info "Updated #{ids.size} existing #{type.pluralize}" if ids.any?
|
|
end
|
|
end
|
|
|
|
def create_importer(filename, file_path)
|
|
# This pattern matches both singular and plural: character(s), weapon(s), summon(s)
|
|
match = filename.match(/\A\d{8}-(character(?:s)?|weapon(?:s)?|summon(?:s)?)-\d+\.csv\z/)
|
|
return unless match
|
|
|
|
matched_type = match[1]
|
|
singular_type = matched_type.sub(/s$/, '')
|
|
importer_class = "Granblue::Importers::#{singular_type.capitalize}Importer".constantize
|
|
|
|
importer_class.new(
|
|
file_path,
|
|
test_mode: @test_mode,
|
|
verbose: @verbose,
|
|
logger: self
|
|
)
|
|
rescue NameError
|
|
log_info "No importer found for type: #{singular_type}"
|
|
nil
|
|
end
|
|
|
|
def already_imported?(filename)
|
|
DataVersion.imported?(filename)
|
|
end
|
|
|
|
def log_import(filename, result = nil)
|
|
return if @test_mode
|
|
|
|
DataVersion.mark_as_imported(filename)
|
|
|
|
if result && @verbose
|
|
result[:new].each do |type, ids|
|
|
log_info "Created #{ids.size} new #{type.pluralize}" if ids.any?
|
|
end
|
|
result[:updated].each do |type, ids|
|
|
log_info "Updated #{ids.size} existing #{type.pluralize}" if ids.any?
|
|
end
|
|
end
|
|
end
|
|
|
|
def log_operation(operation)
|
|
if @test_mode
|
|
@import_logs << operation
|
|
log_info "[TEST MODE] Would perform: #{operation}"
|
|
end
|
|
end
|
|
|
|
def print_summary
|
|
log_info "\nTest Mode Summary:"
|
|
log_info "Would perform #{@import_logs.size} operations"
|
|
if @import_logs.any?
|
|
log_info 'Sample of operations:'
|
|
@import_logs.first(3).each { |log| log_info "- #{log}" }
|
|
log_info '...' if @import_logs.size > 3
|
|
end
|
|
end
|
|
|
|
def log_info(message)
|
|
puts message if @verbose || @test_mode
|
|
end
|
|
end
|
|
end
|