Merge pull request #3 from UPowr/google-page-changes

extend google-page-changes with support for dp method

aws_google_auth/google.py

@@ -304,15 +304,15 @@ def do_login_new(self, first_page):
         if "challenge/totp?" in sess.url:
             sess = self.handle_totp_new(sess, payload)
         elif "challenge/ipp?" in sess.url:
-            raise NotImplementedError('handle_dp not updated')
+            raise NotImplementedError('handle_ipp not updated')
         elif "challenge/az?" in sess.url:
-            raise NotImplementedError('handle_dp not updated')
+            raise NotImplementedError('handle_az not updated')
         elif "challenge/sk?" in sess.url:
-            raise NotImplementedError('handle_dp not updated')
+            raise NotImplementedError('handle_sk not updated')
         elif "challenge/iap?" in sess.url:
-            raise NotImplementedError('handle_dp not updated')
+            raise NotImplementedError('handle_iap not updated')
         elif "challenge/dp?" in sess.url:
-            raise NotImplementedError('handle_dp not updated')
+            sess = self.handle_dp_new(sess)
         elif "challenge/ootp/5" in sess.url:
             raise NotImplementedError(
                 'Offline Google App OOTP not implemented')
@@ -876,6 +876,24 @@ def handle_dp(self, sess):
         # Submit Configuration
         return self.post(challenge_url, data=payload)
 
+    def handle_dp_new(self, sess):
+        response_page = BeautifulSoup(sess.text, 'html.parser')
+
+        input("Check your phone - after you have confirmed response press ENTER to continue.") or None
+
+        form = response_page.find('form')
+        challenge_url = 'https://accounts.google.com' + form.get('action')
+
+        payload = {}
+        for tag in form.find_all('input'):
+            if tag.get('name') is None:
+                continue
+
+            payload[tag.get('name')] = tag.get('value')
+
+        # Submit Configuration
+        return self.post(challenge_url, data=payload)
+
     def handle_iap(self, sess):
         response_page = BeautifulSoup(sess.text, 'html.parser')
         challenge_url = sess.url.split("?")[0]