用Python调用OpenStack API上传Glance镜像

用Python调用OpenStack API上传Glance镜像

import requests
import json

def get_auth_token(controller_ip, domain_name, username, password):
    try:
        url = f"http://{controller_ip}:5000/v3/auth/tokens"
        body = {
            "auth": {
                "identity": {
                    "methods": ['password'],
                    "password": {
                        "user": {
                            "domain": {"name": domain_name},
                            "name": username,
                            "password": password
                        }
                    }
                },
                "scope": {
                    "project": {
                        "domain": {"name": domain_name},
                        "name": username,
                        "password": password
                    }
                }
            }
        }
        headers = {
            "Content-type": "application/json"
        }
        response = requests.post(url, headers=headers, json=body)
        token = response.headers['X-Subject-Token']
        headers = {
            "X-Auth-Token": token
        }
        print(f"获取到的Token值为{token}")
        return headers
    except Exception as e:
        print(f"获取Token值失败原因是{e}")

class ImageManager:
    def __init__(self, headers, url):
        self.headers = headers
        self.url = url

    def CreateImage(self, name, disk_format, container_format):
        body = {
            "name": name,
            "disk_format": disk_format,
            "container_format": container_format,
        }
        response = requests.post(self.url, headers=self.headers, json=body)
        req = response.text
        return req

    def GetID(self, name):
        response = requests.get(self.url, headers=self.headers)
        req = json.loads(response.text)
        for image in req['images']:
            if image['name'] == name: 
                return image['id']
        return "NONE"

    def UploadImage(self, id, image_file):
        url = self.url + "/" + id + "/" + "file"
        self.headers["Content-Type"] = "application/octet-stream"
        req = requests.put(url, headers=self.headers, data=open(image_file, 'rb').read())
        return req

    def DeleteImage(self, id)
        url = self.url + "/" + id
        self.headers["Content-type"] = "application/octet-stream"
        req = requests.delete(url, headers=self.headers)
        return req

if __name__ == "__main__":
    controller_ip = "192.168.200.8"
    domain_name= "demo"
    username = "admin"
    password = "000000"
    headers = get_auth_token(controller_ip, domain_name, username, password)
    image_m = ImageManager(headers, f"http://{controller_ip}:9292/v2/images")

    create = image_m.CreateImage("cirros01", "qcow2", "bare")
    print(f"镜像创建信息为{create}")

    imageid = image_m.GetID("cirros01")
    print(f"镜像ID为{imageid}")

    imageupload = image_m.UploadImage(imageid, "/root/cirros-0.3.4-x86_64-disk.img")
    print(f"镜像上传信息为{imageupload}")
    test = input("镜像创建完成，请输入y删除此镜像，否则退出\n")
    if test == "y":
          imageDelete = image_m.DeleteImage(imageid)
          print(f"镜像已经删除。信息为：{imageDelete}")
    else:
        print("镜像删除失败")