def save_certs(pathname, certs): try: if not os.path.exists(pathname): os.makedirs(pathname) except Exception as e: logger.error(e) for index, cert in enumerate(certs): filename = "/".join([pathname.rstrip("/"), "cert%d.pem" % index]) save_to_file(filename, cert.replace("\\n", '\n'))
def _update_haproxy(self, cfg): if self.running_mode in ["acs"]: if Haproxy.cls_cfg != cfg: logger.info("HAProxy configuration:\n%s" % cfg) Haproxy.cls_cfg = cfg if save_to_file(config.HAPROXY_CONFIG_FILE, cfg): Haproxy.cls_process = UpdateHelper.run_reload( Haproxy.cls_process) elif self.ssl_updated: logger.info("SSL certificates have been changed") Haproxy.cls_process = UpdateHelper.run_reload( Haproxy.cls_process) else: logger.info("HAProxy configuration remains unchanged") logger.info("===========END===========") elif self.running_mode in ["legacy"]: logger.info("HAProxy configuration:\n%s" % cfg) if save_to_file(config.HAPROXY_CONFIG_FILE, cfg): UpdateHelper.run_once()
def test_save_to_file_with_error(self): self.assertFalse(save_to_file("/", "something"))
def test_save_to_file(self): content = str(uuid.uuid4()) self.assertTrue(save_to_file(self.f.name, content)) txt = open(self.f.name).read() self.assertEqual(content, txt)
def create_pid_file(): pid = str(os.getpid()) save_to_file(config.PID_FILE, pid) return pid