define constants for membership types

pull/183/head
Chris Veilleux 2019-06-11 16:48:50 -05:00
parent d07a0fec28
commit 8421868243
3 changed files with 16 additions and 4 deletions

View File

@ -9,5 +9,9 @@ from .entity.membership import Membership
from .entity.skill import AccountSkill
from .repository.account import AccountRepository
from .repository.agreement import AgreementRepository
from .repository.membership import MembershipRepository
from .repository.membership import (
MembershipRepository,
MONTHLY_MEMBERSHIP,
YEARLY_MEMBERSHIP
)
from .repository.skill import AccountSkillRepository

View File

@ -2,6 +2,9 @@ from selene.data.account import AccountMembership
from ..entity.membership import Membership
from ...repository_base import RepositoryBase
MONTHLY_MEMBERSHIP = 'Monthly Membership'
YEARLY_MEMBERSHIP = 'Yearly Membership'
class MembershipRepository(RepositoryBase):
def __init__(self, db):

View File

@ -1,15 +1,20 @@
from decimal import Decimal
from selene.data.account import Membership, MembershipRepository
from selene.data.account import (
Membership,
MembershipRepository,
MONTHLY_MEMBERSHIP,
YEARLY_MEMBERSHIP
)
monthly_membership = dict(
type='Monthly Supporter',
type=MONTHLY_MEMBERSHIP,
rate=Decimal('1.99'),
rate_period='monthly'
)
yearly_membership = dict(
type='Yearly Supporter',
type=YEARLY_MEMBERSHIP,
rate=Decimal('19.99'),
rate_period='yearly'
)