comparison easyhg2.py @ 434:f8aa7ac57993

Refactor load/save config

     """
     parsed_url = urlparse.urlparse(authuri)
     return "%s://%s%s" % (parsed_url.scheme, parsed_url.netloc,
                           parsed_url.path)
 
+def load_config(pfile):
+    fp = None
+    try:
+        fp = open(pfile)
+    except:
+        self.ui.write("failed to open authfile %s\n" % pfile)
+    if fp:
+        pcfg.readfp(fp)
+    fp.close()
+
+def save_config(pcfg, pfile):
+    ofp = None
+    try:
+        ofp = open(pfile, 'w')
+    except:
+        self.ui.write("failed to open authfile %s for writing\n" % pfile)
+        raise
+    try:
+        os.fchmod(ofp.fileno(), stat.S_IRUSR | stat.S_IWUSR) #!!! Windows equivalent?
+    except:
+        ofp.close()
+        ofp = None
+        self.ui.write("failed to set proper permissions on authfile %s\n" % pfile)
+        raise
+    pcfg.write(ofp)
+    ofp.close()
+
 @monkeypatch_method(passwordmgr)
 def find_user_password(self, realm, authuri):
 
     if not self.ui.interactive():
         return orig_find(self, realm, authuri)

 
     uri = canonical_url(authuri)
 
     pkey = base64.b64encode('%s@@%s' % (uri, user)).replace('=', '_')
     pekey = self.ui.config('easyhg', 'authkey')
-    pfile = os.path.expanduser(self.ui.config('easyhg', 'authfile'))
+    pfile = self.ui.config('easyhg', 'authfile')
+    if pfile: pfile = os.path.expanduser(pfile)
     pdata = None
 
     self.ui.write("pekey is %s\n" % pekey)
     self.ui.write("pfile is %s\n" % pfile)
 

     pcfg = None
 
     if pekey and pfile:
         # load pwd from our cache file, decrypt with given key
         pcfg = ConfigParser.RawConfigParser()
-        fp = None
         remember_default = False
+        load_config(pcfg, pfile)
         try:
-            fp = open(pfile)
+            remember_default = pcfg.getboolean('preferences', 'remember')
         except:
-            self.ui.write("failed to open authfile %s\n" % pfile)
-        if fp and not passwd:
-            pcfg.readfp(fp)
-            try:
-                remember_default = pcfg.getboolean('preferences', 'remember')
-            except:
-                remember_default = False
-            try:
-                pdata = pcfg.get('auth', pkey)
-            except ConfigParser.NoOptionError:
-                pdata = None
-            if pdata:
-                cachedpwd = decrypt(pdata, pekey)
-                passfield.setText(cachedpwd)
-        fp.close()
+            remember_default = False
+        try:
+            pdata = pcfg.get('auth', pkey)
+        except ConfigParser.NoOptionError:
+            pdata = None
+        if pdata:
+            cachedpwd = decrypt(pdata, pekey)
+            passfield.setText(cachedpwd)
         remember = QtGui.QCheckBox()
         remember.setChecked(remember_default)
         remember.setText(_('Remember this password until EasyMercurial exits'))
         layout.addWidget(remember, 3, 1)
 

         user = userfield.text()
         passwd = passfield.text()
 
         if pekey and pfile and remember:
 
-            ofp = None
-
-            try:
-                ofp = open(pfile, 'w')
-            except:
-                self.ui.write("failed to open authfile %s for writing\n" % pfile)
-                raise
-
-            try:
-                os.fchmod(ofp.fileno(), stat.S_IRUSR | stat.S_IWUSR) #!!! Windows equivalent?
-            except:
-                ofp.close()
-                ofp = None
-                self.ui.write("failed to set proper permissions on authfile %s\n" % pfile)
-                raise
-
             #!!! add these sections first...
 
             if remember.isChecked():
                 pdata = encrypt(passwd, pekey)
                 pcfg.set('auth', pkey, pdata)
             else:
                 pcfg.set('auth', pkey, '')
 
             pcfg.set('preferences', 'remember', remember.isChecked())
-            pcfg.write(ofp)
-            ofp.close()
+
+            save_config(pcfg, pfile)
 
 #        if passwd and keyring_key != '' and not from_keyring:
 #            keyring_key = '%s@@%s' % (uri, user)
 ##            keyring.set_password('Mercurial', keyring_key, passwd)
         self.add_password(realm, authuri, user, passwd)