summaryrefslogtreecommitdiff
path: root/app/models
diff options
context:
space:
mode:
Diffstat (limited to 'app/models')
-rw-r--r--app/models/invite_code.rb5
1 files changed, 3 insertions, 2 deletions
diff --git a/app/models/invite_code.rb b/app/models/invite_code.rb
index 30a6498..8f69582 100644
--- a/app/models/invite_code.rb
+++ b/app/models/invite_code.rb
@@ -1,4 +1,3 @@
-require 'coupon_code'
class InviteCode < CouchRest::Model::Base
use_database 'invite_codes'
@@ -14,9 +13,11 @@ class InviteCode < CouchRest::Model::Base
def initialize(attributes = {}, options = {})
super(attributes, options)
- write_attribute('invite_code', CouponCode.generate) if new?
end
+ def set_invite_code(code)
+ write_attribute(:invite_code, code)
+ end
end