52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
# Mycroft Server - Backend
|
|
# Copyright (C) 2019 Mycroft AI Inc
|
|
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
#
|
|
# This file is part of the Mycroft Server.
|
|
#
|
|
# The Mycroft Server is free software: you can redistribute it and/or
|
|
# modify it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
from decimal import Decimal
|
|
|
|
from selene.data.account import (
|
|
Membership,
|
|
MembershipRepository,
|
|
MONTHLY_MEMBERSHIP,
|
|
YEARLY_MEMBERSHIP,
|
|
)
|
|
|
|
monthly_membership = dict(
|
|
type=MONTHLY_MEMBERSHIP, rate=Decimal("1.99"), rate_period="monthly"
|
|
)
|
|
|
|
yearly_membership = dict(
|
|
type=YEARLY_MEMBERSHIP, rate=Decimal("19.99"), rate_period="yearly"
|
|
)
|
|
|
|
|
|
def insert_memberships(db):
|
|
monthly = Membership(**monthly_membership)
|
|
yearly = Membership(**yearly_membership)
|
|
membership_repository = MembershipRepository(db)
|
|
monthly.id = membership_repository.add(monthly)
|
|
yearly.id = membership_repository.add(yearly)
|
|
|
|
return monthly, yearly
|
|
|
|
|
|
def delete_memberships(db, memberships):
|
|
membership_repository = MembershipRepository(db)
|
|
for membership in memberships:
|
|
membership_repository.remove(membership)
|