diff options
Diffstat (limited to 'python/reallocate_iam.py')
| -rw-r--r-- | python/reallocate_iam.py | 13 |
1 files changed, 5 insertions, 8 deletions
diff --git a/python/reallocate_iam.py b/python/reallocate_iam.py index b0801393..2846938d 100644 --- a/python/reallocate_iam.py +++ b/python/reallocate_iam.py @@ -84,7 +84,7 @@ class IAMDeal(Deal, deal_type=DealType.IAM, table_name="iams"): def to_globeop(self, action): obj = super().to_globeop(action) - obj["CallNoticeIndicator"] = "24H" if action == "NEW" else None + obj["CallNoticeIndicator"] = "24H" if not self.maturity else None obj["TransactionIndicator"] = "DEPOSIT" if obj["StartMoney"] > 0 else "LOAN" obj["StartMoney"] = abs(obj["StartMoney"]) obj["DealFunction"] = "OTC" @@ -185,7 +185,6 @@ def build_iam(fund: str, cob: datetime.date, conn, upload: bool): def parse_args(): - """Parses command line arguments""" parser = argparse.ArgumentParser(description="Generate IAM file for globeop") parser.add_argument( "cob", @@ -194,16 +193,14 @@ def parse_args(): default=prev_business_day(datetime.date.today()), ) parser.add_argument("-n", "--no-upload", action="store_true", help="do not upload") + parser.add_argument( + "-b", "--backfill", action="store_true", help="backfill an old date" + ) return parser.parse_args() -def main(): - """Generates IAM files for globeop""" +if __name__ == "__main__": conn = dbconn("dawndb") args = parse_args() for fund in ("SERCGMAST", "BOWDST"): build_iam(fund, args.cob, conn, not args.no_upload) - - -if __name__ == "__main__": - main() |