ucloud-image-scanner/main.py

34 lines
No EOL
1.4 KiB
Python

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")