def func_regexsub(self, vals, units, refs, flags):
if len(vals) == 4:
return OperationResult(re.subn(StringsFeature.string(self, vals[1]), StringsFeature.string(self, vals[2]), StringsFeature.string(self, vals[0]), int(vals[3]))[0])
elif len(vals) == 3:
return OperationResult(re.sub(StringsFeature.string(self, vals[1]), StringsFeature.string(self, vals[2]), StringsFeature.string(self, vals[0])))
else:
raise CalculatorException("regexsub requires 3 or 4 arguments, found {0}".format(len(vals)))
def func_regexget(self, vals, units, refs, flags):
found = re.findall(StringsFeature.string(self, vals[1]), StringsFeature.string(self, vals[0]))
if len(vals) == 3:
group = vals[2] - 1
return OperationResult(found[int(group)])
found = [OperandResult(f, None, None) for f in found]
return OperationResult(found)
def op_number_multiply(self, vals, units, refs, flags):
if isinstance(vals[0], UnitPowerList) and isinstance(
vals[1], UnitPowerList):
unit = UnitPowerList.new([vals[0], 1, vals[1], 1])
unit.no_simplify = True
return OperationResult(unit)
for i in range(0, len(vals)):
if isinstance(vals[i], UnitPowerList):
units[i] = vals[i]
vals[i] = Decimal('1')
num_type = self.number(vals[0])[1]
val0 = self.number(vals[0])[0]
val1 = self.number(vals[1])[0]
restored_number = self.restore_number_type(val0 * val1, num_type)
res = OperationResult(restored_number)
if units[0] is not None and units[1] is not None:
unit = UnitPowerList.new([units[0], 1, units[1], 1])
res.set_unit(unit)
elif units[0] is not None:
res.set_unit(units[0])
elif units[1] is not None:
res.set_unit(units[1])
return res
def op_boolean_conditional(self, vals, units, refs, flags):
if BooleansFeature.boolean(self, vals[0]):
res = OperationResult(vals[1])
res.set_unit(units[1])
res.set_ref(refs[1])
return res
res = OperationResult(vals[2])
res.set_unit(units[2])
res.set_ref(refs[2])
return res
def func_substr(self, vals, units, refs, flags):
start = int(vals[1])
if len(vals) == 2:
return OperationResult(
StringsFeature.string(self, vals[0])[start:])
elif len(vals) == 3:
end = int(vals[2]) + 1
if end <= 0:
end += len(vals[0])
return OperationResult(
StringsFeature.string(self, vals[0])[start:end])
else:
raise CalculatorException(
"substr needs either 2 or 3 arguments, found {0}".format(
len(vals)))
def op_number_power(self, vals, units, refs, flags):
power_numtype = self.number(vals[1])
power = power_numtype[0]
num_type = power_numtype[1]
if isinstance(vals[0], UnitPowerList):
unit = vals[0].power(power)
return OperationResult(unit)
num_res = vals[0]**power
restored_number = self.restore_number_type(num_res, num_type)
res = OperationResult(restored_number)
if units[1] is not None:
raise CalculatorException("Power operand must be simple number")
if units[0] is not None:
res.set_unit(units[0].power(power))
return res
def func_split(self, vals, units, refs, flags):
string = StringsFeature.string(self, vals[0])
splitter = StringsFeature.string(self, vals[1])
results = [
OperandResult(r, None, None) for r in string.split(splitter)
]
return OperationResult(results)
def func_delete(self, vals, units, refs, flags):
if isinstance(refs[0], VariableItem):
varname = refs[0].var
del self.vars[varname]
return OperationResult(None)
else:
raise ExecuteException("Expecting variable, received".format(refs[0].desc()), [], None)
def op_bitwise_not(self, vals, units, refs, flags):
dec_num, num_type = self.number(vals[0])
if dec_num < 0 or dec_num % 1 != 0:
raise CalculatorException('Operator requires positive integers')
int_val = int(dec_num)
if num_type and 'width' in num_type.opts:
mask_val = 2**num_type.opts['width'] - 1
else:
mask_val = 2**math.ceil(math.log(int_val, 2)) - 1
flipped_val = ~int_val
masked_flipped_val = flipped_val & mask_val
if num_type:
return OperationResult(num_type.restore(self, masked_flipped_val))
return OperationResult(Decimal(masked_flipped_val))
def op_number_integer_divide(self, vals, units, refs, flags):
op = self.ops_list['/']
division_result = op.call(self, [
OperandResult(vals[0], units[0], None),
OperandResult(vals[1], units[1], None)
], flags)
res = OperationResult(Decimal(math.floor(division_result.value)))
res.set_unit(division_result.unit)
return res
def func_dateformat(self, vals, units, refs, flags):
dt = DatesFeature.string_to_date(self, vals[0])
dt_format, dt_padding = DatesFeature.correct_format_for_date(self, dt)
return OperationResult(
format(
dt.strftime(self.feature_options['dates.datefunctions'][
'Human Readable ' + dt_format]),
self.feature_options['dates.datefunctions']['Human Readable ' +
dt_padding]))
def op_var_set(self, vals, units, refs, flags):
if isinstance(refs[0], VariableItem):
if 'stateless' not in flags.keys() or not flags['stateless']:
self.vars[refs[0].var] = (vals[1], units[1])
res = OperationResult(vals[1])
res.set_unit(units[1])
return res
else:
raise ExecuteException(
"Expecting variable, received".format(refs[0].desc()), [],
None)
def func_dateadd(self, vals, units, refs, flags):
dt = DatesFeature.string_to_date(self, vals[0])
num, fromunit = vals[1], units[1]
tounit = self.unit_normaliser.get_unit('seconds')
seconds, tounit = self.unit_normaliser.unit_conversion(
num, fromunit, tounit, False)
td = timedelta(seconds=float(seconds))
dt += td
res = OperationResult(DatesFeature.date_to_string(self, dt))
res.set_unit(None)
return res
def assignment_operator(self, sym, vals, units, refs, flags):
if isinstance(refs[0], VariableItem):
if 'stateless' not in flags.keys() or not flags['stateless']:
varname = refs[0].var
var = self.vars[varname]
op = self.ops_list[sym]
op_result = op.call(self, [
OperandResult(var[0], var[1], None),
OperandResult(vals[1], units[1], refs[1])
], flags)
self.vars[varname] = (op_result.value, op_result.unit)
res = OperationResult(vals[1])
res.set_unit(units[1])
return res
else:
raise ExecuteException(
"Expecting variable, received".format(refs[0].desc()), [],
None)
def op_unit_conversion(self, vals, units, refs, flags):
if isinstance(vals[0], UnitPowerList):
units[0] = vals[0]
vals[0] = Decimal('1')
num = vals[0]
if vals[1] is not None and not isinstance(vals[1], UnitPowerList) and vals[1] != Decimal('1'):
raise CalculatorException("Second operand must be just a unit")
fromunit = units[0]
tounit = vals[1]
if tounit is None:
raise CalculatorException("Second operand is not set")
if not isinstance(tounit, UnitPowerList):
raise CalculatorException("Second operand is not a unit")
num, tounit = self.unit_normaliser.unit_conversion(num, fromunit, tounit, False)
tounit.no_simplify = True
res = OperationResult(num)
res.set_unit(tounit)
return res
def func_datedifference(self, vals, units, refs, flags):
date1 = DatesFeature.string_to_date(self, vals[0])
date2 = DatesFeature.string_to_date(self, vals[1])
if ((date1.utcoffset() is None) != (date2.utcoffset() is None)):
raise CalculatorException(
"Cannot compare dates with timezones and dates without timezones"
)
td = abs(date1 - date2)
seconds = Decimal(td.total_seconds())
seconds = round(seconds, min(6, getcontext().prec))
fromunit = UnitPowerList.new(
[self.unit_normaliser.get_unit('seconds'), 1])
tounit = UnitPowerList.new(
[self.unit_normaliser.get_unit('seconds'), 1])
value = seconds
if len(vals) >= 3:
tounit = vals[2]
value, tounit = self.unit_normaliser.unit_conversion(
seconds, fromunit, tounit, False)
res = OperationResult(value)
res.set_unit(tounit)
return res
def op_number_morethanequal(self, vals, units, refs, flags):
return OperationResult(vals[0] >= vals[1])
def func_upper(self, vals, units, refs, flags):
return OperationResult(StringsFeature.string(self, vals[0]).upper())
def func_length(self, vals, units, refs, flags):
return OperationResult(
Decimal(len(StringsFeature.string(self, vals[0]))))
def func_strip(self, vals, units, refs, flags):
return OperationResult(
StringsFeature.string(self, vals[0]).strip(
StringsFeature.string(self, vals[1])))
def op_number_modulus(self, vals, units, refs, flags):
return OperationResult(vals[0] % vals[1])
def func_regexcount(self, vals, units, refs, flags):
return OperationResult(Decimal(len(re.findall(StringsFeature.string(self, vals[1]), StringsFeature.string(self, vals[0])))))
def op_regexnot(self, vals, units, refs, flags):
return OperationResult((re.search(StringsFeature.string(self, vals[1]), StringsFeature.string(self, vals[0])) is None))
def func_regexsplit(self, vals, units, refs, flags):
found = re.split(StringsFeature.string(self, vals[1]), StringsFeature.string(self, vals[0]))
found = [OperandResult(f, None, None) for f in found]
return OperationResult(found)
def op_bitwise_rshift(self, vals, units, refs, flags):
return OperationResult(Decimal(int(vals[0]) >> int(vals[1])))
def func_join(self, vals, units, refs, flags):
delimiter = ''
if len(vals) > 1:
delimiter = vals[1]
values = [StringsFeature.string(self, v.value) for v in vals[0]]
return OperationResult(delimiter.join(values))
def op_boolean_not(self, vals, units, refs, flags):
return OperationResult(not BooleansFeature.boolean(self, vals[0]))
def op_number_notequals(self, vals, units, refs, flags):
return OperationResult(vals[0] != vals[1])
def op_boolean_xor(self, vals, units, refs, flags):
return OperationResult(BooleansFeature.boolean(self, vals[0]) != BooleansFeature.boolean(self, vals[1]))
def op_number_lessthan(self, vals, units, refs, flags):
return OperationResult(vals[0] < vals[1])