Source code for synapse.lib.stormlib.math

import synapse.lib.stormtypes as s_stormtypes

@s_stormtypes.registry.registerLib
class MathLib(s_stormtypes.Lib):
    '''
    A Storm library for performing math operations.
    '''
    _storm_locals = (
        {'name': 'number',
         'desc': '''
            Convert a value to a Storm Number object.

            Storm Numbers are high precision fixed point decimals corresponding to
            the hugenum storage type.

            This is not to be used for converting a string to an integer.
         ''',
         'type': {'type': 'function', '_funcname': '_number',
                  'args': (
                      {'name': 'value', 'type': 'any',
                       'desc': 'Value to convert.', },
                  ),
                  'returns': {'type': 'number', 'desc': 'A Number object.', }}},
    )

    _storm_lib_path = ('math',)

    def getObjLocals(self):
        return {
            'number': self._number,
        }

    @s_stormtypes.stormfunc(readonly=True)
    async def _number(self, value):
        if isinstance(value, s_stormtypes.Number):
            return value
        return s_stormtypes.Number(value)