mirror of
https://github.com/eitchtee/WYGIWYH.git
synced 2026-03-18 23:44:03 +01:00
120 lines
5.3 KiB
Python
120 lines
5.3 KiB
Python
from cachalot.api import cachalot_disabled, invalidate
|
|
from dateutil.relativedelta import relativedelta
|
|
from procrastinate.contrib.django import app
|
|
from simpleeval import EvalWithCompoundTypes
|
|
|
|
from apps.accounts.models import Account
|
|
from apps.rules.models import TransactionRule, TransactionRuleAction
|
|
from apps.transactions.models import Transaction, TransactionCategory, TransactionTag
|
|
|
|
|
|
@app.task
|
|
def check_for_transaction_rules(
|
|
instance_id: int,
|
|
signal,
|
|
):
|
|
with cachalot_disabled():
|
|
|
|
instance = Transaction.objects.get(id=instance_id)
|
|
|
|
context = {
|
|
"account_name": instance.account.name,
|
|
"account_id": instance.account.id,
|
|
"account_group_name": (
|
|
instance.account.group.name if instance.account.group else None
|
|
),
|
|
"account_group_id": (
|
|
instance.account.group.id if instance.account.group else None
|
|
),
|
|
"category_name": instance.category.name if instance.category else None,
|
|
"category_id": instance.category.id if instance.category else None,
|
|
"tag_names": [x.name for x in instance.tags.all()],
|
|
"tag_ids": [x.id for x in instance.tags.all()],
|
|
"is_expense": instance.type == Transaction.Type.EXPENSE,
|
|
"is_income": instance.type == Transaction.Type.INCOME,
|
|
"is_paid": instance.is_paid,
|
|
"description": instance.description,
|
|
"amount": instance.amount,
|
|
"notes": instance.notes,
|
|
"date": instance.date,
|
|
"reference_date": instance.reference_date,
|
|
}
|
|
|
|
functions = {"relative_delta": relativedelta}
|
|
|
|
simple = EvalWithCompoundTypes(names=context, functions=functions)
|
|
|
|
if signal == "transaction_created":
|
|
rules = TransactionRule.objects.filter(active=True, on_create=True)
|
|
elif signal == "transaction_updated":
|
|
rules = TransactionRule.objects.filter(active=True, on_update=True)
|
|
else:
|
|
rules = TransactionRule.objects.filter(active=True)
|
|
|
|
for rule in rules:
|
|
if simple.eval(rule.trigger):
|
|
print("True!")
|
|
for action in rule.actions.all():
|
|
if action.field in [
|
|
TransactionRuleAction.Field.type,
|
|
TransactionRuleAction.Field.is_paid,
|
|
TransactionRuleAction.Field.date,
|
|
TransactionRuleAction.Field.reference_date,
|
|
TransactionRuleAction.Field.amount,
|
|
TransactionRuleAction.Field.description,
|
|
TransactionRuleAction.Field.notes,
|
|
]:
|
|
print(1)
|
|
setattr(
|
|
instance,
|
|
action.field,
|
|
simple.eval(action.value),
|
|
)
|
|
|
|
elif action.field == TransactionRuleAction.Field.account:
|
|
print(2)
|
|
value = simple.eval(action.value)
|
|
if isinstance(value, int):
|
|
account = Account.objects.get(id=value)
|
|
instance.account = account
|
|
elif isinstance(value, str):
|
|
account = Account.objects.filter(name=value).first()
|
|
instance.account = account
|
|
|
|
elif action.field == TransactionRuleAction.Field.category:
|
|
print(3)
|
|
value = simple.eval(action.value)
|
|
if isinstance(value, int):
|
|
category = TransactionCategory.objects.get(id=value)
|
|
instance.category = category
|
|
elif isinstance(value, str):
|
|
category = TransactionCategory.objects.get(name=value)
|
|
instance.category = category
|
|
|
|
elif action.field == TransactionRuleAction.Field.tags:
|
|
print(4)
|
|
value = simple.eval(action.value)
|
|
print(value, action.value)
|
|
if isinstance(value, list):
|
|
# Clear existing tags
|
|
instance.tags.clear()
|
|
for tag_value in value:
|
|
if isinstance(tag_value, int):
|
|
tag = TransactionTag.objects.get(id=tag_value)
|
|
instance.tags.add(tag)
|
|
elif isinstance(tag_value, str):
|
|
tag = TransactionTag.objects.get(name=tag_value)
|
|
instance.tags.add(tag)
|
|
|
|
elif isinstance(value, (int, str)):
|
|
# If a single value is provided, treat it as a single tag
|
|
instance.tags.clear()
|
|
if isinstance(value, int):
|
|
tag = TransactionTag.objects.get(id=value)
|
|
else:
|
|
tag = TransactionTag.objects.get(name=value)
|
|
|
|
instance.tags.add(tag)
|
|
|
|
instance.save()
|