little bit tweaking

This commit is contained in:
ahmadbilalkhalid 2019-07-03 18:08:22 +05:00
parent ba2063ed9c
commit cd1efa16f2
3 changed files with 14 additions and 9 deletions

3
.gitignore vendored
View File

@ -1,2 +1,3 @@
.idea/
venv/
.vscode/
venv/

@ -1 +1 @@
Subproject commit 59974e2047bbea08b72d32129fd905bbea6e9d70
Subproject commit cb2a416a17d6789e613ba3b9957917770f4211e1

18
main.py
View File

@ -1,6 +1,7 @@
import os
import json
import subprocess
import shutil
from etcd3_wrapper import Etcd3Wrapper
BASE_PATH = "/var/www"
@ -10,6 +11,7 @@ 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")
@ -17,13 +19,15 @@ for image in images_to_be_created:
print("qcow2 format")
print("Converting it to raw")
subprocess.run(["qemu-img", "convert", "-f", "qcow2",
"-O", "raw", image_full_path, "image.raw"])
if os.path.isfile("image.raw"):
subprocess.run(["rbd", "import", "image.raw",
f"{image.value['store_name']}/{image.key.split('/')[-1]}"])
os.remove("image.raw")
"-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("image.raw not found")
print(f"{image_uuid}.raw not found")
else:
print("not qcow2 format")
else: