From af61dde196674d52258e506c9fadb565e9cd343b Mon Sep 17 00:00:00 2001
From: lulzette <lulzette@cock.li>
Date: Wed, 24 Mar 2021 19:05:34 +0000
Subject: [PATCH] init test

---
 main.py     | 78 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 snapshot.py | 58 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 136 insertions(+)
 create mode 100644 main.py
 create mode 100644 snapshot.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..592b858
--- /dev/null
+++ b/main.py
@@ -0,0 +1,78 @@
+#!/usr/bin/python3
+
+import time
+import argparse
+import libvirt
+from lxml import etree
+from lxml.builder import E
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(description='Create snapshot for libvirt machine')
+    parser.add_argument('--name', required=True, help='Machine name')
+    parser.add_argument('--device', required=True, help='Device path (like vda)')
+    return parser.parse_args()
+
+def create_snapshot_node(disk_node):
+    device = disk_node.xpath('/domain/devices/disk/target/@dev')[0]
+    disk_path = disk_node.xpath('/domain/devices/disk/source/@file')[0]
+
+    now = int(time.time())
+    snap_path = disk_path.rsplit('.', 1)[0]
+    snap_path = f'{snap_path}-{now}.snap'
+    snap_name = snap_path.rsplit('/', 1)[1]
+
+    snap = E.domainsnapshot()
+    snap.append(E.name(snap_name))
+
+    disks = E.disks()
+    disk = E.disk(name=device)
+    disk.append(E.source(file=snap_path))
+    disks.append(disk)
+
+    snap.append(disks)
+    return snap
+
+def get_disk_by_name(vm, name):
+    raw_xml = vm.XMLDesc(0)
+    xml = etree.XML(raw_xml)
+    disks = xml.xpath(f'/domain/devices/disk/target[@dev="{name}"]/..')
+    return disks[0]
+
+def main():
+    # args = parse_args()
+    session = libvirt.open('qemu:///system')
+    tgtDom = 1
+    from time import sleep
+    for i in session.listDefinedDomains():
+        print(i)
+        dom = session.lookupByName(i)
+        print("Is active? "+ str(dom.isActive()))
+        dom.create()
+        sleep(15)
+
+        print(session.listDomainsID())
+        print("Active? "+str(dom.isActive()))
+        sleep(15)
+        dom.shutdown()
+        sleep(15)
+        print("Active? "+str(dom.isActive()))
+
+    # print(session.getHostname())
+
+
+
+    # try:
+    #     vm = session.lookupByName(args.name)
+    # except libvirt.libvirtError:
+    #     print(f'Machine with name {args.name} not found!')
+    #     session.close()
+    #     return
+    # disk = get_disk_by_name(vm, args.device)
+    # snapshot = create_snapshot_node(disk)
+    # vm.snapshotCreateXML(etree.tostring(snapshot, encoding='unicode'),
+    #         libvirt.VIR_DOMAIN_SNAPSHOT_CREATE_DISK_ONLY | libvirt.VIR_DOMAIN_SNAPSHOT_CREATE_ATOMIC)
+    session.close()
+
+if __name__ == '__main__':
+    main()
diff --git a/snapshot.py b/snapshot.py
new file mode 100644
index 0000000..506eebc
--- /dev/null
+++ b/snapshot.py
@@ -0,0 +1,58 @@
+#!/usr/bin/python3
+
+import time
+import argparse
+import libvirt
+from lxml import etree
+from lxml.builder import E
+
+
+def parse_args():
+    parser = argparse.ArgumentParser(description='Create snapshot for libvirt machine')
+    parser.add_argument('--name', required=True, help='Machine name')
+    parser.add_argument('--device', required=True, help='Device path (like vda)')
+    return parser.parse_args()
+
+def create_snapshot_node(disk_node):
+    device = disk_node.xpath('/domain/devices/disk/target/@dev')[0]
+    disk_path = disk_node.xpath('/domain/devices/disk/source/@file')[0]
+
+    now = int(time.time())
+    snap_path = disk_path.rsplit('.', 1)[0]
+    snap_path = f'{snap_path}-{now}.snap'
+    snap_name = snap_path.rsplit('/', 1)[1]
+
+    snap = E.domainsnapshot()
+    snap.append(E.name(snap_name))
+
+    disks = E.disks()
+    disk = E.disk(name=device)
+    disk.append(E.source(file=snap_path))
+    disks.append(disk)
+
+    snap.append(disks)
+    return snap
+
+def get_disk_by_name(vm, name):
+    raw_xml = vm.XMLDesc(0)
+    xml = etree.XML(raw_xml)
+    disks = xml.xpath(f'/domain/devices/disk/target[@dev="{name}"]/..')
+    return disks[0]
+
+def main():
+    args = parse_args()
+    session = libvirt.open('qemu:///system')
+    try:
+        vm = session.lookupByName(args.name)
+    except libvirt.libvirtError:
+        print(f'Machine with name {args.name} not found!')
+        session.close()
+        return
+    disk = get_disk_by_name(vm, args.device)
+    snapshot = create_snapshot_node(disk)
+    vm.snapshotCreateXML(etree.tostring(snapshot, encoding='unicode'),
+            libvirt.VIR_DOMAIN_SNAPSHOT_CREATE_DISK_ONLY | libvirt.VIR_DOMAIN_SNAPSHOT_CREATE_ATOMIC)
+    session.close()
+
+if __name__ == '__main__':
+    main()