Changeset 60

Timestamp:

01/15/05 12:28:54 (20 years ago)

Author:

jerome

Message:

Cleaned the code a bit

Files:

• pykoticon/trunk/bin/pykoticon (modified) (3 diffs)

pykoticon/trunk/bin/pykoticon

@@ -152 +152 @@
         """Retrieves the user account and quota information."""
         info = self.getUserInfo(username)
-        quotas = self.getUserPQuotas(username)
-        return User(username, info, quotas)
-    
-def getWindowsUserName() :    
-    """Retrieves the current user's name under MS Windows."""
-    return win32api.GetUserName()
+        if info :
+            quotas = self.getUserPQuotas(username)
+            return User(username, info, quotas)
     
 class PyKotIcon(wxPython.wx.wxFrame):
@@ -255 +252 @@
 def test() :
     """Runs in test mode (console)."""
-    username = getWindowsUserName()
+    if len(sys.argv) >= 3 :
+        username = sys.argv[2] 
+    else :    
+        username = win32api.GetUserName()
     net = CGINetworkInterface(DUMPYKOTA_URL)
     user = net.getUser(username)
@@ -270 +270 @@
 
 if __name__ == '__main__':
-    if (len(sys.argv) == 2) and (sys.argv[1] == "--test") :
+    if (len(sys.argv) >= 2) and (sys.argv[1] == "--test") :
         test()
     else :