feat: add promo codes and conditional events (#40)

* add extra column
* add conditional events
* refunds
* conditional events working
* adding promo codes
* promo codes logic

---------

Co-authored-by: dni ⚡ <office@dnilabs.com>

models.py

@@ -2,7 +2,29 @@ from datetime import datetime
 from typing import Optional
 
 from fastapi import Query
-from pydantic import BaseModel, EmailStr, root_validator
+from pydantic import BaseModel, EmailStr, Field, root_validator, validator
+
+
+class PromoCode(BaseModel):
+    code: str
+    discount_percent: float = 0.0
+    active: bool = True
+
+    # make the promo code uppercase
+    @validator("code")
+    def uppercase_code(cls, v):
+        return v.upper()
+
+    @validator("discount_percent")
+    def validate_discount_percent(cls, v):
+        assert 0 <= v <= 100, "Discount must be between 0 and 100."
+        return v
+
+
+class EventExtra(BaseModel):
+    promo_codes: list[PromoCode] = Field(default_factory=list)
+    conditional: bool = False
+    min_tickets: int = 1
 
 
 class CreateEvent(BaseModel):
@@ -16,20 +38,23 @@ class CreateEvent(BaseModel):
     amount_tickets: int = Query(..., ge=0)
     price_per_ticket: float = Query(..., ge=0)
     banner: str | None = None
+    extra: EventExtra = Field(default_factory=EventExtra)
 
 
 class CreateTicket(BaseModel):
     name: Optional[str] = None
     email: Optional[EmailStr] = None
     user_id: Optional[str] = None
-    
+    promo_code: Optional[str] = None
+    refund_address: Optional[str] = None
+
     @root_validator
     def validate_identifiers(cls, values):
         # Ensure either (name AND email) OR user_id is provided
         name = values.get('name')
         email = values.get('email')
         user_id = values.get('user_id')
-        
+
         if not user_id and not (name and email):
             raise ValueError("Either user_id or both name and email must be provided")
         if user_id and (name or email):
@@ -43,6 +68,7 @@ class Event(BaseModel):
     name: str
     info: str
     closing_date: str
+    canceled: bool = False
     event_start_date: str
     event_end_date: str
     currency: str
@@ -51,6 +77,14 @@ class Event(BaseModel):
     time: datetime
     sold: int = 0
     banner: str | None = None
+    extra: EventExtra = Field(default_factory=EventExtra)
+
+
+class TicketExtra(BaseModel):
+    applied_promo_code: str | None = None
+    sats_paid: int | None = None
+    refund_address: str | None = None
+    refunded: bool = False
 
 
 class Ticket(BaseModel):
@@ -64,3 +98,4 @@ class Ticket(BaseModel):
     paid: bool
     time: datetime
     reg_timestamp: datetime
+    extra: TicketExtra = Field(default_factory=TicketExtra)