aboutsummaryrefslogtreecommitdiffstats
path: root/python/collateral/jpm.py
blob: 8a5123c2963509ae1c66128bbea91e345d9beba2 (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
46
47
48
49
50
51
52
53
54
55
import datetime
import pandas as pd
from io import BytesIO
from pikepdf import Pdf
from . import DAILY_DIR
from .common import load_pdf


def load_file(d, fund):
    try:
        fname = next(
            (DAILY_DIR / fund / "JPM_reports").glob(f"CSCFTCSTMT-*-{d:%y%m%d}*.pdf")
        )
    except StopIteration:
        raise FileNotFoundError(f"JPM file not found for date {d}")
    return pd.read_excel(fname, skiprows=6, skipfooter=2)


paths = {
    # "Serenitas": ["NYops", "Margin Calls JPM"],
    "BowdSt": ["BowdoinOps", "Margin JPM"],
}


def load_file(d, fund):
    try:
        fname = next(
            (DAILY_DIR / fund / "JPM_reports").glob(
                f"CSCFTCSTMT-*-{d:%y%m%d}-909271_2.pdf"
            )
        )
    except StopIteration:
        raise FileNotFoundError(f"JPM file not found for date {d}")
    return fname


def get_collateral(d: datetime.date, fund):
    pdf_file = load_file(d, fund)
    collat_page = load_pdf(pdf_file, pages=True)[3]
    return float(get_col(pomme, 200, 300, 1000, 1100)[0].replace(",", ""))


def download_files(em, count=20, *, fund="BowdSt", **kwargs):
    if fund not in paths:
        return
    emails = em.get_msgs(path=paths[fund], count=count, subject__startswith="909271")
    DATA_DIR = DAILY_DIR / fund / "JPM_reports"
    for msg in emails:
        for attach in msg.attachments:
            fname = attach.name
            p = DATA_DIR / fname
            if not p.exists():
                stream = BytesIO(attach.content)
                pdf = Pdf.open(stream, password="tm64EO")
                pdf.save(p)