Exemple #1
0
    def test_hash_file(self):
        with TemporaryDirectory() as temp_dir:
            path = Path(temp_dir) / 'temp.txt'
            path.write_text('abcdef')
            expected = 'bef57ec7f53a6d40beb640a780a639c83bc29ac8a9816f1fc6c5c6dcd93c4721'

            actual = utils.hash_file(path)
            self.assertEqual(actual, expected)
Exemple #2
0
def secure_hash(env, rel_path):
    """
    Return a secure hash of the contents of the file, using SHA-256.

    Returns the hash as a hexadecimal string.

    Args:
      env: a Jinja2 Environment object.
      rel_path: a path relative to the output directory configured in the
        Jinja2 Environment object. This can be any path-like object.
    """
    path = get_output_path(env, rel_path)
    sha = utils.hash_file(path)

    return sha
def format_output(
    output_dir,
    build_time,
    election_title=None,
    results_title=None,
    rel_home_page=None,
    zip_file_path=None,
    initial_data=None,
):
    """
    Print and return the output data.
    """
    if initial_data is None:
        initial_data = {}

    output_dir = Path(output_dir)
    # Rename the variable for clarity.
    output_data = initial_data

    if election_title is not None:
        output_data['election_title'] = election_title

    if results_title is not None:
        output_data['results_title'] = results_title

    if rel_home_page is not None:
        output_data['rel_home_page'] = rel_home_page

    if zip_file_path is not None:
        path = output_dir / zip_file_path
        zip_size = path.stat().st_size
        zip_hash = utils.hash_file(path)

        output_data['zip_file'] = {
            'path': str(zip_file_path),
            'bytes': zip_size,
            'hash': str(zip_hash),
        }

    output_data.update(
        build_time=build_time.isoformat(),
        output_dir=str(output_dir),
    )

    # TODO: allow changing the stdout output format (e.g. YAML or text)?
    output = json.dumps(output_data, **DEFAULT_JSON_DUMPS_ARGS)

    return (output_data, output)