def change_account(request):
split_guid = request.POST.get('split_guid', '')
account_guid = request.POST.get('account_guid', '')
split = Split.objects.get(guid=split_guid)
Lock.obtain()
try:
split.account = Account.objects.get(guid=account_guid)
split.save()
finally:
Lock.release()
return {
'split_guid': split_guid,
'account_guid': account_guid,
}
def change_memo(request):
split_guid = request.POST.get('split_guid', '')
memo = request.POST.get('memo', '')
split = Split.objects.get(guid=split_guid)
Lock.obtain()
try:
split.memo = request.POST.get('memo', '')
split.save()
finally:
Lock.release()
return {
'split_guid': split_guid,
'memo': memo,
}
def change_account(request):
split_guid = request.POST.get('split_guid', '')
account_guid = request.POST.get('account_guid', '')
split = Split.objects.get(guid=split_guid)
Lock.obtain()
try:
split.account = Account.objects.get(guid=account_guid)
split.save()
finally:
Lock.release()
return {
'split_guid': split_guid,
'account_guid': account_guid,
}
def change_memo(request):
split_guid = request.POST.get('split_guid', '')
memo = request.POST.get('memo', '')
split = Split.objects.get(guid=split_guid)
Lock.obtain()
try:
split.memo = request.POST.get('memo', '')
split.save()
finally:
Lock.release()
return {
'split_guid': split_guid,
'memo': memo,
}
def apply(**kwargs):
splits = kwargs['splits']
accounts = splits.accounts
tx_desc = kwargs.get('tx_desc', None)
is_regex = kwargs.get('is_regex', False)
opposing_account = kwargs['opposing_account']
min_amount = kwargs.get('min_amount', None)
max_amount = kwargs.get('max_amount', None)
save_rule = kwargs['save_rule']
if min_amount is None:
min_amount = 0
elif min_amount < 0:
raise ValueError('min_amount (%s) < 0' % min_amount)
if max_amount is None:
max_amount = 0
elif max_amount < 0:
raise ValueError('max_amount (%s) < 0' % max_amount)
if min_amount and max_amount and min_amount > max_amount:
raise ValueError('min_amount (%s) > max_amount (%s)' %
(min_amount, max_amount))
filtered_splits = splits.filtered_splits
if tx_desc:
# Need to do tx_desc filter ourselves
if is_regex:
filtered_splits = \
filtered_splits.filter(transaction__description__iregex=tx_desc)
else:
filtered_splits = \
filtered_splits.filter(transaction__description__icontains=tx_desc)
else:
# Any tx_desc filtering has already been done for us
tx_desc = splits.tx_desc
is_regex = splits.tx_desc_is_regex
tx_guids = filtered_splits.distinct().values('transaction__guid')
splits_real = Split.objects \
.filter(transaction__guid__in=tx_guids) \
.exclude(account__guid__in=[a.guid for a in accounts])
split_guids = list(splits_real.distinct().values_list('guid',
flat=True))
tx_count = len(split_guids)
modified_tx_count = 0
if tx_count > 0:
Lock.obtain()
try:
splits = Split.objects.filter(guid__in=split_guids)
if opposing_account is None:
tx_guids = list(
splits.values_list('transaction__guid', flat=True))
Transaction.objects.filter(guid__in=tx_guids).delete()
Split.objects.filter(guid__in=split_guids).delete()
else:
splits.update(account=opposing_account)
modified_tx_count = tx_count
finally:
Lock.release()
if save_rule and tx_desc:
rule = Rule()
if opposing_account:
rule.opposing_account_guid = opposing_account.guid
else:
rule.opposing_account_guid = None
rule.match_tx_desc = tx_desc
rule.is_regex = is_regex
if min_amount: rule.min_amount = min_amount
if max_amount: rule.max_amount = max_amount
rule.save()
for a in accounts:
rule_account = RuleAccount()
rule_account.rule = rule
rule_account.account_guid = a.guid
rule_account.save()
return modified_tx_count
def apply(**kwargs):
splits = kwargs['splits']
account = splits.account
tx_desc = kwargs.get('tx_desc', None)
is_regex = kwargs.get('is_regex', False)
opposing_account = kwargs['opposing_account']
min_amount = kwargs.get('min_amount', None)
max_amount = kwargs.get('max_amount', None)
save_rule = kwargs['save_rule']
if min_amount is None:
min_amount = 0
elif min_amount < 0:
raise ValueError('min_amount (%s) < 0' % min_amount)
if max_amount is None:
max_amount = 0
elif max_amount < 0:
raise ValueError('max_amount (%s) < 0' % max_amount)
if min_amount and max_amount and min_amount > max_amount:
raise ValueError('min_amount (%s) > max_amount (%s)' % (min_amount, max_amount))
filtered_splits = splits.filtered_splits
if tx_desc:
# Need to do tx_desc filter ourselves
if is_regex:
filtered_splits = \
filtered_splits.filter(transaction__description__iregex=tx_desc)
else:
filtered_splits = \
filtered_splits.filter(transaction__description__icontains=tx_desc)
else:
# Any tx_desc filtering has already been done for us
tx_desc = splits.tx_desc
is_regex = splits.tx_desc_is_regex
tx_guids = filtered_splits.distinct().values('transaction__guid')
splits_real = Split.objects.filter(transaction__guid__in=tx_guids).exclude(account=account)
# ugh
if min_amount: min_amount -= Decimal('1e-8')
if max_amount: max_amount += Decimal('1e-8')
if min_amount and max_amount:
splits_real = splits_real.filter(
(Q(value_num__gte=F('value_denom') * min_amount) & Q(value_num__lte=F('value_denom') * max_amount)) |
(Q(value_num__lte=F('value_denom') * -min_amount) & Q(value_num__gte=F('value_denom') * -max_amount)))
elif min_amount:
splits_real = splits_real.filter(
(Q(value_num__gte=F('value_denom') * min_amount)) |
(Q(value_num__lte=F('value_denom') * -min_amount)))
elif max_amount:
splits_real = splits_real.filter(
(Q(value_num__lte=F('value_denom') * max_amount)) |
(Q(value_num__gte=F('value_denom') * -max_amount)))
split_guids = list(splits_real.distinct().values_list('guid', flat=True))
tx_count = len(split_guids)
modified_tx_count = 0
if tx_count > 0:
Lock.obtain()
try:
Split.objects.filter(guid__in=split_guids).update(account=opposing_account)
modified_tx_count = tx_count
finally:
Lock.release()
if save_rule and tx_desc:
rule = Rule()
rule.opposing_account_guid = opposing_account.guid
rule.match_tx_desc = tx_desc
rule.is_regex = is_regex
if min_amount: rule.min_amount = min_amount
if max_amount: rule.max_amount = max_amount
rule.save()
rule_account = RuleAccount()
rule_account.rule = rule
rule_account.account_guid = account.guid
rule_account.save()
return modified_tx_count