mirror of
https://github.com/RoboSats/robosats.git
synced 2024-08-17 01:40:39 +03:00
New notifications
This commit is contained in:
parent
e9a49d31f1
commit
0175e73ca1
@ -210,6 +210,7 @@ class OrderAdmin(AdminChangeLinksMixin, admin.ModelAdmin):
|
||||
f"Dispute of order {order.id} solved successfully on favor of the maker",
|
||||
messages.SUCCESS,
|
||||
)
|
||||
send_notification.delay(order_id=order.id, message="dispute_closed")
|
||||
|
||||
else:
|
||||
self.message_user(
|
||||
@ -248,6 +249,7 @@ class OrderAdmin(AdminChangeLinksMixin, admin.ModelAdmin):
|
||||
f"Dispute of order {order.id} solved successfully on favor of the taker",
|
||||
messages.SUCCESS,
|
||||
)
|
||||
send_notification.delay(order_id=order.id, message="dispute_closed")
|
||||
|
||||
else:
|
||||
self.message_user(
|
||||
|
@ -10,6 +10,7 @@ from django.db import models
|
||||
from django.db.models.signals import pre_delete
|
||||
from django.dispatch import receiver
|
||||
from django.utils import timezone
|
||||
from api.tasks import send_notification
|
||||
|
||||
if config("TESTING", cast=bool, default=False):
|
||||
import random
|
||||
@ -91,10 +92,7 @@ class Order(models.Model):
|
||||
decimal_places=2,
|
||||
default=0,
|
||||
null=True,
|
||||
validators=[
|
||||
MinValueValidator(Decimal(-100)),
|
||||
MaxValueValidator(Decimal(999))
|
||||
],
|
||||
validators=[MinValueValidator(Decimal(-100)), MaxValueValidator(Decimal(999))],
|
||||
blank=True,
|
||||
)
|
||||
# explicit
|
||||
@ -352,6 +350,8 @@ class Order(models.Model):
|
||||
self.log(
|
||||
f"Order state went from {old_status}: <i>{Order.Status(old_status).label}</i> to {new_status}: <i>{Order.Status(new_status).label}</i>"
|
||||
)
|
||||
if new_status == Order.Status.FAI:
|
||||
send_notification.delay(order_id=self.id, message="lightning_failed")
|
||||
|
||||
|
||||
@receiver(pre_delete, sender=Order)
|
||||
|
@ -219,3 +219,47 @@ class Notifications:
|
||||
title = f"🛠️ Your order with ID {order.id} has been cancelled by the coordinator {config('COORDINATOR_ALIAS', cast=str, default='NoAlias')} for the upcoming maintenance stop."
|
||||
self.send_message(order, order.maker.robot, title)
|
||||
return
|
||||
|
||||
def dispute_closed(self, order):
|
||||
lang = order.maker.robot.telegram_lang_code
|
||||
if order.status == Order.Status.MLD:
|
||||
# Maker lost dispute
|
||||
looser = order.maker
|
||||
winner = order.taker
|
||||
elif order.status == Order.Status.TLD:
|
||||
# Taker lost dispute
|
||||
looser = order.taker
|
||||
winner = order.maker
|
||||
|
||||
lang = looser.robot.telegram_lang_code
|
||||
if lang == "es":
|
||||
title = f"⚖️ Hey {looser.username}, has perdido la disputa en la orden con ID {str(order.id)}."
|
||||
else:
|
||||
title = f"⚖️ Hey {looser.username}, you lost the dispute on your order with ID {str(order.id)}."
|
||||
self.send_message(order, looser.robot, title)
|
||||
|
||||
lang = winner.robot.telegram_lang_code
|
||||
if lang == "es":
|
||||
title = f"⚖️ Hey {winner.username}, has ganado la disputa en la orden con ID {str(order.id)}."
|
||||
else:
|
||||
title = f"⚖️ Hey {winner.username}, you won the dispute on your order with ID {str(order.id)}."
|
||||
self.send_message(order, winner.robot, title)
|
||||
|
||||
return
|
||||
|
||||
def lightning_failed(self, order):
|
||||
lang = order.maker.robot.telegram_lang_code
|
||||
if order.type == Order.Types.BUY:
|
||||
buyer = order.maker
|
||||
else:
|
||||
buyer = order.taker
|
||||
|
||||
if lang == "es":
|
||||
title = f"⚡❌ Hey {buyer.username}, el pago lightning en la order con ID {str(order.id)} ha fallado."
|
||||
description = "Intentalo de nuevo con una nueva factura o con otra wallet."
|
||||
else:
|
||||
title = f"⚡❌ Hey {buyer.username}, the lightning payment on your order with ID {str(order.id)} failed."
|
||||
description = "Try again with a new invoice or from another wallet."
|
||||
|
||||
self.send_message(order, buyer.robot, title, description)
|
||||
return
|
||||
|
@ -493,7 +493,7 @@ class OrderDetailSerializer(serializers.ModelSerializer):
|
||||
class ListNotificationSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = Notification
|
||||
fields = ("title", "description", "order_id")
|
||||
fields = ("title", "description", "order_id", "order_status")
|
||||
|
||||
|
||||
class OrderPublicSerializer(serializers.ModelSerializer):
|
||||
|
@ -303,4 +303,10 @@ def send_notification(order_id=None, chat_message_id=None, message=None):
|
||||
elif message == "coordinator_cancelled":
|
||||
notifications.coordinator_cancelled(order)
|
||||
|
||||
elif message == "dispute_closed":
|
||||
notifications.dispute_closed(order)
|
||||
|
||||
elif message == "lightning_failed":
|
||||
notifications.lightning_failed(order)
|
||||
|
||||
return
|
||||
|
@ -764,6 +764,7 @@ class NotificationsView(ListAPIView):
|
||||
data["title"] = str(notification.title)
|
||||
data["description"] = str(notification.description)
|
||||
data["order_id"] = notification.order.id
|
||||
data["order_status"] = notification.order.status
|
||||
|
||||
notification_data.append(data)
|
||||
|
||||
|
@ -1163,6 +1163,60 @@ class TradeTest(BaseAPITestCase):
|
||||
f"⚖️ Hey {data['taker_nick']}, a dispute has been opened on your order with ID {str(trade.order_id)}.",
|
||||
)
|
||||
|
||||
# def test_dispute_closed_maker_wins(self):
|
||||
# trade = Trade(self.client)
|
||||
# trade.publish_order()
|
||||
# trade.take_order()
|
||||
# trade.lock_taker_bond()
|
||||
# trade.lock_escrow(trade.taker_index)
|
||||
# trade.submit_payout_invoice(trade.maker_index)
|
||||
|
||||
# trade.change_order_status(Order.Status.TLD)
|
||||
|
||||
# trade.clean_orders()
|
||||
|
||||
# maker_headers = trade.get_robot_auth(trade.maker_index)
|
||||
# response = self.client.get(reverse("notifications"), **maker_headers)
|
||||
# self.assertResponse(response)
|
||||
# notifications_data = list(response.json())
|
||||
# self.assertEqual(notifications_data[0]["order_id"], trade.order_id)
|
||||
# self.assertEqual(
|
||||
# notifications_data[0]["title"],
|
||||
# f"⚖️ Hey {data['maker_nick']}, you won the dispute on your order with ID {str(trade.order_id)}."
|
||||
# )
|
||||
# taker_headers = trade.get_robot_auth(trade.taker_index)
|
||||
# response = self.client.get(reverse("notifications"), **taker_headers)
|
||||
# self.assertResponse(response)
|
||||
# notifications_data = list(response.json())
|
||||
# self.assertEqual(notifications_data[0]["order_id"], trade.order_id)
|
||||
# self.assertEqual(
|
||||
# notifications_data[0]["title"],
|
||||
# f"⚖️ Hey {data['taker_nick']}, you lost the dispute on your order with ID {str(trade.order_id)}."
|
||||
# )
|
||||
|
||||
def test_lightning_payment_failed(self):
|
||||
trade = Trade(self.client)
|
||||
trade.publish_order()
|
||||
trade.take_order()
|
||||
trade.lock_taker_bond()
|
||||
trade.lock_escrow(trade.taker_index)
|
||||
trade.submit_payout_invoice(trade.maker_index)
|
||||
|
||||
trade.change_order_status(Order.Status.FAI)
|
||||
|
||||
trade.clean_orders()
|
||||
|
||||
maker_headers = trade.get_robot_auth(trade.maker_index)
|
||||
maker_nick = read_file(f"tests/robots/{trade.maker_index}/nickname")
|
||||
response = self.client.get(reverse("notifications"), **maker_headers)
|
||||
self.assertResponse(response)
|
||||
notifications_data = list(response.json())
|
||||
self.assertEqual(notifications_data[0]["order_id"], trade.order_id)
|
||||
self.assertEqual(
|
||||
notifications_data[0]["title"],
|
||||
f"⚡❌ Hey {maker_nick}, the lightning payment on your order with ID {str(trade.order_id)} failed.",
|
||||
)
|
||||
|
||||
def test_withdraw_reward_after_unilateral_cancel(self):
|
||||
"""
|
||||
Tests withdraw rewards as taker after maker cancels order unilaterally
|
||||
|
@ -271,3 +271,10 @@ class Trade:
|
||||
order = Order.objects.get(id=self.order_id)
|
||||
order.expires_at = datetime.now()
|
||||
order.save()
|
||||
|
||||
@patch("api.tasks.send_notification.delay", send_notification)
|
||||
def change_order_status(self, status):
|
||||
# Change order expiry to now
|
||||
order = Order.objects.get(id=self.order_id)
|
||||
order.status = status
|
||||
order.save()
|
||||
|
Loading…
Reference in New Issue
Block a user