Source code for richy.shares.models

from django.core.cache import cache
from django.db import models

from ..core.models import BaseDividend, Item




class Share(Item):
    item_type = Item.Type.SHARE

    def is_share(self):
        return True

    def is_index(self):
        return False

    def is_etf(self):
        return False

    def is_coin(self):
        return False



    def update_cache(self):
        super().update_cache()

        from ..transactions.transactions import Performance

        cache.delete("shares-transaction-performance-and-assets")
        cache.delete("shares-transaction-profit-performance")
        Performance("share", profit=False, use_cache=False).get_data()
        Performance("share", profit=True, use_cache=False).get_data()






class Dividend(BaseDividend):
    share = models.ForeignKey(Share, on_delete=models.CASCADE)