[pyKoL-users] SF.net SVN: pykol: [18] kolsite.py

[<mi...@us...>]

Revision: 18
          http://pykol.svn.sourceforge.net/pykol/?rev=18&view=rev
Author:   misza13
Date:     2007-05-04 14:56:36 -0700 (Fri, 04 May 2007)

Log Message:
-----------
Added temporary cookie storage.

Modified Paths:
--------------
    kolsite.py

Modified: kolsite.py
===================================================================
--- kolsite.py	2007-05-04 21:03:07 UTC (rev 17)
+++ kolsite.py	2007-05-04 21:56:36 UTC (rev 18)
@@ -90,6 +90,20 @@
 
 
     def doLogin(self, nick, password):
+        try:
+            cookie = file('cookie','r').readlines()
+            self.cookie = cookie[1][:-1]
+            self.hostname = cookie[0][:-1]
+        except:
+            self.cookie = None
+
+        if self.cookie:
+            print 'Trying a recovered cookie...'
+            response, data = self.getPage('main.php')
+            if data or response.getheader('location')=='fight.php':
+                print "Successfully recovered cookie file."
+                return
+
         challenge = self.getChallenge()
 
         formFields = {}
@@ -108,6 +122,8 @@
 
         if response.status == 302:
             self.cookie = response.getheader('set-cookie')
+            file('cookie','w+').write('%s\n%s\n' % (self.hostname,self.cookie))
+            print 'Saved cookie file.'
             return
 
         if re.search('Bad password.',data):


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.