/usr/lib/python3/dist-packages/artifacts/writer.py is in python3-artifacts 20161022-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 | # -*- coding: utf-8 -*-
"""The artifact writer objects."""
import abc
import json
import yaml
class BaseArtifactsWriter(object):
"""Class that implements the artifacts writer interface."""
@abc.abstractmethod
def WriteArtifactsFile(self, artifacts, filename):
"""Writes artifact definitions to a file.
Args:
artifacts: a list of ArtifactDefinition objects to be written.
filename: the filename to write artifacts to.
"""
@abc.abstractmethod
def FormatArtifacts(self, artifacts):
"""Formats artifacts to desired output format.
Args:
artifacts: an ArtifactDefinition instance or list of ArtifactDefinitions.
Returns:
formatted string of artifact definition.
"""
class ArtifactWriter(BaseArtifactsWriter):
"""Class that implements the artifacts writer interface."""
def WriteArtifactsFile(self, artifacts, filename):
"""Writes artifact definitions to a file.
Args:
artifacts: a list of ArtifactDefinition objects to be written.
filename: the filename to write artifacts to.
"""
with open(filename, 'w') as file_object:
file_object.write(self.FormatArtifacts(artifacts))
class JsonArtifactsWriter(ArtifactWriter):
"""Class that implements the JSON artifacts writer interface."""
def FormatArtifacts(self, artifacts):
"""Formats artifacts to desired output format.
Args:
artifacts: a list of ArtifactDefinitions.
Returns:
formatted string of artifact definition.
"""
artifact_definitions = [artifact.AsDict() for artifact in artifacts]
json_data = json.dumps(artifact_definitions)
return json_data
class YamlArtifactsWriter(ArtifactWriter):
"""Class that implements the YAML artifacts writer interface."""
def FormatArtifacts(self, artifacts):
"""Formats artifacts to desired output format.
Args:
artifacts: a list of ArtifactDefinitions.
Returns:
formatted string of artifact definition.
"""
# TODO: improve output formatting of yaml
artifact_definitions = [artifact.AsDict() for artifact in artifacts]
yaml_data = yaml.safe_dump_all(artifact_definitions)
return yaml_data
|