X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=market.py;h=e16641c476865bc3977ceaa8f30ccce5296925ab;hb=882d55e99489d9131b5171f23e505b0dfd1c8738;hp=ac3aa14a9213ba817db4a431cbc3ddd0649e460e;hpb=79ac9364372ad0cd99f3d29b7dbc2b79dcc84bf5;p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git diff --git a/market.py b/market.py index ac3aa14..e16641c 100644 --- a/market.py +++ b/market.py @@ -25,9 +25,11 @@ class Market: self.balances = BalanceStore(self) self.processor = Processor(self) - for key in ["user_id", "market_id", "report_path", "pg_config"]: + for key in ["user_id", "market_id", "pg_config"]: setattr(self, key, kwargs.get(key, None)) + self.report.log_market(self.args, self.user_id, self.market_id) + @classmethod def from_config(cls, config, args, **kwargs): config["apiKey"] = config.pop("key", None) @@ -39,14 +41,14 @@ class Market: def store_report(self): self.report.merge(Portfolio.report) date = datetime.now() - if self.report_path is not None: + if self.args.report_path is not None: self.store_file_report(date) - if self.pg_config is not None: + if self.pg_config is not None and self.args.report_db: self.store_database_report(date) def store_file_report(self, date): try: - report_file = "{}/{}_{}".format(self.report_path, date.isoformat(), self.user_id) + report_file = "{}/{}_{}".format(self.args.report_path, date.isoformat(), self.user_id) with open(report_file + ".json", "w") as f: f.write(self.report.to_json()) with open(report_file + ".log", "w") as f: