diff options
Diffstat (limited to 'python/analytics/option.py')
| -rw-r--r-- | python/analytics/option.py | 34 |
1 files changed, 25 insertions, 9 deletions
diff --git a/python/analytics/option.py b/python/analytics/option.py index 803f6500..c334a47e 100644 --- a/python/analytics/option.py +++ b/python/analytics/option.py @@ -1,13 +1,15 @@ import bottleneck as bn import datetime +import logging import math import numpy as np import pandas as pd import analytics from .black import black, Nx +from .exceptions import MissingDataError from .sabr import sabr -from .utils import GHquad, build_table +from .utils import GHquad, build_table, bus_day from .index import g, ForwardIndex, CreditIndex from .db import serenitas_engine, dawn_engine from .utils import memoize @@ -30,6 +32,8 @@ from scipy.integrate import quad from scipy.special import logit, expit +logger = logging.getLogger(__name__) + def calib(S0, fp, tilt, w, ctx): return expected_pv(tilt, w, S0, ctx) - fp @@ -94,22 +98,34 @@ class BlackSwaption(ForwardIndex): instance._orig_params = (rec.strike, index.factor, index.cumloss) return instance - def mark(self, source_list=[], surface_id=None, **args): + def mark(self, source_list=[], surface_id=None, **kwargs): ind = self.index ind.mark() # add None so that we always try everything source_list = source_list + [None] - vs = BlackSwaptionVolSurface(ind.index_type, - ind.series, - ind.tenor, - ind.value_date, - **args) + surface_date = kwargs.get("surface_date", ind.value_date) + i = 0 + while i < 5: + try: + vs = BlackSwaptionVolSurface(ind.index_type, + ind.series, + ind.tenor, + surface_date, + **kwargs) + + except MissingDataError as e: + logger.warning(str(e)) + surface_date -= bus_day + logger.info(f"trying {self.value_date - bus_day}") + i += 1 + else: + break if surface_id is None: for source in source_list: if len(vs.list(source, self.option_type)) >= 1: break else: - raise ValueError("No market data available for this day") + raise MissingDataError(f"{type(self).__name__}: No quote for type {self.option_type} and date {self.value_date}") surface_id = vs.list(source, self.option_type)[-1] try: self.sigma = float(vs[surface_id](self.T, np.log(self.moneyness))) @@ -508,7 +524,7 @@ class QuoteSurface(): self._quotes.loc[(self._quotes.quote_source == "GS") & (self._quotes['index'] =="HY"), ["pay_bid", "pay_offer", "rec_bid", "rec_offer"]] *=100 if self._quotes.empty: - raise ValueError(f"No quotes for date: {value_date}") + raise MissingDataError(f"{type(self).__name__}: No market quote for date {value_date}") self._quotes['quotedate'] = (self._quotes['quotedate'].dt. tz_convert('America/New_York'). dt.tz_localize(None)) |