mirror of
https://github.com/RoboSats/robosats.git
synced 2024-12-13 19:06:26 +00:00
166 lines
7.3 KiB
Python
166 lines
7.3 KiB
Python
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)
|
|
secret = models.CharField(max_length=300, unique=False, null=True, default=None)
|
|
expires_at = models.DateTimeField()
|
|
amount = models.DecimalField(max_digits=9, decimal_places=4, 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)
|
|
|
|
|
|
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(MIN_TRADE), MaxValueValidator(MAX_TRADE)])
|
|
payment_method = models.CharField(max_length=30, null=False, default="Not specified")
|
|
|
|
# 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)])
|
|
t0_market_satoshis = models.PositiveBigIntegerField(null=True, validators=[MinValueValidator(MIN_TRADE), MaxValueValidator(MAX_TRADE)])
|
|
# explicit
|
|
satoshis = models.PositiveBigIntegerField(null=True, validators=[MinValueValidator(MIN_TRADE), MaxValueValidator(MAX_TRADE)])
|
|
|
|
# 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) # 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)
|
|
taker_bond = models.ForeignKey(LNPayment, related_name='taker_bond', on_delete=models.SET_NULL, null=True, default=None)
|
|
trade_escrow = models.ForeignKey(LNPayment, related_name='trade_escrow', on_delete=models.SET_NULL, null=True, default=None)
|
|
|
|
# buyer payment LN invoice
|
|
buyer_invoice = models.ForeignKey(LNPayment, related_name='buyer_invoice', on_delete=models.SET_NULL, null=True, default=None)
|
|
|
|
|
|
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]) # Will only store latest ratings
|
|
avg_rating = models.DecimalField(max_digits=4, decimal_places=1, default=None, null=True, validators=[MinValueValidator(0), MaxValueValidator(100)])
|
|
|
|
# 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')
|
|
|
|
@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('<img src="%s" width="50" height="50" />' % self.get_avatar())
|