diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/tasks/invite_code.rake | 21 |
1 files changed, 15 insertions, 6 deletions
diff --git a/lib/tasks/invite_code.rake b/lib/tasks/invite_code.rake index f3bafac..6623640 100644 --- a/lib/tasks/invite_code.rake +++ b/lib/tasks/invite_code.rake @@ -1,16 +1,25 @@ desc "Generate a batch of invite codes" -task :generate_invites, [:n] => :environment do |task, args| +task :generate_invites, [:n, :u] => :environment do |task, args| - codes = args.n - codes = codes.to_i + codes = args.n + codes = codes.to_i - codes.times do |x| + if args.u == nil + max_uses = 1 + + elsif + max_uses = args.u + max_uses = max_uses.to_i + end + + codes.times do |x| x = InviteCode.new + x.invite_max_uses = max_uses x.save - puts "#{x.invite_code} Code generated." - + puts "#{x.invite_code} Code generated with #{x.invite_max_uses} uses." end + end |