42 lines
No EOL
1.4 KiB
Python
42 lines
No EOL
1.4 KiB
Python
from app import db
|
|
from flask import current_app
|
|
from flask_login import current_user
|
|
from sqlalchemy import TEXT, Column, Boolean, ForeignKey, TEXT, INTEGER, VARCHAR
|
|
import jwt
|
|
from datetime import datetime, timezone, timedelta
|
|
|
|
|
|
class AmbassadorMap(db.Model):
|
|
__tablename__ = 'ambassador_map'
|
|
id = Column('id', INTEGER(), primary_key=True)
|
|
campaign_id = Column('campaign_id', INTEGER(), nullable=False)
|
|
user_id = Column('user_id', INTEGER(), nullable=False)
|
|
goal = Column('ambassador_goal', INTEGER())
|
|
|
|
|
|
|
|
class Campaign(db.Model):
|
|
__tablename__ = 'campaign'
|
|
id = Column('id', INTEGER(), primary_key=True)
|
|
title = Column('title', TEXT(), nullable=False)
|
|
active = Column('active', Boolean(), nullable=False, default=True)
|
|
goal = Column('goal', INTEGER(), default=0)
|
|
archived = Column('archived', Boolean(), default=False)
|
|
|
|
def get_donations(self):
|
|
from app.main.models import Donation
|
|
return Donation.query.filter_by(campaign_id=self.id).all()
|
|
|
|
def get_amount_raised(self):
|
|
donations = self.get_donations()
|
|
total = 0
|
|
for donation in donations:
|
|
total += donation.amount
|
|
return total
|
|
|
|
def get_ambassadors(self):
|
|
return AmbassadorMap.query.filter_by(campaign_id=self.id).all()
|
|
|
|
|
|
def __repr__(self) -> str:
|
|
return f"{self.id} - {self.title}" |