diff options
| -rw-r--r-- | python/isosel_reports.py | 6 | ||||
| -rw-r--r-- | python/report_ops/remote.py | 17 |
2 files changed, 10 insertions, 13 deletions
diff --git a/python/isosel_reports.py b/python/isosel_reports.py index e1a9a9b7..f6d92f9e 100644 --- a/python/isosel_reports.py +++ b/python/isosel_reports.py @@ -20,8 +20,4 @@ if __name__ == "__main__": AccruedReport, AllReport, ): - report = report_cls(args.date) - try: - report.to_db() - except ValueError as e: - logger.info(e) + report_cls.to_db(args.date) diff --git a/python/report_ops/remote.py b/python/report_ops/remote.py index 5d66fd7f..944019e1 100644 --- a/python/report_ops/remote.py +++ b/python/report_ops/remote.py @@ -57,18 +57,19 @@ class Report: with cls._sftp.client.open(fname) as fh: return load_citco_report(fh, kd, cls.date_cols) - def to_db(self): - p = self.get_newest_report(self.date) - df = self.to_df(p) - with self._conn.cursor() as c: + @classmethod + def to_db(cls, date): + p = cls.get_newest_report(date) + df = cls.to_df(p) + with cls._conn.cursor() as c: c.execute( - f"DELETE FROM {self.table} WHERE period_end_date= %s", - (self.date,), + f"DELETE FROM {cls.table} WHERE period_end_date= %s", + (date,), ) - self._conn.commit() + cls._conn.commit() if "strategy" in df.columns: df["strategy"] = df["strategy"].str.replace("/M_|/SER_", "/", regex=True) - df.to_sql(self.table, dawn_engine, if_exists="append", index=False) + df.to_sql(cls.table, dawn_engine, if_exists="append", index=False) class AccruedReport( |