Files
nairobi/db/dev_seeds/polls.rb
2025-10-16 11:09:34 +02:00

292 lines
10 KiB
Ruby

section "Creating polls" do
def create_poll!(attributes)
poll = Poll.create!(attributes.merge(starts_at: 1.day.from_now, ends_at: 2.days.from_now))
poll.update_columns(
starts_at: attributes[:starts_at].beginning_of_minute,
ends_at: attributes[:ends_at].beginning_of_minute
)
end
create_poll!(name: I18n.t("seeds.polls.current_poll"),
slug: I18n.t("seeds.polls.current_poll").parameterize,
starts_at: 7.days.ago,
ends_at: 7.days.from_now,
geozone_restricted: false)
create_poll!(name: I18n.t("seeds.polls.current_poll_geozone_restricted"),
slug: I18n.t("seeds.polls.current_poll_geozone_restricted").parameterize,
starts_at: 5.days.ago,
ends_at: 5.days.from_now,
geozone_restricted_to: Geozone.sample(3))
create_poll!(name: I18n.t("seeds.polls.recounting_poll"),
slug: I18n.t("seeds.polls.recounting_poll").parameterize,
starts_at: 15.days.ago,
ends_at: 2.days.ago)
create_poll!(name: I18n.t("seeds.polls.expired_poll_without_stats"),
slug: I18n.t("seeds.polls.expired_poll_without_stats").parameterize,
starts_at: 2.months.ago,
ends_at: 1.month.ago)
create_poll!(name: I18n.t("seeds.polls.expired_poll_with_stats"),
slug: I18n.t("seeds.polls.expired_poll_with_stats").parameterize,
starts_at: 2.months.ago,
ends_at: 1.month.ago,
results_enabled: true,
stats_enabled: true)
Poll.find_each do |poll|
name = poll.name
Setting.enabled_locales.map do |locale|
Globalize.with_locale(locale) do
poll.name = "#{name} (#{locale})"
poll.summary = "Summary for locale #{locale}"
poll.description = "Description for locale #{locale}"
end
end
poll.save!
end
end
section "Creating Poll Questions & Options" do
Poll.find_each do |poll|
(3..5).to_a.sample.times do
vote_type = VotationType.vote_types.keys.sample
question_title = Faker::Lorem.sentence(word_count: 3).truncate(60) + "?"
question = Poll::Question.new(author: User.sample,
title: question_title,
poll: poll)
Setting.enabled_locales.map do |locale|
Globalize.with_locale(locale) do
question.title = "#{question_title} (#{locale})"
end
end
question.save!
question.create_votation_type!(vote_type: vote_type, max_votes: (3 if vote_type == "multiple"))
if question.accepts_options?
Faker::Lorem.words(number: (2..4).to_a.sample).each_with_index do |title, index|
description = "<p>#{Faker::Lorem.paragraphs.join("</p><p>")}</p>"
option = Poll::Question::Option.new(question: question,
title: title.capitalize,
description: description,
given_order: index + 1)
Setting.enabled_locales.map do |locale|
Globalize.with_locale(locale) do
option.title = "#{title} (#{locale})"
option.description = "#{description} (#{locale})"
end
end
option.save!
end
end
end
end
end
section "Creating Poll Booths & BoothAssignments" do
20.times do |i|
Poll::Booth.create(name: "Booth #{i}",
location: Faker::Address.street_address,
polls: [Poll.sample])
end
end
section "Creating Poll Shifts for Poll Officers" do
Poll.find_each do |poll|
Poll::BoothAssignment.where(poll: poll).find_each do |booth_assignment|
scrutiny = (poll.ends_at.to_datetime..poll.ends_at.to_datetime + Poll::RECOUNT_DURATION)
Poll::Officer.find_each do |poll_officer|
{
vote_collection: (poll.starts_at.to_datetime..poll.ends_at.to_datetime),
recount_scrutiny: scrutiny
}.each do |task_name, task_dates|
task_dates.each do |shift_date|
Poll::Shift.create(booth: booth_assignment.booth,
officer: poll_officer,
date: shift_date,
officer_name: poll_officer.name,
officer_email: poll_officer.email,
task: task_name)
end
end
end
end
end
end
section "Commenting Polls" do
30.times do
author = User.sample
poll = Poll.sample
Comment.create!(user: author,
created_at: rand(poll.created_at..Time.current),
commentable: poll,
body: Faker::Lorem.sentence)
end
end
section "Creating Poll Voters" do
def vote_poll_on_booth(user, poll)
officer = Poll::Officer.sample
Poll::Voter.create!(document_type: user.document_type,
document_number: user.document_number,
user: user,
poll: poll,
origin: "booth",
officer: officer,
officer_assignment: officer.officer_assignments.sample,
booth_assignment: poll.booth_assignments.sample)
end
def vote_poll_on_web(user, poll)
randomly_answer_questions(poll, user)
Poll::Voter.create!(document_type: user.document_type,
document_number: user.document_number,
user: user,
poll: poll,
origin: "web")
end
def randomly_answer_questions(poll, user)
poll.questions.each do |question|
next unless [true, false].sample
if question.accepts_options?
option = question.question_options.sample
Poll::Answer.create!(question_id: question.id,
author: user,
option: option,
answer: option.title)
else
text = Faker::Lorem.sentence(word_count: (6..14).to_a.sample)
Poll::Answer.create!(question_id: question.id, author: user, answer: text)
end
end
end
(Poll.expired + Poll.current + Poll.recounting).uniq.each do |poll|
verified_users = User.level_two_or_three_verified
if poll.geozone_restricted?
verified_users = verified_users.where(geozone_id: poll.geozone_ids)
end
user_groups = verified_users.in_groups(2)
user_groups.first.each { |user| vote_poll_on_booth(user, poll) }
user_groups.second.compact.each { |user| vote_poll_on_web(user, poll) }
end
end
section "Creating Poll Recounts" do
Poll.find_each do |poll|
poll.booth_assignments.each do |booth_assignment|
officer_assignment = poll.officer_assignments.first
author = Poll::Officer.first.user
total_amount = white_amount = null_amount = 0
booth_assignment.voters.count.times do
case rand
when 0...0.1 then null_amount += 1
when 0.1...0.2 then white_amount += 1
else total_amount += 1
end
end
Poll::Recount.create!(officer_assignment: officer_assignment,
booth_assignment: booth_assignment,
author: author,
date: poll.ends_at,
white_amount: white_amount,
null_amount: null_amount,
total_amount: total_amount,
origin: "booth")
end
end
end
section "Creating Poll Results" do
Poll.find_each do |poll|
poll.booth_assignments.each do |booth_assignment|
officer_assignment = poll.officer_assignments.first
author = Poll::Officer.first.user
poll.questions.each do |question|
question.question_options.each do |option|
Poll::PartialResult.create!(officer_assignment: officer_assignment,
booth_assignment: booth_assignment,
date: Date.current,
question: question,
answer: option.title,
author: author,
amount: rand(999),
origin: "booth")
end
end
end
end
end
section "Creating Poll Questions from Proposals" do
3.times do
proposal = Proposal.sample
poll = Poll.current.first
question = Poll::Question.new(poll: poll)
question.copy_attributes_from_proposal(proposal)
question_title = question.title
Setting.enabled_locales.map do |locale|
Globalize.with_locale(locale) do
question.title = "#{question_title} (#{locale})"
end
end
question.save!
Faker::Lorem.words(number: (2..4).to_a.sample).each_with_index do |title, index|
description = "<p>#{Faker::ChuckNorris.fact}</p>"
option = Poll::Question::Option.new(question: question,
title: title.capitalize,
description: description,
given_order: index + 1)
Setting.enabled_locales.map do |locale|
Globalize.with_locale(locale) do
option.title = "#{title} (#{locale})"
option.description = "#{description} (#{locale})"
end
end
option.save!
end
question.create_votation_type!(vote_type: "unique")
end
end
section "Creating Successful Proposals" do
10.times do
proposal = Proposal.sample
poll = Poll.current.first
question = Poll::Question.new(poll: poll)
question.copy_attributes_from_proposal(proposal)
question_title = question.title
Setting.enabled_locales.map do |locale|
Globalize.with_locale(locale) do
question.title = "#{question_title} (#{locale})"
end
end
question.save!
Faker::Lorem.words(number: (2..4).to_a.sample).each_with_index do |title, index|
description = "<p>#{Faker::ChuckNorris.fact}</p>"
option = Poll::Question::Option.new(question: question,
title: title.capitalize,
description: description,
given_order: index + 1)
Setting.enabled_locales.map do |locale|
Globalize.with_locale(locale) do
option.title = "#{title} (#{locale})"
option.description = "#{description} (#{locale})"
end
end
option.save!
end
end
end