コード例 #1
0
ファイル: config_actor.py プロジェクト: zerolugithub/beeswarm
    def _get_zmq_keys(self, id):
        cert_path = os.path.join(self.work_dir, 'certificates')
        public_keys = os.path.join(cert_path, 'public_keys')
        private_keys = os.path.join(cert_path, 'private_keys')
        public_key_path = os.path.join(public_keys, '{0}.pub'.format(id))
        private_key_path = os.path.join(private_keys, '{0}.pri'.format(id))

        if not os.path.isfile(public_key_path) or not os.path.isfile(
                private_key_path):
            logging.debug('Generating ZMQ keys for: {0}.'.format(id))
            for _path in [cert_path, public_keys, private_keys]:
                if not os.path.isdir(_path):
                    os.mkdir(_path)

            tmp_key_dir = tempfile.mkdtemp()
            try:
                public_key, private_key = create_certificates(tmp_key_dir, id)
                # the final location for keys
                shutil.move(public_key, public_key_path)
                shutil.move(private_key, private_key_path)
            finally:
                shutil.rmtree(tmp_key_dir)

        # return copy of keys
        return open(private_key_path,
                    "r").readlines(), open(public_key_path, "r").readlines()
コード例 #2
0
ファイル: config_actor.py プロジェクト: nghiemnv/beeswarm
    def _generate_zmq_keys(self, key_name):
        cert_path = os.path.join(self.work_dir, 'certificates')
        public_keys = os.path.join(cert_path, 'public_keys')
        private_keys = os.path.join(cert_path, 'private_keys')
        for _path in [cert_path, public_keys, private_keys]:
            if not os.path.isdir(_path):
                os.mkdir(_path)

        tmp_key_dir = tempfile.mkdtemp()
        try:
            public_key, private_key = create_certificates(tmp_key_dir, key_name)
            # the final location for keys
            public_key_final = os.path.join(public_keys, '{0}.pub'.format(key_name))
            private_key_final = os.path.join(private_keys, '{0}.pri'.format(key_name))
            shutil.move(public_key, public_key_final)
            shutil.move(private_key, private_key_final)
        finally:
            shutil.rmtree(tmp_key_dir)

        # return copy of keys
        return open(private_key_final, "r").readlines(), open(public_key_final, "r").readlines()
コード例 #3
0
ファイル: config_actor.py プロジェクト: cmsmith1977/beeswarm
    def _get_zmq_keys(self, id):
        cert_path = os.path.join(self.work_dir, 'certificates')
        public_keys = os.path.join(cert_path, 'public_keys')
        private_keys = os.path.join(cert_path, 'private_keys')
        public_key_path = os.path.join(public_keys, '{0}.pub'.format(id))
        private_key_path = os.path.join(private_keys, '{0}.pri'.format(id))

        if not os.path.isfile(public_key_path) or not os.path.isfile(private_key_path):
            logging.debug('Generating ZMQ keys for: {0}.'.format(id))
            for _path in [cert_path, public_keys, private_keys]:
                if not os.path.isdir(_path):
                    os.mkdir(_path)

            tmp_key_dir = tempfile.mkdtemp()
            try:
                public_key, private_key = create_certificates(tmp_key_dir, id)
                # the final location for keys
                shutil.move(public_key, public_key_path)
                shutil.move(private_key, private_key_path)
            finally:
                shutil.rmtree(tmp_key_dir)

        # return copy of keys
        return open(private_key_path, "r").readlines(), open(public_key_path, "r").readlines()