Subscription.py

from Frequency import FrequencyType
from tinydb.queries import Query
from Content import ContentType
from db import DB

class Subscription:    
    def __init__(self) -> None:
        self.db = DB().db.table('Subscription')
        self.SubscriptionQuery = Query()

    def searchSubscriptionByFrequency(self, frequency: FrequencyType):
        return self.db.search(self.SubscriptionQuery()['frequency'] == frequency.name)

    def createSubscription(self, username, contentType: ContentType, frequency: FrequencyType):
        self.db.insert({
            'username': username,
            'frequency': frequency.name,
            'contentType': contentType.name
        })

    def validateSubscription(self, username, contentType: ContentType):
        if self.db.search((self.SubscriptionQuery['username'] == username) & (self.SubscriptionQuery['contentType'] == contentType)):
            return True
        return False

    def getAllSubscriptions(self, username):
        return self.db.search(self.SubscriptionQuery['username'] == username)

    def delSubscription(self, username, contentType: ContentType):
        self.db.remove((SubscriptionQuery['username'] == username) & (SubscriptionQuery['contentType'] == contentType.name))

    def updateSubscription(self, username, contentType: ContentType, frequency: FrequencyType):
        self.db.update({
            'frequency': frequency.name
        }, (self.SubscriptionQuery['username'] == username) & (self.SubscriptionQuery['contentType'] == contentType.name))