yisroelbaum/PilznoProduction
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

functions for campaign model and ambassador goal

Browse source
This commit is contained in:
Yisroel Baum 2024-09-24 14:36:24 +03:00
parent d2a04f8b5a
commit 959c98fe79
1 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
app/campaigns/models.py
View file

@ -11,6 +11,7 @@ class AmbassadorMap(db.Model):
id = Column('id', INTEGER(), primary_key=True) id = Column('id', INTEGER(), primary_key=True)
campaign_id = Column('campaign_id', INTEGER(), nullable=False) campaign_id = Column('campaign_id', INTEGER(), nullable=False)
user_id = Column('user_id', INTEGER(), nullable=False) user_id = Column('user_id', INTEGER(), nullable=False)
goal = Column('ambassador_goal', INTEGER())
@ -26,7 +27,16 @@ class Campaign(db.Model):
from app.main.models import Donation from app.main.models import Donation
return Donation.query.filter_by(campaign_id=self.id).all() 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: def __repr__(self) -> str:
return f"{self.id} - {self.title}" return f"{self.id} - {self.title}"
# donation_id = Column(INTEGER, ForeignKey('donation.id'))