# 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)
