def create_temp_file(self, new_code=None): self.folder = Files.temp_folder('tmp_lambda_') if new_code: self.lambda_code = new_code self.tmp_file = Files.path_combine(self.folder, '{0}.py'.format(self.file_name)) Files.write(self.tmp_file, self.lambda_code) assert Files.exists(self.tmp_file) return self
def __init__(self,lambda_name): self.lambda_name = lambda_name self.aws_lambda = Lambda(self.lambda_name) self.s3_bucket = AWS_Config().lambda_s3_bucket() self.s3_key = f'{AWS_Config().lambda_s3_folder_lambdas()}/{self.lambda_name}.zip' self.role_arn = Temp_Aws_Roles().for_lambda_invocation__role_arn() self.tmp_folder = Files.temp_folder('tmp_lambda_') (self.aws_lambda.set_s3_bucket (self.s3_bucket ) .set_s3_key (self.s3_key ) .set_role (self.role_arn ) .set_folder_code (self.tmp_folder ))
def test_temp_folder(self): assert Files.exists(Files.temp_folder()) assert 'aa_' in Files.temp_folder('_bb', 'aa_', '/tmp')
def __enter__(self): if Files.exists(self.zip_file): if self.target_folder is None: self.target_folder = Files.temp_folder("unzipped_") return Files.unzip_file(self.zip_file, self.target_folder) return None
def __init__(self, contents='...', extension='tmp'): self.tmp_file = random_filename(extension) self.tmp_folder = Files.temp_folder() self.file_path = Files.path_combine(self.tmp_folder, self.tmp_file) self.contents = contents