/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to bzrlib/config.py

  • Committer: Parth Malwankar
  • Date: 2010-07-21 03:07:02 UTC
  • mto: (5050.3.14 2.2)
  • mto: This revision was merged to the branch mainline in revision 5365.
  • Revision ID: parth.malwankar@gmail.com-20100721030702-ud1x2wcs6um57fhr
copy config file ownership only if a new file is created

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/config.py
74
74
 
75
75
import bzrlib
76
76
from bzrlib import (
 
77
    atomicfile,
77
78
    debug,
78
79
    errors,
79
80
    mail_client,
257
258
 
258
259
        Something similar to 'Martin Pool <mbp@sourcefrog.net>'
259
260
 
260
 
        $BZR_EMAIL can be set to override this (as well as the
261
 
        deprecated $BZREMAIL), then
 
261
        $BZR_EMAIL can be set to override this, then
262
262
        the concrete policy type is checked, and finally
263
263
        $EMAIL is examined.
264
 
        If none is found, a reasonable default is (hopefully)
265
 
        created.
 
264
        If no username can be found, errors.NoWhoami exception is raised.
266
265
 
267
266
        TODO: Check it's reasonably well-formed.
268
267
        """
278
277
        if v:
279
278
            return v.decode(osutils.get_user_encoding())
280
279
 
281
 
        name, email = _auto_user_id()
282
 
        if name:
283
 
            return '%s <%s>' % (name, email)
284
 
        else:
285
 
            return email
 
280
        raise errors.NoWhoami()
 
281
 
 
282
    def ensure_username(self):
 
283
        """Raise errors.NoWhoami if username is not set.
 
284
 
 
285
        This method relies on the username() function raising the error.
 
286
        """
 
287
        self.username()
286
288
 
287
289
    def signature_checking(self):
288
290
        """What is the current policy for signature checking?."""
476
478
    def _get_nickname(self):
477
479
        return self.get_user_option('nickname')
478
480
 
 
481
    def _write_config_file(self):
 
482
        filename = self._get_filename()
 
483
        file_existed = os.path.isfile(filename)
 
484
        atomic_file = atomicfile.AtomicFile(filename)
 
485
        self._get_parser().write(atomic_file)
 
486
        atomic_file.commit()
 
487
        atomic_file.close()
 
488
        if not file_existed:
 
489
            osutils.copy_ownership_from_path(filename)
 
490
 
479
491
 
480
492
class GlobalConfig(IniBasedConfig):
481
493
    """The configuration that should be used for a specific location."""
517
529
        self._get_parser().setdefault(section, {})[option] = value
518
530
        self._write_config_file()
519
531
 
520
 
    def _write_config_file(self):
521
 
        path = self._get_filename()
522
 
        f = open(path, 'wb')
523
 
        osutils.copy_ownership_from_path(path)
524
 
        self._get_parser().write(f)
525
 
        f.close()
526
 
 
527
532
 
528
533
class LocationConfig(IniBasedConfig):
529
534
    """A configuration object that gives the policy for a location."""
663
668
        self._get_parser()[location][option]=value
664
669
        # the allowed values of store match the config policies
665
670
        self._set_option_policy(location, option, store)
666
 
        self._get_parser().write(file(self._get_filename(), 'wb'))
 
671
        self._write_config_file()
667
672
 
668
673
 
669
674
class BranchConfig(Config):
841
846
                                  ' or HOME set')
842
847
        return osutils.pathjoin(base, 'bazaar', '2.0')
843
848
    else:
844
 
        # cygwin, linux, and darwin all have a $HOME directory
845
849
        if base is None:
846
850
            base = os.path.expanduser("~")
847
851
        return osutils.pathjoin(base, ".bazaar")
899
903
        return os.path.expanduser('~/.cache')
900
904
 
901
905
 
902
 
def _auto_user_id():
903
 
    """Calculate automatic user identification.
904
 
 
905
 
    Returns (realname, email).
906
 
 
907
 
    Only used when none is set in the environment or the id file.
908
 
 
909
 
    This previously used the FQDN as the default domain, but that can
910
 
    be very slow on machines where DNS is broken.  So now we simply
911
 
    use the hostname.
912
 
    """
913
 
    import socket
914
 
 
915
 
    if sys.platform == 'win32':
916
 
        name = win32utils.get_user_name_unicode()
917
 
        if name is None:
918
 
            raise errors.BzrError("Cannot autodetect user name.\n"
919
 
                                  "Please, set your name with command like:\n"
920
 
                                  'bzr whoami "Your Name <name@domain.com>"')
921
 
        host = win32utils.get_host_name_unicode()
922
 
        if host is None:
923
 
            host = socket.gethostname()
924
 
        return name, (name + '@' + host)
925
 
 
926
 
    try:
927
 
        import pwd
928
 
        uid = os.getuid()
929
 
        try:
930
 
            w = pwd.getpwuid(uid)
931
 
        except KeyError:
932
 
            raise errors.BzrCommandError('Unable to determine your name.  '
933
 
                'Please use "bzr whoami" to set it.')
934
 
 
935
 
        # we try utf-8 first, because on many variants (like Linux),
936
 
        # /etc/passwd "should" be in utf-8, and because it's unlikely to give
937
 
        # false positives.  (many users will have their user encoding set to
938
 
        # latin-1, which cannot raise UnicodeError.)
939
 
        try:
940
 
            gecos = w.pw_gecos.decode('utf-8')
941
 
            encoding = 'utf-8'
942
 
        except UnicodeError:
943
 
            try:
944
 
                encoding = osutils.get_user_encoding()
945
 
                gecos = w.pw_gecos.decode(encoding)
946
 
            except UnicodeError:
947
 
                raise errors.BzrCommandError('Unable to determine your name.  '
948
 
                   'Use "bzr whoami" to set it.')
949
 
        try:
950
 
            username = w.pw_name.decode(encoding)
951
 
        except UnicodeError:
952
 
            raise errors.BzrCommandError('Unable to determine your name.  '
953
 
                'Use "bzr whoami" to set it.')
954
 
 
955
 
        comma = gecos.find(',')
956
 
        if comma == -1:
957
 
            realname = gecos
958
 
        else:
959
 
            realname = gecos[:comma]
960
 
        if not realname:
961
 
            realname = username
962
 
 
963
 
    except ImportError:
964
 
        import getpass
965
 
        try:
966
 
            user_encoding = osutils.get_user_encoding()
967
 
            realname = username = getpass.getuser().decode(user_encoding)
968
 
        except UnicodeDecodeError:
969
 
            raise errors.BzrError("Can't decode username as %s." % \
970
 
                    user_encoding)
971
 
 
972
 
    return realname, (username + '@' + socket.gethostname())
973
 
 
974
 
 
975
906
def parse_username(username):
976
907
    """Parse e-mail username and return a (name, address) tuple."""
977
908
    match = re.match(r'(.*?)\s*<?([\w+.-]+@[\w+.-]+)>?', username)
1063
994
        """Save the config file, only tests should use it for now."""
1064
995
        conf_dir = os.path.dirname(self._filename)
1065
996
        ensure_config_dir_exists(conf_dir)
1066
 
        self._get_config().write(file(self._filename, 'wb'))
 
997
        f = file(self._filename, 'wb')
 
998
        try:
 
999
            self._get_config().write(f)
 
1000
        finally:
 
1001
            f.close()
1067
1002
 
1068
1003
    def _set_option(self, section_name, option_name, value):
1069
1004
        """Set an authentication configuration option"""
1517
1452
            return StringIO()
1518
1453
 
1519
1454
    def _get_configobj(self):
1520
 
        return ConfigObj(self._get_config_file(), encoding='utf-8')
 
1455
        f = self._get_config_file()
 
1456
        try:
 
1457
            return ConfigObj(f, encoding='utf-8')
 
1458
        finally:
 
1459
            f.close()
1521
1460
 
1522
1461
    def _set_configobj(self, configobj):
1523
1462
        out_file = StringIO()