from django.db import models from django.contrib.auth.models import User from django.core.validators import MaxValueValidator, MinValueValidator, validate_comma_separated_integer_list from django.db.models.signals import post_save, pre_delete from django.dispatch import receiver from django.utils.html import mark_safe from pathlib import Path ############################# # TODO # Load hparams from .env file MIN_TRADE = 10*1000 #In sats MAX_TRADE = 500*1000 FEE = 0.002 # Trade fee in % BOND_SIZE = 0.01 # Bond in % class LNPayment(models.Model): class Types(models.IntegerChoices): NORM = 0, 'Regular invoice' # Only outgoing HTLCs will be regular invoices (Non-hodl) HODL = 1, 'Hodl invoice' class Concepts(models.IntegerChoices): MAKEBOND = 0, 'Maker bond' TAKEBOND = 1, 'Taker-buyer bond' TRESCROW = 2, 'Trade escrow' PAYBUYER = 3, 'Payment to buyer' class Status(models.IntegerChoices): INVGEN = 0, 'Hodl invoice was generated' LOCKED = 1, 'Hodl invoice has HTLCs locked' CHRGED = 2, 'Hodl invoice was charged' RETNED = 3, 'Hodl invoice was returned' MISSNG = 4, 'Buyer invoice is missing' IVALID = 5, 'Buyer invoice is valid' INPAID = 6, 'Buyer invoice was paid' INFAIL = 7, 'Buyer invoice routing failed' # payment use case type = models.PositiveSmallIntegerField(choices=Types.choices, null=False, default=Types.HODL) concept = models.PositiveSmallIntegerField(choices=Concepts.choices, null=False, default=Concepts.MAKEBOND) status = models.PositiveSmallIntegerField(choices=Status.choices, null=False, default=Status.INVGEN) # payment details invoice = models.CharField(max_length=300, unique=False, null=True, default=None, blank=True) secret = models.CharField(max_length=300, unique=False, null=True, default=None, blank=True) expires_at = models.DateTimeField() amount = models.PositiveBigIntegerField(validators=[MinValueValidator(MIN_TRADE*BOND_SIZE), MaxValueValidator(MAX_TRADE*(1+BOND_SIZE+FEE))]) # payment relationals sender = models.ForeignKey(User, related_name='sender', on_delete=models.CASCADE, null=True, default=None) receiver = models.ForeignKey(User, related_name='receiver', on_delete=models.CASCADE, null=True, default=None) def __str__(self): # Make relational back to ORDER return (f'HTLC {self.id}: {self.Concepts(self.concept).label}') class Order(models.Model): class Types(models.IntegerChoices): BUY = 0, 'BUY' SELL = 1, 'SELL' class Currencies(models.IntegerChoices): USD = 1, 'USD' EUR = 2, 'EUR' ETH = 3, 'ETH' class Status(models.IntegerChoices): WFB = 0, 'Waiting for bond' PUB = 1, 'Published in order book' DEL = 2, 'Deleted from order book' TAK = 3, 'Taken' UCA = 4, 'Unilaterally cancelled' RET = 5, 'Returned to order book' # Probably same as 1 in most cases. WF2 = 6, 'Waiting for trade collateral and buyer invoice' WTC = 7, 'Waiting only for trade collateral' WBI = 8, 'Waiting only for buyer invoice' EXF = 9, 'Exchanging fiat / In chat' CCA = 10, 'Collaboratively cancelled' FSE = 11, 'Fiat sent' FCO = 12, 'Fiat confirmed' SUC = 13, 'Sucessfully settled' FAI = 14, 'Failed lightning network routing' UPI = 15, 'Updated invoice' DIS = 16, 'In dispute' MLD = 17, 'Maker lost dispute' TLD = 18, 'Taker lost dispute' EXP = 19, 'Expired' # order info status = models.PositiveSmallIntegerField(choices=Status.choices, null=False, default=Status.WFB) created_at = models.DateTimeField(auto_now_add=True) expires_at = models.DateTimeField() # order details type = models.PositiveSmallIntegerField(choices=Types.choices, null=False) currency = models.PositiveSmallIntegerField(choices=Currencies.choices, null=False) amount = models.DecimalField(max_digits=9, decimal_places=4, validators=[MinValueValidator(0.00001)]) payment_method = models.CharField(max_length=30, null=False, default="not specified", blank=True) # order pricing method. A explicit amount of sats, or a relative premium above/below market. is_explicit = models.BooleanField(default=False, null=False) # marked to marked premium = models.DecimalField(max_digits=5, decimal_places=2, default=0, null=True, validators=[MinValueValidator(-100), MaxValueValidator(999)], blank=True) t0_market_satoshis = models.PositiveBigIntegerField(null=True, validators=[MinValueValidator(MIN_TRADE), MaxValueValidator(MAX_TRADE)], blank=True) # explicit satoshis = models.PositiveBigIntegerField(null=True, validators=[MinValueValidator(MIN_TRADE), MaxValueValidator(MAX_TRADE)], blank=True) # order participants maker = models.ForeignKey(User, related_name='maker', on_delete=models.CASCADE, null=True, default=None) # unique = True, a maker can only make one order taker = models.ForeignKey(User, related_name='taker', on_delete=models.SET_NULL, null=True, default=None, blank=True) # unique = True, a taker can only take one order # order collateral maker_bond = models.ForeignKey(LNPayment, related_name='maker_bond', on_delete=models.SET_NULL, null=True, default=None, blank=True) taker_bond = models.ForeignKey(LNPayment, related_name='taker_bond', on_delete=models.SET_NULL, null=True, default=None, blank=True) trade_escrow = models.ForeignKey(LNPayment, related_name='trade_escrow', on_delete=models.SET_NULL, null=True, default=None, blank=True) # buyer payment LN invoice buyer_invoice = models.ForeignKey(LNPayment, related_name='buyer_invoice', on_delete=models.SET_NULL, null=True, default=None, blank=True) class Profile(models.Model): user = models.OneToOneField(User,on_delete=models.CASCADE) # Ratings stored as a comma separated integer list total_ratings = models.PositiveIntegerField(null=False, default=0) latest_ratings = models.CharField(max_length=999, null=True, default=None, validators=[validate_comma_separated_integer_list], blank=True) # Will only store latest ratings avg_rating = models.DecimalField(max_digits=4, decimal_places=1, default=None, null=True, validators=[MinValueValidator(0), MaxValueValidator(100)], blank=True) # Disputes num_disputes = models.PositiveIntegerField(null=False, default=0) lost_disputes = models.PositiveIntegerField(null=False, default=0) # RoboHash avatar = models.ImageField(default="static/assets/misc/unknown_avatar.png", verbose_name='Avatar', blank=True) @receiver(post_save, sender=User) def create_user_profile(sender, instance, created, **kwargs): if created: Profile.objects.create(user=instance) @receiver(post_save, sender=User) def save_user_profile(sender, instance, **kwargs): instance.profile.save() @receiver(pre_delete, sender=User) def del_avatar_from_disk(sender, instance, **kwargs): avatar_file=Path('frontend/' + instance.profile.avatar.url) avatar_file.unlink() # FIX deleting user fails if avatar is not found def __str__(self): return self.user.username # to display avatars in admin panel def get_avatar(self): if not self.avatar: return 'static/assets/misc/unknown_avatar.png' return self.avatar.url # method to create a fake table field in read only mode def avatar_tag(self): return mark_safe('' % self.get_avatar())