Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Generating file path using checksum address.

pull/632/head
jMyles 2019-01-10 14:30:06 -05:00
parent 4c33b02854
commit 9d356af5a8
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nucypher/config/storages.py
View File

@ -213,16 +213,11 @@ class ForgetfulNodeStorage(NodeStorage):
@validate_checksum_address @validate_checksum_address
def generate_certificate_filepath(self, def generate_certificate_filepath(self,
checksum_address: str = None, checksum_address: str = None) -> str:
host: str = None) -> str:
if not bool(checksum_address) ^ bool(host): prefix = '{}{}-'.format(self.__base_prefix, checksum_address)
message = "Either pass checksum_address or host; Not both. Got ({} {})".format(checksum_address, host)
raise ValueError(message)
prefix = '{}{}-'.format(self.__base_prefix, checksum_address or host)
temp_file = tempfile.NamedTemporaryFile(prefix=prefix, suffix=self.TLS_CERTIFICATE_EXTENSION, delete=False) temp_file = tempfile.NamedTemporaryFile(prefix=prefix, suffix=self.TLS_CERTIFICATE_EXTENSION, delete=False)
certificate = self.__certificates[checksum_address or host] certificate = self.__certificates[checksum_address]
certificate_bytes = certificate.public_bytes(self.TLS_CERTIFICATE_ENCODING) certificate_bytes = certificate.public_bytes(self.TLS_CERTIFICATE_ENCODING)
temp_file.write(certificate_bytes) temp_file.write(certificate_bytes)