root/lang/python/pit/pit.py @ 37388

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import os, yaml, tempfile
from subprocess import Popen

class Pit:
    VERSION   = "0.0.1"
    DIRECTORY = os.path.expanduser('~/.pit')
    _config = os.path.join(DIRECTORY, 'pit.yaml')
    _profile = os.path.join(DIRECTORY, 'default.yaml')
    
    @staticmethod
    def set(name, opts={}):
        profile = Pit._load()
        result = {}
        if opts.has_key('data'):
            result = opts['data']
        else:
            if not os.environ.has_key('EDITOR'):
                return {}
            
            temp_fd, path = tempfile.mkstemp()
            t = os.fdopen(temp_fd, "w")
            c = yaml.dump(opts['config'] if opts.has_key('config') else Pit.get(name) ,default_flow_style=False)
            t.write(c)
            t.close()
            Popen([os.environ['EDITOR'],path]).communicate()
            t = open(path)
            result = t.read()
            t.close()
            os.remove(path)

            if result == c:
                print 'No Changes'
                print profile
                return profile[name]

            result = yaml.load(result)

        profile[name] = result
        yaml.dump(profile,
                  open(Pit._profile, 'w'),
                  default_flow_style=False)
        return result

    @staticmethod
    def get(name, opts={}):
        load_data = Pit._load()
        ret = load_data[name] if load_data.has_key(name) else {} 
        if opts.has_key('require'):
            keys = set(opts['require'].keys()) - set(ret.keys())
            if keys:
                for key in keys:
                    ret[key] = opts['require'][key] 
                ret = Pit.set(name,{'config' : ret})
        
        return ret or {'username' : '', 'password' : ''}

    @staticmethod
    def switch(name, opts={}):
        Pit._profile = os.path.join(Pit.DIRECTORY, '%s.yaml' % name)
        config = Pit.config()
        ret = config['profile']
        config['profile'] = name
        yaml.dump(config,
                  open(Pit._config, 'w'),
                  default_flow_style=False)
        return ret

    @staticmethod
    def _load():
        if not os.path.exists(Pit.DIRECTORY):
            os.mkdir(Pit.DIRECTORY)
            os.chmod(Pit.DIRECTORY, 0700)

        if not os.path.exists(Pit._config):
            yaml.dump({'profile' : 'default'},
                      open(Pit._config, 'w'),
                      default_flow_style=False)
            os.chmod(Pit._config, 0600)

        Pit.switch(Pit.config()['profile'])

        if not os.path.exists(Pit._profile):
            yaml.dump({}, 
                      open(Pit._profile, 'w'), 
                      default_flow_style=False)
            os.chmod(Pit._profile, 0600)
        return yaml.load(open(Pit._profile)) or {}

    @staticmethod
    def config():
        return yaml.load(open(Pit._config))

if __name__ == '__main__':
    config = Pit.get('34twitter.com',{'require': {'email':'your email','password':'your password'}})
    print config
    print config['email']
    print config['password']