mirror of https://github.com/pulumi/pulumi.git
45 lines
1.6 KiB
Go
45 lines
1.6 KiB
Go
// Copyright 2022-2024, Pulumi Corporation.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package python
|
|
|
|
import "fmt"
|
|
|
|
// Provides code for a method which will be placed in the program preamble if deemed
|
|
// necessary. Because many tasks in Go such as reading a file require extensive error
|
|
// handling, it is much prettier to encapsulate that error handling boilerplate as its
|
|
// own function in the preamble.
|
|
func getHelperMethodIfNeeded(functionName string, indent string) (string, bool) {
|
|
switch functionName {
|
|
case "filebase64sha256":
|
|
return `def computeFilebase64sha256(path):
|
|
fileData = open(path).read().encode()
|
|
hashedData = hashlib.sha256(fileData.encode()).digest()
|
|
return base64.b64encode(hashedData).decode()`, true
|
|
case "notImplemented":
|
|
return fmt.Sprintf(`
|
|
%sdef not_implemented(msg):
|
|
%s raise NotImplementedError(msg)`, indent, indent), true
|
|
case "singleOrNone":
|
|
return fmt.Sprintf(
|
|
`%sdef single_or_none(elements):
|
|
%s if len(elements) != 1:
|
|
%s raise Exception("single_or_none expected input list to have a single element")
|
|
%s return elements[0]
|
|
`, indent, indent, indent, indent), true
|
|
default:
|
|
return "", false
|
|
}
|
|
}
|