600955edd4
* Vendor ruamel.yaml v0.17.21 * Add unit test for whitespace regression * Add an abstraction layer in Spack to wrap ruamel.yaml All YAML operations are routed through spack.util.spack_yaml The custom classes have been adapted to the new ruamel.yaml class hierarchy. Fixed line annotation issue in "spack config blame"
128 lines
4.2 KiB
Python
128 lines
4.2 KiB
Python
# coding: utf-8
|
|
|
|
from ruamel.yaml.anchor import Anchor
|
|
|
|
if False: # MYPY
|
|
from typing import Text, Any, Dict, List # NOQA
|
|
|
|
__all__ = ['ScalarInt', 'BinaryInt', 'OctalInt', 'HexInt', 'HexCapsInt', 'DecimalInt']
|
|
|
|
|
|
class ScalarInt(int):
|
|
def __new__(cls, *args, **kw):
|
|
# type: (Any, Any, Any) -> Any
|
|
width = kw.pop('width', None)
|
|
underscore = kw.pop('underscore', None)
|
|
anchor = kw.pop('anchor', None)
|
|
v = int.__new__(cls, *args, **kw)
|
|
v._width = width
|
|
v._underscore = underscore
|
|
if anchor is not None:
|
|
v.yaml_set_anchor(anchor, always_dump=True)
|
|
return v
|
|
|
|
def __iadd__(self, a): # type: ignore
|
|
# type: (Any) -> Any
|
|
x = type(self)(self + a)
|
|
x._width = self._width # type: ignore
|
|
x._underscore = ( # type: ignore
|
|
self._underscore[:] if self._underscore is not None else None # type: ignore
|
|
) # NOQA
|
|
return x
|
|
|
|
def __ifloordiv__(self, a): # type: ignore
|
|
# type: (Any) -> Any
|
|
x = type(self)(self // a)
|
|
x._width = self._width # type: ignore
|
|
x._underscore = ( # type: ignore
|
|
self._underscore[:] if self._underscore is not None else None # type: ignore
|
|
) # NOQA
|
|
return x
|
|
|
|
def __imul__(self, a): # type: ignore
|
|
# type: (Any) -> Any
|
|
x = type(self)(self * a)
|
|
x._width = self._width # type: ignore
|
|
x._underscore = ( # type: ignore
|
|
self._underscore[:] if self._underscore is not None else None # type: ignore
|
|
) # NOQA
|
|
return x
|
|
|
|
def __ipow__(self, a): # type: ignore
|
|
# type: (Any) -> Any
|
|
x = type(self)(self ** a)
|
|
x._width = self._width # type: ignore
|
|
x._underscore = ( # type: ignore
|
|
self._underscore[:] if self._underscore is not None else None # type: ignore
|
|
) # NOQA
|
|
return x
|
|
|
|
def __isub__(self, a): # type: ignore
|
|
# type: (Any) -> Any
|
|
x = type(self)(self - a)
|
|
x._width = self._width # type: ignore
|
|
x._underscore = ( # type: ignore
|
|
self._underscore[:] if self._underscore is not None else None # type: ignore
|
|
) # NOQA
|
|
return x
|
|
|
|
@property
|
|
def anchor(self):
|
|
# type: () -> Any
|
|
if not hasattr(self, Anchor.attrib):
|
|
setattr(self, Anchor.attrib, Anchor())
|
|
return getattr(self, Anchor.attrib)
|
|
|
|
def yaml_anchor(self, any=False):
|
|
# type: (bool) -> Any
|
|
if not hasattr(self, Anchor.attrib):
|
|
return None
|
|
if any or self.anchor.always_dump:
|
|
return self.anchor
|
|
return None
|
|
|
|
def yaml_set_anchor(self, value, always_dump=False):
|
|
# type: (Any, bool) -> None
|
|
self.anchor.value = value
|
|
self.anchor.always_dump = always_dump
|
|
|
|
|
|
class BinaryInt(ScalarInt):
|
|
def __new__(cls, value, width=None, underscore=None, anchor=None):
|
|
# type: (Any, Any, Any, Any) -> Any
|
|
return ScalarInt.__new__(cls, value, width=width, underscore=underscore, anchor=anchor)
|
|
|
|
|
|
class OctalInt(ScalarInt):
|
|
def __new__(cls, value, width=None, underscore=None, anchor=None):
|
|
# type: (Any, Any, Any, Any) -> Any
|
|
return ScalarInt.__new__(cls, value, width=width, underscore=underscore, anchor=anchor)
|
|
|
|
|
|
# mixed casing of A-F is not supported, when loading the first non digit
|
|
# determines the case
|
|
|
|
|
|
class HexInt(ScalarInt):
|
|
"""uses lower case (a-f)"""
|
|
|
|
def __new__(cls, value, width=None, underscore=None, anchor=None):
|
|
# type: (Any, Any, Any, Any) -> Any
|
|
return ScalarInt.__new__(cls, value, width=width, underscore=underscore, anchor=anchor)
|
|
|
|
|
|
class HexCapsInt(ScalarInt):
|
|
"""uses upper case (A-F)"""
|
|
|
|
def __new__(cls, value, width=None, underscore=None, anchor=None):
|
|
# type: (Any, Any, Any, Any) -> Any
|
|
return ScalarInt.__new__(cls, value, width=width, underscore=underscore, anchor=anchor)
|
|
|
|
|
|
class DecimalInt(ScalarInt):
|
|
"""needed if anchor"""
|
|
|
|
def __new__(cls, value, width=None, underscore=None, anchor=None):
|
|
# type: (Any, Any, Any, Any) -> Any
|
|
return ScalarInt.__new__(cls, value, width=width, underscore=underscore, anchor=anchor)
|