<commit_before>import os
import stat
import shutil
import datetime
import ConfigParser

from awsident.identity import Identity
from awsident.storage import identity_store

class ConfigHandler(object):
    """Handler for getting and setting config file credentials

    """
    conf_filename = None
    section_name = 'default'
    attr_map = {
        'access_key_id':'aws_access_key_id',
        'secret_access_key':'aws_secret_access_key',
    }
    @classmethod
    def save_identity(cls, identity=None):
        """Saves any currently configured credentials to an :class:`Identity`

        """
        if identity is None:
            now = datetime.datetime.now()
            name = now.strftime('Saved Identity - %Y%m%d-%H%M%S')
            identity = Identity(name=name)
        for _cls in ConfigHandler.__subclasses__():
            _cls()._save_identity(identity)
        identity_store.add_identity(identity)
        return identity
    @classmethod
    def change_identity(cls, identity):
        """Sets all config files to the credentials of the provided :class:`Identity`

        """
        if not isinstance(identity, Identity):
            identity = identity_store.get(identity)
        assert identity is not None
        for _cls in ConfigHandler.__subclasses__():
            _cls()._change_identity(identity)
        return identity
    def _save_identity(self, identity):
        self._handle_identity_save(identity)
    def _change_identity(self, identity):
        self.handle_paths()
        self._handle_identity_change(identity)
        self.handle_perms()
    def handle_paths(self):
        fn = os.path.expanduser(self.conf_filename)
        bak_fn = '.'.join([fn, 'bak'])
        if os.path.exists(fn) and not os.path.exists(bak_fn):
            shutil.copy2(fn, bak_fn)
        if os.path.dirname(self.conf_filename) != '~':
            dirname = os.path.dirname(fn)
            if not os.path.exists(dirname):
                os.makedirs(dirname)
                os.chmod(dirname, stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR)
    def load_config(self):
        fn = os.path.expanduser(self.conf_filename)
        p = ConfigParser.SafeConfigParser()
        if os.path.exists(fn):
            p.read(fn)
        if self.section_name.lower() != 'default':
            if not p.has_section(self.section_name):
                p.add_section(self.section_name)
        return p
    def iter_conf_vals(self, identity=None):
        val = None
        for ident_key, option_name in self.attr_map.items():
            if identity is not None:
                val = getattr(identity, ident_key)
            yield option_name, ident_key, val
    def _handle_identity_save(self, identity):
        p = self.load_config()
        for option_name, ident_key, val in self.iter_conf_vals(identity):
            if val is not None:
                continue
            try:
                val = p.get(self.section_name, option_name)
            except ConfigParser.NoOptionError:
                val = None
            if not val:
                continue
            setattr(identity, ident_key, val)
    def _handle_identity_change(self, identity):
        p = self.load_config()
        fn = os.path.expanduser(self.conf_filename)
        for option_name, ident_key, val in self.iter_conf_vals(identity):
            p.set(self.section_name, option_name, val)
        with open(fn, 'w') as f:
            p.write(f)
    def handle_perms(self):
        fn = os.path.expanduser(self.conf_filename)
        os.chmod(fn, stat.S_IRUSR | stat.S_IWUSR)

class BotoConfigHandler(ConfigHandler):
    conf_filename = '~/.boto'
    section_name = 'Credentials'

class S3CmdConfigHandler(ConfigHandler):
    conf_filename = '~/.s3cfg'
    attr_map = {
        'access_key_id':'access_key',
        'secret_access_key':'secret_key',
    }

class AwsConfigHandler(ConfigHandler):
    conf_filename = '~/.aws/credentials'
<commit_msg>Use proper get/set methods if section_name is 'default'<commit_after>import os
import stat
import shutil
import datetime
import ConfigParser

from awsident.identity import Identity
from awsident.storage import identity_store

class ConfigHandler(object):
    """Handler for getting and setting config file credentials

    """
    conf_filename = None
    section_name = 'default'
    attr_map = {
        'access_key_id':'aws_access_key_id',
        'secret_access_key':'aws_secret_access_key',
    }
    @classmethod
    def save_identity(cls, identity=None):
        """Saves any currently configured credentials to an :class:`Identity`

        """
        if identity is None:
            now = datetime.datetime.now()
            name = now.strftime('Saved Identity - %Y%m%d-%H%M%S')
            identity = Identity(name=name)
        for _cls in ConfigHandler.__subclasses__():
            _cls()._save_identity(identity)
        identity_store.add_identity(identity)
        return identity
    @classmethod
    def change_identity(cls, identity):
        """Sets all config files to the credentials of the provided :class:`Identity`

        """
        if not isinstance(identity, Identity):
            identity = identity_store.get(identity)
        assert identity is not None
        for _cls in ConfigHandler.__subclasses__():
            _cls()._change_identity(identity)
        return identity
    def _save_identity(self, identity):
        self._handle_identity_save(identity)
    def _change_identity(self, identity):
        self.handle_paths()
        self._handle_identity_change(identity)
        self.handle_perms()
    def handle_paths(self):
        fn = os.path.expanduser(self.conf_filename)
        bak_fn = '.'.join([fn, 'bak'])
        if os.path.exists(fn) and not os.path.exists(bak_fn):
            shutil.copy2(fn, bak_fn)
        if os.path.dirname(self.conf_filename) != '~':
            dirname = os.path.dirname(fn)
            if not os.path.exists(dirname):
                os.makedirs(dirname)
                os.chmod(dirname, stat.S_IRUSR | stat.S_IWUSR | stat.S_IXUSR)
    def load_config(self):
        fn = os.path.expanduser(self.conf_filename)
        p = ConfigParser.SafeConfigParser()
        if os.path.exists(fn):
            p.read(fn)
        if self.section_name.lower() != 'default':
            if not p.has_section(self.section_name):
                p.add_section(self.section_name)
        return p
    def iter_conf_vals(self, identity=None):
        val = None
        for ident_key, option_name in self.attr_map.items():
            if identity is not None:
                val = getattr(identity, ident_key)
            yield option_name, ident_key, val
    def _handle_identity_save(self, identity):
        p = self.load_config()
        for option_name, ident_key, val in self.iter_conf_vals(identity):
            if val is not None:
                continue
            if self.section_name.lower() == 'default':
                val = p.defaults().get(option_name)
            else:
                try:
                    val = p.get(self.section_name, option_name)
                except ConfigParser.NoOptionError:
                    val = None
            if not val:
                continue
            setattr(identity, ident_key, val)
    def _handle_identity_change(self, identity):
        p = self.load_config()
        fn = os.path.expanduser(self.conf_filename)
        for option_name, ident_key, val in self.iter_conf_vals(identity):
            if self.section_name.lower() == 'default':
                p.defaults()[option_name] = val
            else:
                p.set(self.section_name, option_name, val)
        with open(fn, 'w') as f:
            p.write(f)
    def handle_perms(self):
        fn = os.path.expanduser(self.conf_filename)
        os.chmod(fn, stat.S_IRUSR | stat.S_IWUSR)

class BotoConfigHandler(ConfigHandler):
    conf_filename = '~/.boto'
    section_name = 'Credentials'

class S3CmdConfigHandler(ConfigHandler):
    conf_filename = '~/.s3cfg'
    attr_map = {
        'access_key_id':'access_key',
        'secret_access_key':'secret_key',
    }

class AwsConfigHandler(ConfigHandler):
    conf_filename = '~/.aws/credentials'

{'fraction_non_alphanumeric': 0.0993409812057603, 'fraction_numerical': 0.000732243104710764, 'mean_word_length': 2.350367947669665, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '72297', 'n_tokens_mistral': 2427, 'n_tokens_neox': 2395, 'n_words': 527}