aboutsummaryrefslogtreecommitdiffstats
path: root/python/analytics/portfolio.py
blob: 3d89ae113be1d02d0371285b60e39b0a2499fe2c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
from .index import Index

class Portfolio:
    def __init__(self, trades):
        self.trades = trades
        self.index = next(t for t in trades if isinstance(t, Index))
        self.swaptions = (t for t in trades if isinstance(t, BlackSwaption))

    @property
    def pnl(self):
        return sum(t.pnl for t in self.trades)

    def set_original_pv(self):
        for t in self.trades:
            t.set_original_pv()

    @property
    def trade_date(self):
        return self.index.trade_date

    @trade_date.setter
    def trade_date(self, d):
        print("pomme")
        self.index.trade_date = d

    @property
    def ref(self):
        return self.index.ref

    @ref.setter
    def ref(self, val):
        self.index.ref = val

    @property
    def delta(self):
        """returns the equivalent protection notional"""
        return sum([getattr(t, 'delta', -t._direction) * t.notional for t in self.trades])

    @property
    def gamma(self):
        return sum([getattr(t, 'gamma', 0) * t.notional for t in self.trades])

    @property
    def dv01(self):
        return sum(t.dv01 for t in self.trades)