diff options
author | Philipp Hahn <hahn@univention.de> | 2020-04-20 18:39:54 +0200 |
---|---|---|
committer | Jano Tomko <jtomko@redhat.com> | 2020-09-01 13:26:01 +0000 |
commit | 7f021c21d6a091ca33615852d6061e75b2500f3f (patch) | |
tree | ba80cbb393ff61d9b1816bdf000790e7a81e4063 /generator.py | |
parent | 5a2c12278a1984632e344fd5b4578b274ac0016e (diff) | |
download | libvirt-python-7f021c21d6a091ca33615852d6061e75b2500f3f.tar.gz |
generator: Merge code for __init__ genration
Signed-off-by: Philipp Hahn <hahn@univention.de>
Diffstat (limited to 'generator.py')
-rwxr-xr-x | generator.py | 16 |
1 files changed, 5 insertions, 11 deletions
diff --git a/generator.py b/generator.py index f3ed905..a138dd5 100755 --- a/generator.py +++ b/generator.py @@ -1479,24 +1479,18 @@ def buildWrappers(module): "virStorageVol", "virNodeDevice", "virSecret", "virStream", "virNWFilter", "virNWFilterBinding"]: classes.write(" def __init__(self, conn, _obj=None):\n") - elif classname in ["virDomainCheckpoint", "virDomainSnapshot"]: - classes.write(" def __init__(self, dom, _obj=None):\n") - elif classname in ["virNetworkPort"]: - classes.write(" def __init__(self, net, _obj=None):\n") - else: - classes.write(" def __init__(self, _obj=None):\n") - if classname in ["virDomain", "virNetwork", "virInterface", - "virNodeDevice", "virSecret", "virStream", - "virNWFilter", "virNWFilterBinding"]: classes.write(" self._conn = conn\n") - elif classname in ["virStorageVol", "virStoragePool"]: - classes.write(" self._conn = conn if isinstance(conn, virConnect) else conn._conn # type: virConnect\n") elif classname in ["virDomainCheckpoint", "virDomainSnapshot"]: + classes.write(" def __init__(self, net, _obj=None):\n") classes.write(" self._dom = dom\n") classes.write(" self._conn = dom.connect()\n") elif classname in ["virNetworkPort"]: + classes.write(" def __init__(self, net, _obj=None) -> None:\n") classes.write(" self._net = net\n") classes.write(" self._conn = net.connect()\n") + else: + classes.write(" def __init__(self, _obj=None):\n") + classes.write(" if type(_obj).__name__ not in [\"PyCapsule\", \"PyCObject\"]:\n") classes.write(" raise Exception(\"Expected a wrapped C Object but got %s\" % type(_obj))\n") classes.write(" self._o = _obj\n\n") |