mirror of
https://github.com/discourse/discourse.git
synced 2024-12-13 13:13:44 +08:00
61 lines
1.5 KiB
Ruby
61 lines
1.5 KiB
Ruby
|
# frozen_string_literal: true
|
||
|
|
||
|
require 'discourse_dev/record'
|
||
|
require 'rails'
|
||
|
require 'faker'
|
||
|
|
||
|
module DiscourseDev
|
||
|
class Category < Record
|
||
|
|
||
|
def initialize
|
||
|
super(::Category, DiscourseDev.config.category[:count])
|
||
|
@parent_category_ids = ::Category.where(parent_category_id: nil).pluck(:id)
|
||
|
end
|
||
|
|
||
|
def data
|
||
|
name = Faker::Discourse.unique.category
|
||
|
parent_category_id = nil
|
||
|
|
||
|
if Faker::Boolean.boolean(true_ratio: 0.6)
|
||
|
offset = Faker::Number.between(from: 0, to: @parent_category_ids.count - 1)
|
||
|
parent_category_id = @parent_category_ids[offset]
|
||
|
@permissions = ::Category.find(parent_category_id).permissions_params.presence
|
||
|
else
|
||
|
@permissions = nil
|
||
|
end
|
||
|
|
||
|
{
|
||
|
name: name,
|
||
|
description: Faker::Lorem.paragraph,
|
||
|
user_id: ::Discourse::SYSTEM_USER_ID,
|
||
|
color: Faker::Color.hex_color.last(6),
|
||
|
parent_category_id: parent_category_id
|
||
|
}
|
||
|
end
|
||
|
|
||
|
def permissions
|
||
|
return @permissions if @permissions.present?
|
||
|
return { everyone: :full } if Faker::Boolean.boolean(true_ratio: 0.75)
|
||
|
|
||
|
permission = {}
|
||
|
group = Group.random
|
||
|
permission[group.id] = Faker::Number.between(from: 1, to: 3)
|
||
|
|
||
|
permission
|
||
|
end
|
||
|
|
||
|
def create!
|
||
|
super do |category|
|
||
|
category.set_permissions(permissions)
|
||
|
category.save!
|
||
|
|
||
|
@parent_category_ids << category.id if category.parent_category_id.blank?
|
||
|
end
|
||
|
end
|
||
|
|
||
|
def self.random
|
||
|
super(::Category)
|
||
|
end
|
||
|
end
|
||
|
end
|