From 1bf3827921fad04b417b26ae54fb1d2b60041822 Mon Sep 17 00:00:00 2001
From: steffen123 <steffen@schaumburger.info>
Date: Wed, 7 Jul 2010 08:17:18 +0200
Subject: [PATCH] copied bit of error handling from Carl

---
 pyfpdb/ImapSummaries.py | 83 ++++++++++++++++++++++-------------------
 1 file changed, 45 insertions(+), 38 deletions(-)

diff --git a/pyfpdb/ImapSummaries.py b/pyfpdb/ImapSummaries.py
index c8e651b1..50079488 100755
--- a/pyfpdb/ImapSummaries.py
+++ b/pyfpdb/ImapSummaries.py
@@ -30,46 +30,53 @@ def splitPokerStarsSummaries(emailText):
     return splitSummaries
 #end def emailText
 
-#TODO: move all these into the config file. until then usage is: ./ImapSummaries.py YourImapHost YourImapUser YourImapPw 
-configHost=sys.argv[1]
-configUser=sys.argv[2]
-configPw=sys.argv[3]
-#TODO: specify folder, whether to use SSL
+if __name__ == '__main__':
+    #TODO: move all these into the config file. until then usage is: ./ImapSummaries.py YourImapHost YourImapUser YourImapPw 
+    configHost=sys.argv[1]
+    configUser=sys.argv[2]
+    configPw=sys.argv[3]
+    #TODO: specify folder, whether to use SSL
 
-server = IMAP4_SSL(configHost) #TODO: optionally non-SSL
-response = server.login(configUser, configPw) #TODO catch authentication error
-#print "response to logging in:",response
-#print "server.list():",server.list() #prints list of folders
+    try:
+        server = IMAP4_SSL(configHost) #TODO: optionally non-SSL
+        response = server.login(configUser, configPw) #TODO catch authentication error
+        #print "response to logging in:",response
+        #print "server.list():",server.list() #prints list of folders
 
-response = server.select("INBOX")
-#print "response to selecting INBOX:",response
-if response[0]!="OK":
-    raise error #TODO: show error message
+        response = server.select("INBOX")
+        #print "response to selecting INBOX:",response
+        if response[0]!="OK":
+            raise error #TODO: show error message
 
-neededMessages=[]
-response, searchData = server.search(None, "SUBJECT", "PokerStars Tournament History Request")
-for messageNumber in searchData[0].split(" "):
-    response, headerData = server.fetch(messageNumber, "(BODY[HEADER.FIELDS (SUBJECT)])")
-    #print "response to fetch subject:",response
-    if response!="OK":
-        raise error #TODO: show error message
-    if headerData[1].find("Subject: PokerStars Tournament History Request - Last x")!=1:
-        neededMessages.append(("PS", messageNumber))
+        neededMessages=[]
+        response, searchData = server.search(None, "SUBJECT", "PokerStars Tournament History Request")
+        for messageNumber in searchData[0].split(" "):
+            response, headerData = server.fetch(messageNumber, "(BODY[HEADER.FIELDS (SUBJECT)])")
+            #print "response to fetch subject:",response
+            if response!="OK":
+                raise error #TODO: show error message
+            if headerData[1].find("Subject: PokerStars Tournament History Request - Last x")!=1:
+                neededMessages.append(("PS", messageNumber))
         
-if (len(neededMessages)==0):
-    raise error #TODO: show error message
-for messageData in neededMessages:
-    response, bodyData = server.fetch(messageData[1], "(UID BODY[TEXT])")
-    bodyData=bodyData[0][1]
-    if response!="OK":
-        raise error #TODO: show error message
-    if messageData[0]=="PS":
-        summaryTexts=(splitPokerStarsSummaries(bodyData))
-        for summaryText in summaryTexts:
-            result=PokerStarsSummary.PokerStarsSummary(sitename="PokerStars", gametype=None, summaryText=summaryText, builtFrom = "IMAP")
-            #print "result:",result
-            #TODO: count results and output to shell like hand importer does
+        if (len(neededMessages)==0):
+            raise error #TODO: show error message
+        for messageData in neededMessages:
+            response, bodyData = server.fetch(messageData[1], "(UID BODY[TEXT])")
+            bodyData=bodyData[0][1]
+            if response!="OK":
+                raise error #TODO: show error message
+            if messageData[0]=="PS":
+                summaryTexts=(splitPokerStarsSummaries(bodyData))
+                for summaryText in summaryTexts:
+                    result=PokerStarsSummary.PokerStarsSummary(sitename="PokerStars", gametype=None, summaryText=summaryText, builtFrom = "IMAP")
+                    #print "result:",result
+                    #TODO: count results and output to shell like hand importer does
             
-print "completed running Imap import, closing server connection"
-server.close()
-server.logout()
+        print "completed running Imap import, closing server connection"
+    finally:
+        try:
+            server.close()
+        finally:
+            pass
+        server.logout()
+        
\ No newline at end of file