Picking up on some stray areas

[gc-dialer] / tests / gv_samples / generate_gv_samples.py

diff --git a/tests/gv_samples/generate_gv_samples.py b/tests/gv_samples/generate_gv_samples.py
index 6ba1c30..5f1b399 100755 (executable)
--- a/tests/gv_samples/generate_gv_samples.py
+++ b/tests/gv_samples/generate_gv_samples.py
@@ -1,81 +1,22 @@
 #!/usr/bin/env python
 
-import sys
-import urllib
-import urllib2
-import traceback
-import warnings
+from __future__ import with_statement
+
+import logging
 
 import sys
+sys.path.append("/usr/lib/dialcentral")
 sys.path.append("../../src")
 
-import browser_emu
-import gv_backend
-
-webpages = [
-       ("login", gv_backend.GVDialer._loginURL),
-       ("clicktocall", gv_backend.GVDialer._clicktocallURL),
-       ("sendsms", gv_backend.GVDialer._sendSmsURL),
-       ("setforward", gv_backend.GVDialer._setforwardURL),
-       ("contacts", gv_backend.GVDialer._contactsURL),
-       ("contactdetails", gv_backend.GVDialer._contactDetailURL),
-       ("voicemail", gv_backend.GVDialer._voicemailURL),
-       ("sms", gv_backend.GVDialer._smsURL),
-       ("forward", gv_backend.GVDialer._forwardURL),
-       ("recent", gv_backend.GVDialer._recentCallsURL),
-       ("placed", gv_backend.GVDialer._placedCallsURL),
-       ("recieved", gv_backend.GVDialer._receivedCallsURL),
-       ("missed", gv_backend.GVDialer._missedCallsURL),
-]
-
-
-browser = browser_emu.MozillaEmulator(1)
-for name, url in webpages:
-       try:
-               page = browser.download(url)
-       except StandardError, e:
-               print e.message
-               continue
-       with open("not_loggedin_%s.txt" % name, "w") as f:
-               f.write(page)
-
-username = sys.argv[1]
-password = sys.argv[2]
-
-loginPostData = urllib.urlencode({
-       'Email' : username,
-       'Passwd' : password,
-       'service': "grandcentral",
-       "ltmpl": "mobile",
-       "btmpl": "mobile",
-       "PersistentCookie": "yes",
-})
+from backends import gv_backend
 
-try:
-       loginSuccessOrFailurePage = browser.download(gv_backend.GVDialer._loginURL, loginPostData)
-except urllib2.URLError, e:
-       warnings.warn(traceback.format_exc())
-       raise RuntimeError("%s is not accesible" % gv_backend.GVDialer._loginURL)
 
-forwardPage = browser.download(gv_backend.GVDialer._forwardURL)
+def main():
+       username = sys.argv[1]
+       password = sys.argv[2]
+       gv_backend.grab_debug_info(username, password)
 
-tokenGroup = gv_backend.GVDialer._tokenRe.search(forwardPage)
-if tokenGroup is None:
-       print forwardPage
-       raise RuntimeError("Could not extract authentication token from GoogleVoice")
-token = tokenGroup.group(1)
 
-browser = browser_emu.MozillaEmulator(1)
-for name, url in webpages:
-       try:
-               #data = urllib.urlencode({
-               #       "_rnr_se": token,
-               #})
-               #page = browser.download(url, data)
-               page = browser.download(url)
-       except StandardError, e:
-               warnings.warn(traceback.format_exc())
-               continue
-       print "Writing to file"
-       with open("loggedin_%s.txt" % name, "w") as f:
-               f.write(page)
+if __name__ == "__main__":
+       logging.basicConfig(level=logging.DEBUG)
+       main()