__init__.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809
  1. #
  2. # The Qubes OS Project, https://www.qubes-os.org/
  3. #
  4. # Copyright (C) 2010-2015 Joanna Rutkowska <joanna@invisiblethingslab.com>
  5. # Copyright (C) 2011-2015 Marek Marczykowski-Górecki
  6. # <marmarek@invisiblethingslab.com>
  7. # Copyright (C) 2014-2015 Wojtek Porczyk <woju@invisiblethingslab.com>
  8. #
  9. # This library is free software; you can redistribute it and/or
  10. # modify it under the terms of the GNU Lesser General Public
  11. # License as published by the Free Software Foundation; either
  12. # version 2.1 of the License, or (at your option) any later version.
  13. #
  14. # This library is distributed in the hope that it will be useful,
  15. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  16. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17. # Lesser General Public License for more details.
  18. #
  19. # You should have received a copy of the GNU Lesser General Public
  20. # License along with this library; if not, see <https://www.gnu.org/licenses/>.
  21. #
  22. '''
  23. Qubes OS
  24. :copyright: © 2010-2015 Invisible Things Lab
  25. '''
  26. import builtins
  27. import collections.abc
  28. import os
  29. import os.path
  30. import string
  31. import lxml.etree
  32. import qubes.config
  33. import qubes.events
  34. import qubes.exc
  35. __author__ = 'Invisible Things Lab'
  36. __license__ = 'GPLv2 or later'
  37. __version__ = 'R3'
  38. class Label:
  39. '''Label definition for virtual machines
  40. Label specifies colour of the padlock displayed next to VM's name.
  41. When this is a :py:class:`qubes.vm.dispvm.DispVM`, padlock is overlayed
  42. with recycling pictogram.
  43. :param int index: numeric identificator of label
  44. :param str color: colour specification as in HTML (``#abcdef``)
  45. :param str name: label's name like "red" or "green"
  46. '''
  47. def __init__(self, index, color, name):
  48. #: numeric identificator of label
  49. self.index = index
  50. #: colour specification as in HTML (``#abcdef``)
  51. self.color = color
  52. #: label's name like "red" or "green"
  53. self.name = name
  54. #: freedesktop icon name, suitable for use in
  55. #: :py:meth:`PyQt4.QtGui.QIcon.fromTheme`
  56. self.icon = 'appvm-' + name
  57. #: freedesktop icon name, suitable for use in
  58. #: :py:meth:`PyQt4.QtGui.QIcon.fromTheme` on DispVMs
  59. self.icon_dispvm = 'dispvm-' + name
  60. @classmethod
  61. def fromxml(cls, xml):
  62. '''Create label definition from XML node
  63. :param lxml.etree._Element xml: XML node reference
  64. :rtype: :py:class:`qubes.Label`
  65. '''
  66. index = int(xml.get('id').split('-', 1)[1])
  67. color = xml.get('color')
  68. name = xml.text
  69. return cls(index, color, name)
  70. def __xml__(self):
  71. element = lxml.etree.Element(
  72. 'label', id='label-{}'.format(self.index), color=self.color)
  73. element.text = self.name
  74. return element
  75. def __str__(self):
  76. return self.name
  77. def __repr__(self):
  78. return '{}({!r}, {!r}, {!r})'.format(
  79. self.__class__.__name__,
  80. self.index,
  81. self.color,
  82. self.name)
  83. def __eq__(self, other):
  84. if isinstance(other, Label):
  85. return self.name == other.name
  86. return NotImplemented
  87. def __hash__(self):
  88. return hash(self.name)
  89. @builtins.property
  90. def icon_path(self):
  91. '''Icon path
  92. .. deprecated:: 2.0
  93. use :py:meth:`PyQt4.QtGui.QIcon.fromTheme` and :py:attr:`icon`
  94. '''
  95. return os.path.join(qubes.config.system_path['qubes_icon_dir'],
  96. self.icon) + ".png"
  97. @builtins.property
  98. def icon_path_dispvm(self):
  99. '''Icon path
  100. .. deprecated:: 2.0
  101. use :py:meth:`PyQt4.QtGui.QIcon.fromTheme` and :py:attr:`icon_dispvm`
  102. '''
  103. return os.path.join(qubes.config.system_path['qubes_icon_dir'],
  104. self.icon_dispvm) + ".png"
  105. class property: # pylint: disable=redefined-builtin,invalid-name
  106. '''Qubes property.
  107. This class holds one property that can be saved to and loaded from
  108. :file:`qubes.xml`. It is used for both global and per-VM properties.
  109. Property can be unset by ordinary ``del`` statement or assigning
  110. :py:attr:`DEFAULT` special value to it. After deletion (or before first
  111. assignment/load) attempting to read a property will get its default value
  112. or, when no default, py:class:`exceptions.AttributeError`.
  113. :param str name: name of the property
  114. :param collections.abc.Callable setter: if not :py:obj:`None`, this is \
  115. used to initialise value; first parameter to the function is holder \
  116. instance and the second is value; this is called before ``type``
  117. :param collections.abc.Callable saver: function to coerce value to \
  118. something readable by setter
  119. :param type type: if not :py:obj:`None`, value is coerced to this type
  120. :param object default: default value; if callable, will be called with \
  121. holder as first argument
  122. :param int load_stage: stage when property should be loaded (see \
  123. :py:class:`Qubes` for description of stages)
  124. :param int order: order of evaluation (bigger order values are later)
  125. :param bool clone: :py:meth:`PropertyHolder.clone_properties` will not \
  126. include this property by default if :py:obj:`False`
  127. :param str doc: docstring; this should be one paragraph of plain RST, no \
  128. sphinx-specific features
  129. Setters and savers have following signatures:
  130. .. :py:function:: setter(self, prop, value)
  131. :noindex:
  132. :param self: instance of object that is holding property
  133. :param prop: property object
  134. :param value: value being assigned
  135. .. :py:function:: saver(self, prop, value)
  136. :noindex:
  137. :param self: instance of object that is holding property
  138. :param prop: property object
  139. :param value: value being saved
  140. :rtype: str
  141. :raises property.DontSave: when property should not be saved at all
  142. '''
  143. #: Assigning this value to property means setting it to its default value.
  144. #: If property has no default value, this will unset it.
  145. DEFAULT = object()
  146. # internal use only
  147. _NO_DEFAULT = object()
  148. def __init__(self, name, setter=None, saver=None, type=None,
  149. default=_NO_DEFAULT, write_once=False, load_stage=2, order=0,
  150. save_via_ref=False, clone=True,
  151. doc=None):
  152. # pylint: disable=redefined-builtin
  153. self.__name__ = name
  154. if setter is None and type is bool:
  155. setter = qubes.property.bool
  156. self._setter = setter
  157. self._saver = saver if saver is not None else (
  158. lambda self, prop, value: str(value))
  159. self.type = type
  160. self._default = default
  161. self._default_function = None
  162. if isinstance(default, collections.abc.Callable):
  163. self._default_function = default
  164. self._write_once = write_once
  165. self.order = order
  166. self.load_stage = load_stage
  167. self.save_via_ref = save_via_ref
  168. self.clone = clone
  169. self.__doc__ = doc
  170. self._attr_name = '_qubesprop_' + name
  171. def __get__(self, instance, owner):
  172. if instance is None:
  173. return self
  174. # XXX this violates duck typing, shall we keep it?
  175. if not isinstance(instance, PropertyHolder):
  176. raise AttributeError('qubes.property should be used on '
  177. 'qubes.PropertyHolder instances only')
  178. try:
  179. return getattr(instance, self._attr_name)
  180. except AttributeError:
  181. return self.get_default(instance)
  182. def get_default(self, instance):
  183. if self._default is self._NO_DEFAULT:
  184. raise AttributeError(
  185. 'property {!r} have no default'.format(self.__name__))
  186. if self._default_function:
  187. return self._default_function(instance)
  188. return self._default
  189. def __set__(self, instance, value):
  190. self._enforce_write_once(instance)
  191. if value is self.__class__.DEFAULT:
  192. self.__delete__(instance)
  193. return
  194. try:
  195. oldvalue = getattr(instance, self.__name__)
  196. has_oldvalue = True
  197. except AttributeError:
  198. has_oldvalue = False
  199. if self._setter is not None:
  200. value = self._setter(instance, self, value)
  201. if self.type not in (None, type(value)):
  202. value = self.type(value)
  203. if has_oldvalue:
  204. instance.fire_event('property-pre-set:' + self.__name__,
  205. pre_event=True,
  206. name=self.__name__, newvalue=value, oldvalue=oldvalue)
  207. else:
  208. instance.fire_event('property-pre-set:' + self.__name__,
  209. pre_event=True,
  210. name=self.__name__, newvalue=value)
  211. instance._property_init(self, value) # pylint: disable=protected-access
  212. if has_oldvalue:
  213. instance.fire_event('property-set:' + self.__name__,
  214. name=self.__name__, newvalue=value, oldvalue=oldvalue)
  215. else:
  216. instance.fire_event('property-set:' + self.__name__,
  217. name=self.__name__, newvalue=value)
  218. def __delete__(self, instance):
  219. self._enforce_write_once(instance)
  220. try:
  221. oldvalue = getattr(instance, self.__name__)
  222. has_oldvalue = True
  223. except AttributeError:
  224. has_oldvalue = False
  225. if has_oldvalue:
  226. instance.fire_event('property-pre-reset:' + self.__name__,
  227. pre_event=True,
  228. name=self.__name__, oldvalue=oldvalue)
  229. # deprecated, to be removed in Qubes 5.0
  230. instance.fire_event('property-pre-del:' + self.__name__,
  231. pre_event=True,
  232. name=self.__name__, oldvalue=oldvalue)
  233. try:
  234. delattr(instance, self._attr_name)
  235. except AttributeError:
  236. pass
  237. instance.fire_event('property-reset:' + self.__name__,
  238. name=self.__name__, oldvalue=oldvalue)
  239. # deprecated, to be removed in Qubes 5.0
  240. instance.fire_event('property-del:' + self.__name__,
  241. name=self.__name__, oldvalue=oldvalue)
  242. else:
  243. instance.fire_event('property-pre-reset:' + self.__name__,
  244. pre_event=True,
  245. name=self.__name__)
  246. # deprecated, to be removed in Qubes 5.0
  247. instance.fire_event('property-pre-del:' + self.__name__,
  248. pre_event=True,
  249. name=self.__name__)
  250. instance.fire_event('property-reset:' + self.__name__,
  251. name=self.__name__)
  252. # deprecated, to be removed in Qubes 5.0
  253. instance.fire_event('property-del:' + self.__name__,
  254. name=self.__name__)
  255. def __repr__(self):
  256. default = ' default={!r}'.format(self._default) \
  257. if self._default is not self._NO_DEFAULT \
  258. else ''
  259. return '<{} object at {:#x} name={!r}{}>'.format(
  260. self.__class__.__name__, id(self), self.__name__, default)
  261. def __str__(self):
  262. return self.__name__
  263. def __hash__(self):
  264. return hash(self.__name__)
  265. def __lt__(self, other):
  266. if isinstance(other, property):
  267. return (self.load_stage, self.order, self.__name__) <\
  268. (other.load_stage, other.order, other.__name__)
  269. return NotImplemented
  270. def __eq__(self, other):
  271. if isinstance(other, str):
  272. return self.__name__ == other
  273. return isinstance(other, property) and self.__name__ == other.__name__
  274. def _enforce_write_once(self, instance):
  275. if self._write_once and not instance.property_is_default(self):
  276. raise AttributeError(
  277. 'property {!r} is write-once and already set'.format(
  278. self.__name__))
  279. def sanitize(self, *, untrusted_newvalue):
  280. '''Coarse sanitization of value to be set, before sending it to a
  281. setter. Can raise QubesValueError if the value is invalid.
  282. :param untrusted_newvalue: value to be validated
  283. :return: sanitized value
  284. :raises: qubes.exc.QubesValueError
  285. '''
  286. # do not treat type='str' as sufficient validation
  287. if self.type is not None and self.type is not str:
  288. # assume specific type will preform enough validation
  289. try:
  290. untrusted_newvalue = untrusted_newvalue.decode('ascii',
  291. errors='strict')
  292. except UnicodeDecodeError:
  293. raise qubes.exc.QubesValueError
  294. if self.type is bool:
  295. return self.bool(None, None, untrusted_newvalue)
  296. try:
  297. return self.type(untrusted_newvalue)
  298. except ValueError:
  299. raise qubes.exc.QubesValueError
  300. else:
  301. # 'str' or not specified type
  302. try:
  303. untrusted_newvalue = untrusted_newvalue.decode('ascii',
  304. errors='strict')
  305. except UnicodeDecodeError:
  306. raise qubes.exc.QubesValueError
  307. allowed_set = string.printable
  308. if not all(x in allowed_set for x in untrusted_newvalue):
  309. raise qubes.exc.QubesValueError(
  310. 'Invalid characters in property value')
  311. return untrusted_newvalue
  312. #
  313. # exceptions
  314. #
  315. class DontSave(Exception):
  316. '''This exception may be raised from saver to sign that property should
  317. not be saved.
  318. '''
  319. @staticmethod
  320. def dontsave(self, prop, value):
  321. '''Dummy saver that never saves anything.'''
  322. # pylint: disable=bad-staticmethod-argument,unused-argument
  323. raise property.DontSave()
  324. #
  325. # some setters provided
  326. #
  327. @staticmethod
  328. def forbidden(self, prop, value):
  329. '''Property setter that forbids loading a property.
  330. This is used to effectively disable property in classes which inherit
  331. unwanted property. When someone attempts to load such a property, it
  332. :throws AttributeError: always
  333. ''' # pylint: disable=bad-staticmethod-argument,unused-argument
  334. raise AttributeError(
  335. 'setting {} property on {} instance is forbidden'.format(
  336. prop.__name__, self.__class__.__name__))
  337. @staticmethod
  338. def bool(self, prop, value):
  339. '''Property setter for boolean properties.
  340. It accepts (case-insensitive) ``'0'``, ``'no'`` and ``false`` as
  341. :py:obj:`False` and ``'1'``, ``'yes'`` and ``'true'`` as
  342. :py:obj:`True`.
  343. ''' # pylint: disable=bad-staticmethod-argument,unused-argument
  344. if isinstance(value, str):
  345. lcvalue = value.lower()
  346. if lcvalue in ('0', 'no', 'false', 'off'):
  347. return False
  348. if lcvalue in ('1', 'yes', 'true', 'on'):
  349. return True
  350. raise qubes.exc.QubesValueError(
  351. 'Invalid literal for boolean property: {!r}'.format(value))
  352. return bool(value)
  353. def stateless_property(func):
  354. '''Decorator similar to :py:class:`builtins.property`, but for properties
  355. exposed through management API (including qvm-prefs etc)'''
  356. return property(func.__name__,
  357. setter=property.forbidden,
  358. saver=property.dontsave,
  359. default=func,
  360. doc=func.__doc__)
  361. class PropertyHolder(qubes.events.Emitter):
  362. '''Abstract class for holding :py:class:`qubes.property`
  363. Events fired by instances of this class:
  364. .. event:: property-load (subject, event)
  365. Fired once after all properties are loaded from XML. Individual
  366. ``property-set`` events are not fired.
  367. .. event:: property-set:<propname> \
  368. (subject, event, name, newvalue[, oldvalue])
  369. Fired when property changes state. Signature is variable,
  370. *oldvalue* is present only if there was an old value.
  371. :param name: Property name
  372. :param newvalue: New value of the property
  373. :param oldvalue: Old value of the property
  374. .. event:: property-pre-set:<propname> \
  375. (subject, event, name, newvalue[, oldvalue])
  376. Fired before property changes state. Signature is variable,
  377. *oldvalue* is present only if there was an old value.
  378. :param name: Property name
  379. :param newvalue: New value of the property
  380. :param oldvalue: Old value of the property
  381. .. event:: property-del:<propname> \
  382. (subject, event, name[, oldvalue])
  383. Fired when property gets deleted (is set to default). Signature is
  384. variable, *oldvalue* is present only if there was an old value.
  385. This event is deprecated and will be removed in Qubes 5.0.
  386. Use property-reset instead.
  387. :param name: Property name
  388. :param oldvalue: Old value of the property
  389. .. event:: property-pre-del:<propname> \
  390. (subject, event, name[, oldvalue])
  391. Fired before property gets deleted (is set to default). Signature
  392. is variable, *oldvalue* is present only if there was an old value.
  393. This event is deprecated and will be removed in Qubes 5.0.
  394. Use property-pre-reset instead.
  395. :param name: Property name
  396. :param oldvalue: Old value of the property
  397. .. event:: property-reset:<propname> \
  398. (subject, event, name[, oldvalue])
  399. Fired when property gets reset to the (possibly dynamic) default.
  400. This even may be also fired when the property is already in
  401. "default" state, but the calculated default value changes.
  402. Signature is variable, *oldvalue* is present only if there was an
  403. old value.
  404. :param name: Property name
  405. :param oldvalue: Old value of the property
  406. .. event:: property-pre-reset:<propname> \
  407. (subject, event, name[, oldvalue])
  408. Fired before property gets reset to the (possibly dynamic) default.
  409. Signature is variable, *oldvalue* is present only if there was an
  410. old value.
  411. :param name: Property name
  412. :param oldvalue: Old value of the property
  413. .. event:: clone-properties (subject, event, src, proplist)
  414. :param src: object, from which we are cloning
  415. :param proplist: list of properties
  416. Members:
  417. '''
  418. def __init__(self, xml, **kwargs):
  419. self.xml = xml
  420. propvalues = {}
  421. all_names = self.property_dict()
  422. for key in list(kwargs):
  423. if not key in all_names:
  424. continue
  425. propvalues[key] = kwargs.pop(key)
  426. super(PropertyHolder, self).__init__(**kwargs)
  427. for key, value in propvalues.items():
  428. setattr(self, key, value)
  429. if self.xml is not None:
  430. # check if properties are appropriate
  431. for node in self.xml.xpath('./properties/property'):
  432. name = node.get('name')
  433. if name not in all_names:
  434. raise TypeError(
  435. 'property {!r} not applicable to {!r}'.format(
  436. name, self.__class__.__name__))
  437. # pylint: disable=too-many-nested-blocks
  438. @classmethod
  439. def property_dict(cls, load_stage=None):
  440. '''List all properties attached to this VM's class
  441. :param load_stage: Filter by load stage
  442. :type load_stage: :py:func:`int` or :py:obj:`None`
  443. '''
  444. # use cls.__dict__ since we must not look at parent classes
  445. if "_property_dict" not in cls.__dict__:
  446. cls._property_dict = {}
  447. memo = cls._property_dict
  448. if load_stage not in memo:
  449. props = dict()
  450. if load_stage is None:
  451. for class_ in cls.__mro__:
  452. for name in class_.__dict__:
  453. # don't overwrite props with those from base classes
  454. if name not in props:
  455. prop = class_.__dict__[name]
  456. if isinstance(prop, property):
  457. assert name == prop.__name__
  458. props[name] = prop
  459. else:
  460. for prop in cls.property_dict().values():
  461. if prop.load_stage == load_stage:
  462. props[prop.__name__] = prop
  463. memo[load_stage] = props
  464. return memo[load_stage]
  465. @classmethod
  466. def property_list(cls, load_stage=None):
  467. '''List all properties attached to this VM's class
  468. :param load_stage: Filter by load stage
  469. :type load_stage: :py:func:`int` or :py:obj:`None`
  470. '''
  471. # use cls.__dict__ since we must not look at parent classes
  472. if "_property_list" not in cls.__dict__:
  473. cls._property_list = {}
  474. memo = cls._property_list
  475. if load_stage not in memo:
  476. memo[load_stage] = sorted(cls.property_dict(load_stage).values())
  477. return memo[load_stage]
  478. def _property_init(self, prop, value):
  479. '''Initialise property to a given value, without side effects.
  480. :param qubes.property prop: property object of particular interest
  481. :param value: value
  482. '''
  483. # pylint: disable=protected-access
  484. setattr(self, self.property_get_def(prop)._attr_name, value)
  485. def property_is_default(self, prop):
  486. '''Check whether property is in it's default value.
  487. Properties when unset may return some default value, so
  488. ``hasattr(vm, prop.__name__)`` is wrong in some circumstances. This
  489. method allows for checking if the value returned is in fact it's
  490. default value.
  491. :param qubes.property prop: property object of particular interest
  492. :rtype: bool
  493. ''' # pylint: disable=protected-access
  494. # both property_get_def() and ._attr_name may throw AttributeError,
  495. # which we don't want to catch
  496. attrname = self.property_get_def(prop)._attr_name
  497. return not hasattr(self, attrname)
  498. def property_get_default(self, prop):
  499. '''Get property default value.
  500. :param qubes.property or str prop: property object of particular
  501. interest
  502. '''
  503. return self.property_get_def(prop).get_default(self)
  504. @classmethod
  505. def property_get_def(cls, prop):
  506. '''Return property definition object.
  507. If prop is already :py:class:`qubes.property` instance, return the same
  508. object.
  509. :param prop: property object or name
  510. :type prop: qubes.property or str
  511. :rtype: qubes.property
  512. '''
  513. if isinstance(prop, qubes.property):
  514. return prop
  515. props = cls.property_dict()
  516. if prop in props:
  517. return props[prop]
  518. raise AttributeError('No property {!r} found in {!r}'.format(
  519. prop, cls))
  520. def load_properties(self, load_stage=None):
  521. '''Load properties from immediate children of XML node.
  522. ``property-set`` events are not fired for each individual property.
  523. :param int load_stage: Stage of loading.
  524. '''
  525. if self.xml is None:
  526. return
  527. all_names = set(
  528. prop.__name__ for prop in self.property_list(load_stage))
  529. for node in self.xml.xpath('./properties/property'):
  530. name = node.get('name')
  531. value = node.get('ref') or node.text
  532. if not name in all_names:
  533. continue
  534. setattr(self, name, value)
  535. def xml_properties(self, with_defaults=False):
  536. '''Iterator that yields XML nodes representing set properties.
  537. :param bool with_defaults: If :py:obj:`True`, then it also includes \
  538. properties which were not set explicite, but have default values \
  539. filled.
  540. '''
  541. properties = lxml.etree.Element('properties')
  542. for prop in self.property_list():
  543. # pylint: disable=protected-access
  544. try:
  545. value = getattr(
  546. self, (prop.__name__ if with_defaults else prop._attr_name))
  547. except AttributeError:
  548. continue
  549. try:
  550. value = prop._saver(self, prop, value)
  551. except property.DontSave:
  552. continue
  553. element = lxml.etree.Element('property', name=prop.__name__)
  554. if prop.save_via_ref:
  555. element.set('ref', value)
  556. else:
  557. element.text = value
  558. properties.append(element)
  559. return properties
  560. # this was clone_attrs
  561. def clone_properties(self, src, proplist=None):
  562. '''Clone properties from other object.
  563. :param PropertyHolder src: source object
  564. :param iterable proplist: list of properties \
  565. (:py:obj:`None` or omit for all properties except those with \
  566. :py:attr:`property.clone` set to :py:obj:`False`)
  567. '''
  568. if proplist is None:
  569. proplist = [prop for prop in self.property_list()
  570. if prop.clone]
  571. else:
  572. proplist = [prop for prop in self.property_list()
  573. if prop.__name__ in proplist or prop in proplist]
  574. for prop in proplist:
  575. try:
  576. # pylint: disable=protected-access
  577. self._property_init(prop, getattr(src, prop._attr_name))
  578. except AttributeError:
  579. continue
  580. self.fire_event('clone-properties', src=src, proplist=proplist)
  581. def property_require(self, prop, allow_none=False, hard=False):
  582. '''Complain badly when property is not set.
  583. :param prop: property name or object
  584. :type prop: qubes.property or str
  585. :param bool allow_none: if :py:obj:`True`, don't complain if \
  586. :py:obj:`None` is found
  587. :param bool hard: if :py:obj:`True`, raise :py:class:`AssertionError`; \
  588. if :py:obj:`False`, log warning instead
  589. '''
  590. if isinstance(prop, qubes.property):
  591. prop = prop.__name__
  592. try:
  593. value = getattr(self, prop)
  594. if value is None and not allow_none:
  595. msg = 'Property {!r} cannot be None'.format(prop)
  596. if hard:
  597. raise ValueError(msg)
  598. self.log.fatal(msg)
  599. except AttributeError:
  600. # pylint: disable=no-member
  601. msg = 'Required property {!r} not set on {!r}'.format(prop, self)
  602. if hard:
  603. raise ValueError(msg)
  604. # pylint: disable=no-member
  605. self.log.fatal(msg)
  606. def close(self):
  607. super().close()
  608. # Remove all properties -- somewhere in them there are cyclic
  609. # references. This just removes all the properties, just in case.
  610. # They are removed directly, bypassing write_once.
  611. for prop in self.property_list():
  612. # pylint: disable=protected-access
  613. try:
  614. delattr(self, prop._attr_name)
  615. except AttributeError:
  616. pass
  617. # pylint: disable=wrong-import-position
  618. from qubes.vm import VMProperty
  619. from qubes.app import Qubes
  620. __all__ = [
  621. 'Label',
  622. 'PropertyHolder',
  623. 'Qubes',
  624. 'VMProperty',
  625. 'property',
  626. ]