shaba/libvirt-python
1
0
Fork 0
mirror of https://gitlab.com/libvirt/libvirt-python.git synced 2025-07-14 16:59:34 +03:00
Code Issues Projects Releases Activity

examples: Convert to ArgumentParser

Browse Source 
Replace getopt() and hand-rolled-parser with argparse.ArgumentParser.

Fix wrong header comments copy-pasted from domstart.py

Signed-off-by: Philipp Hahn <hahn@univention.de>
This commit is contained in:
Philipp Hahn
2020-07-28 07:05:19 +02:00
committed by Philipp Hahn
parent 9cf539a2a8
commit 06aba185a8
12 changed files with 142 additions and 235 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
examples/sparsestream.py
View File

@ -1,10 +1,15 @@
#!/usr/bin/env python3
"""
Either uploads local FILE to libvirt VOLUME, or downloads libvirt
VOLUME into local FILE while preserving FILE/VOLUME sparseness
"""
# Example of sparse streams usage
#
# Authors:
# Michal Privoznik <mprivozn@redhat.com>
import libvirt, sys, os
from argparse import ArgumentParser
def bytesWriteHandler(stream: libvirt.virStream, buf: bytes, opaque: int) -> int:
fd = opaque
@ -94,24 +99,22 @@ def upload(vol: libvirt.virStorageVol, st: libvirt.virStream, filename: str) ->
os.close(fd)
# main
if len(sys.argv) != 5:
print("Usage: ", sys.argv[0], " URI --upload/--download VOLUME FILE")
print("Either uploads local FILE to libvirt VOLUME, or downloads libvirt ")
print("VOLUME into local FILE while preserving FILE/VOLUME sparseness")
sys.exit(1)
parser = ArgumentParser(description=__doc__)
parser.add_argument("uri")
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument("--upload", action="store_const", const=upload, dest="operation")
group.add_argument("--download", action="store_const", const=download, dest="operation")
parser.add_argument("volume")
parser.add_argument("file")
args = parser.parse_args()
conn = libvirt.open(sys.argv[1])
vol = conn.storageVolLookupByKey(sys.argv[3])
conn = libvirt.open(args.uri)
vol = conn.storageVolLookupByKey(args.volume)
st = conn.newStream()
if sys.argv[2] == "--download":
download(vol, st, sys.argv[4])
elif sys.argv[2] == "--upload":
upload(vol, st, sys.argv[4])
else:
print("Unknown operation: %s " % sys.argv[1])
sys.exit(1)
args.operation(vol, st, args.file)
st.finish()
conn.close()