ucloud-image-scanner/main.py

import os
import json
import subprocess
import shutil
from etcd3_wrapper import Etcd3Wrapper

BASE_PATH = "/var/www"

client = Etcd3Wrapper()
images = list(client.get_prefix("/v1/image/", value_in_json=True))
images_to_be_created = list(filter(lambda e: e.value["status"] == "TO_BE_CREATED", images))

for image in images_to_be_created:
    image_uuid = image.key.split("/")[-1]
    image_full_path = f"{BASE_PATH}/{image.value['owner']}/{image.value['filename']}"
    if os.path.isfile(image_full_path):
        output = subprocess.check_output(["qemu-img", "info", image_full_path]).decode("utf-8")
        if "qcow2" in output:
            print("qcow2 format")
            print("Converting it to raw")
            subprocess.run(["qemu-img", "convert", "-f", "qcow2",
                            "-O", "raw", image_full_path, f"{image_uuid}.raw"])
            if os.path.isfile(f"{image_uuid}.raw"):
                shutil.move(f"{image_uuid}.raw", f"/var/vm/{image_uuid}.raw")
                # subprocess.run(["rbd", "import", "image.raw",
                #                f"{image.value['store_name']}/{image.key.split('/')[-1]}"])
                image.value["status"] = "CREATED"
                client.put(image.key, json.dumps(image.value))
            else:
                print(f"{image_uuid}.raw not found")
        else:
            print("not qcow2 format")
    else:
        print("File does not exists")