From 76cc852527a117e1cbff7f2ea80afa9033abb422 Mon Sep 17 00:00:00 2001
From: ankonym <ankonym@gmail.com>
Date: Thu, 8 Oct 2015 15:37:27 +0200
Subject: Adjust the rake task to make id = invite code

This will simplify invite code deletion after running tests.
---
 lib/tasks/invite_code.rake | 21 +++++++++++++--------
 1 file changed, 13 insertions(+), 8 deletions(-)

(limited to 'lib')

diff --git a/lib/tasks/invite_code.rake b/lib/tasks/invite_code.rake
index f3bafac..b379d3a 100644
--- a/lib/tasks/invite_code.rake
+++ b/lib/tasks/invite_code.rake
@@ -1,16 +1,21 @@
-
+require 'base64'
+require 'securerandom'
 
 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
+  if args.u != nil
+    max_uses = args.u
+  end
 
-    codes.times do |x|
+  codes.times do |x|
     x = InviteCode.new
+    x.max_uses = max_uses
     x.save
-    puts "#{x.invite_code} Code generated."
-
+    puts x.invite_code
   end
-end
 
+end
-- 
cgit v1.2.3