martlet 0.0.2 → 0.0.3

data/lib/martlet/client.rb

@@ -6,15 +6,25 @@ module Martlet
     
     def initialize(email, password)
       @agent = Mechanize.new
-      @email = email
+      @email = format_email(email)
       
       authenticator = Authenticator.new(@agent)
-      authenticator.authenticate(email, password)
+      authenticator.authenticate(@email, password)
     end
     
     def grades
       transcript = Transcript.new(@agent)
       transcript.fetch_grades
     end
+    
+    private
+    
+    def format_email(email)
+      if email.include?('@')
+        email
+      else
+        "#{email}@mail.mcgill.ca"
+      end
+    end
   end
 end

data/lib/martlet/version.rb

@@ -1,3 +1,3 @@
 module Martlet
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: martlet
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors: