新增：导入另外一个平台项目数据脚本
This commit is contained in:
parent
4a12ab2d90
commit
dd4b21115f
|
|
@ -0,0 +1,138 @@
|
|||
|
||||
desc "导入导出issue、version、journal数据"
|
||||
namespace :import_from_another_forge do
|
||||
desc "导出数据"
|
||||
# 执行示例 bundle exec rake "import_from_another_forge:export[110]"
|
||||
# RAILS_ENV=production bundle exec rake "import_from_another_forge:export[110]"
|
||||
task :export, [:project_id] => :environment do |t, args|
|
||||
project_id = args.project_id
|
||||
project = Project.find_by_id project_id
|
||||
Axlsx::Package.new do |p|
|
||||
p.workbook.add_worksheet(:name => 'version') do |sheet|
|
||||
sheet.add_row ['id', 'name', 'description', 'effective_date', 'created_on', 'updated_on', 'status', 'creator','issues_count', 'closed_issues_count', 'percent']
|
||||
project.versions.each do |version|
|
||||
sheet.add_row [version.id, version.name, version.description, version.effective_date, version.created_on.strftime("%Y-%m-%d %H:%M:%S"), version.updated_on.strftime("%Y-%m-%d %H:%M:%S"), version.status, version.user.try(:login), version.issues_count, version.closed_issues_count, version.percent]
|
||||
end
|
||||
end
|
||||
p.workbook.add_worksheet(:name => 'issue') do |sheet|
|
||||
sheet.add_row ['id', 'project_issues_index', 'subject', 'description', 'creator', 'created_on', 'changer', 'updated_on', 'status_id', 'assigners', 'priority_id', 'issue_tags', 'version_id']
|
||||
project.issues.issue_issue.each do |issue|
|
||||
sheet.add_row [issue.id, issue.project_issues_index, issue.subject, issue.description, issue.user.try(:login), issue.created_on.strftime("%Y-%m-%d %H:%M:%S"), issue.changer.try(:login), issue.updated_on.strftime("%Y-%m-%d %H:%M:%S"), issue.status_id, issue.assigners.pluck(:login).join(","),issue.priority_id, issue.issue_tags.pluck(:name, :color).join(","), issue.fixed_version_id]
|
||||
end
|
||||
end
|
||||
issue_ids = project.issues.issue_issue
|
||||
p.workbook.add_worksheet(:name => 'journal') do |sheet|
|
||||
sheet.add_row ['id', 'journalized_type', 'journalized_id', 'creator', 'notes', 'created_on', 'parent_id', 'comments_count', 'reply_id', 'updated_on', 'operate_by']
|
||||
Journal.where(journalized_type: 'Issue', journalized_id: issue_ids).where.not(notes: nil).each do |journal|
|
||||
sheet.add_row [journal.id, journal.journalized_type, journal.journalized_id, journal.user.try(:login), journal.notes, journal.created_on.strftime("%Y-%m-%d %H:%M:%S"), journal.parent_id, journal.comments_count, journal.reply_id, journal.updated_on.strftime("%Y-%m-%d %H:%M:%S"), journal.operate_by]
|
||||
end
|
||||
end
|
||||
p.serialize('public/version_issue_journal_data.xlsx')
|
||||
end
|
||||
end
|
||||
|
||||
def find_or_create_by(login)
|
||||
return nil unless login.present?
|
||||
user = User.find_by(login: login)
|
||||
return user if user.present?
|
||||
|
||||
user = User.new(admin: false, login: login, mail: "#{login}@forge.com", nickname: login, platform: 'forge' , type: "User")
|
||||
user.password = "12345678"
|
||||
user.activate
|
||||
interactor = Gitea::RegisterInteractor.call({username: login, email: "#{login}@forge.com", password: "12345678"})
|
||||
gitea_user = interactor.result
|
||||
result = Gitea::User::GenerateTokenService.call(username, password)
|
||||
user.gitea_token = result['sha1']
|
||||
user.gitea_uid = gitea_user[:body]['id']
|
||||
UserExtension.create!(user_id: user.id) if user.save!
|
||||
|
||||
return user
|
||||
rescue
|
||||
return nil
|
||||
end
|
||||
|
||||
# 执行示例 bundle exec rake "import_from_another_forge:import[filepath, 365, ceshi_org]"
|
||||
# RAILS_ENV=production bundle exec rake "import_from_another_forge:import[public/version_issue_journal_data.xlsx, 110]"
|
||||
task :import, [:attachment_uuid, :project_id] => :environment do |t, args|
|
||||
attachment_uuid = args.attachment_uuid
|
||||
project_id = args.project_id
|
||||
a =Attachment.find_by(uuid: attachment_uuid)
|
||||
project = Project.find_by_id project_id
|
||||
version_hash = {}
|
||||
issue_hash = {}
|
||||
journal_hash = {}
|
||||
ActiveRecord::Base.transaction do
|
||||
doc = SimpleXlsxReader.open("#{Rails.root}/tmp/files/#{a.relative_path_filename}")
|
||||
doc.sheets.each do |sheet|
|
||||
case sheet.name
|
||||
when 'version'
|
||||
sheet.rows.each.with_index do |row, index|
|
||||
next if index == 0
|
||||
version = Version.new(project_id: project_id)
|
||||
version.name = row[1]
|
||||
version.description = row[2]
|
||||
version.effective_date = row[3]
|
||||
version.created_on = row[4]
|
||||
version.updated_on = row[5]
|
||||
version.status = row[6]
|
||||
version.user = find_or_create_by(row[7])
|
||||
version.issues_count = row[8].to_i
|
||||
version.closed_issues_count = row[9].to_i
|
||||
version.percent = row[10].to_f
|
||||
version.save!
|
||||
version_hash["#{row[0]}"] = version.id
|
||||
end
|
||||
when 'issue'
|
||||
sheet.rows.each.with_index do |row, index|
|
||||
next if index == 0
|
||||
issue = Issue.new(issue_classify: "issue", project_id: project_id, tracker_id: Tracker.first.id)
|
||||
issue.project_issues_index = row[1]
|
||||
issue.subject = row[2]
|
||||
issue.description = row[3]
|
||||
issue.user = find_or_create_by(row[4])
|
||||
issue.created_on = row[5]
|
||||
issue.changer = find_or_create_by(row[6])
|
||||
issue.updated_on = row[7]
|
||||
issue.status_id = row[8]
|
||||
if row[9].present?
|
||||
row[9].split(',').each do |a|
|
||||
u = find_or_create_by(a)
|
||||
next unless u.present?
|
||||
issue.assigners << u
|
||||
end
|
||||
end
|
||||
issue.priority_id = row[10]
|
||||
issue.issue_tags
|
||||
if row[11].present?
|
||||
row[11].split(',').each_slice(2).to_a.each do |t|
|
||||
tag = IssueTag.find_by(project_id: project_id, name: t[0])
|
||||
if tag.present?
|
||||
issue.issue_tags << tag
|
||||
else
|
||||
tag = IssueTag.create(project_id: project_id, name: t[0], color: t[1])
|
||||
issue.issue_tags << tag
|
||||
end
|
||||
end
|
||||
end
|
||||
issue.fixed_version_id = row[12]
|
||||
issue.save!
|
||||
issue_hash["#{row[0]}"] = issue.id
|
||||
when 'journal'
|
||||
sheet.rows.each.with_index do |row, index|
|
||||
next if index == 0
|
||||
next if row[6].present? || row[8].present?
|
||||
journal = Journal.new
|
||||
journal.journalized_type = row[1]
|
||||
journal.journalized_id = row[2]
|
||||
journal.user = find_or_create_by(row[3])
|
||||
journal.notes = row[4]
|
||||
journal.created_on = row[5]
|
||||
journal.updated_on = row[9]
|
||||
journal.save!
|
||||
journal_hash["#{row[0]}"] = journal.id
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
end
|
||||
Loading…
Reference in New Issue