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