File: 04_labels.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (49 lines) | stat: -rw-r--r-- 1,410 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# frozen_string_literal: true

class Gitlab::Seeder::GroupLabels
  def initialize(group, label_per_group: 10)
    @group = group
    @label_per_group = label_per_group
  end

  def seed!
    @label_per_group.times do
      label_title = FFaker::Product.brand
      Labels::CreateService
        .new(title: label_title, color: "#{::Gitlab::Color.color_for(label_title)}")
        .execute(group: @group)
      print '.'
    end
  end
end

class Gitlab::Seeder::ProjectLabels
  def initialize(project, label_per_project: 5)
    @project = project
    @label_per_project = label_per_project
  end

  def seed!
    @label_per_project.times do
      label_title = FFaker::Vehicle.model
      Labels::CreateService
        .new(title: label_title, color: "#{::Gitlab::Color.color_for(label_title)}")
        .execute(project: @project)
      print '.'
    end
  end
end

Gitlab::Seeder.quiet do
  label_per_group = 10
  puts "\nGenerating group labels: #{Group.not_mass_generated.count * label_per_group}"
  Group.not_mass_generated.find_each do |group|
    Gitlab::Seeder::GroupLabels.new(group, label_per_group: label_per_group).seed!
  end

  label_per_project = 5
  puts "\nGenerating project labels: #{Project.not_mass_generated.count * label_per_project}"
  Project.not_mass_generated.find_each do |project|
    Gitlab::Seeder::ProjectLabels.new(project, label_per_project: label_per_project).seed!
  end
end