#! /usr/bin/env python
# -*- coding: utf-8 -*-

from sqlalchemy import Index

from newspipe.bootstrap import db
from newspipe.models.right_mixin import RightMixin


class Category(db.Model, RightMixin):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String())

    # relationships
    user_id = db.Column(db.Integer, db.ForeignKey("user.id"))
    feeds = db.relationship("Feed", cascade="all,delete-orphan")
    articles = db.relationship("Article", cascade="all,delete-orphan")

    # index
    idx_category_uid = Index("user_id")

    # api whitelists
    @staticmethod
    def _fields_base_read():
        return {"id", "user_id"}

    @staticmethod
    def _fields_base_write():
        return {"name"}