blender-docker/main.py

import os
import pathlib
import json
import docker

# TODO Update readme file badge with blender version
# TODO add a check if the image build are stored in the docker register


def get_data_from_json() -> (str, str):
    """From the json file, get blender version (major and minor), and return this information."""
    file = pathlib.Path(os.getcwd(), 'build.json')

    with open(file, 'r') as f:
        data = json.load(f)
        f.close()

    version_values = data['tags'][0]
    major, minor = version_values['b3d_vs_major'], version_values['b3d_vs_minor']
    print(f'1 -> Setup the package with blender version "{major}.{minor}"')

    return major, minor


def build_docker(b3d_version: (str, str), latest: bool = False):
    """
    Generate docker image, request the blender version (major, minor) with the keyword b3d_version, and can be
    tagged with latest or only blender version info.
    """
    client = docker.from_env()
    if os.environ.get('docker_hub_password'):
        client.login(username=os.environ['docker_hub_user'], password=os.environ['docker_hub_password'])

    # Init all data to build the docker image
    args = {
        'b3d_vs_major': str(b3d_version[0]),
        'b3d_vs_minor': str(b3d_version[1]),
    }
    docker_repo = 'stilobique/blender'
    if latest:
        tag = 'latest'
    else:
        tag = f'{args["b3d_vs_major"]}.{args["b3d_vs_minor"]}'
    completed_tag_name = f'{docker_repo}:{args["b3d_vs_major"]}.{args["b3d_vs_minor"]}'

    print('2 -> Build from image file')
    client.images.build(path=os.getcwd(), buildargs=args, tag=completed_tag_name)

    print('3 -> Push the new images')
    client.images.push(repository=docker_repo, tag=tag)


if __name__ == "__main__":
    version = get_data_from_json()
    build_docker(b3d_version=version)
    build_docker(b3d_version=version, latest=True)