From e7d7c0e5645da35adcbfec9e51deb68f012c422f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Isma=C3=ABl=20Bouya?= Date: Sat, 7 Apr 2018 17:39:29 +0200 Subject: Acceptance test preparation Save some headers for http requests Wait for all threads after the end of main Simplify library imports for mocking --- market.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'market.py') diff --git a/market.py b/market.py index e16641c..7a37cf6 100644 --- a/market.py +++ b/market.py @@ -5,6 +5,7 @@ import psycopg2 from store import * from cachetools.func import ttl_cache from datetime import datetime +import datetime from retry import retry import portfolio @@ -28,7 +29,7 @@ class Market: 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) + self.report.log_market(self.args) @classmethod def from_config(cls, config, args, **kwargs): @@ -40,7 +41,7 @@ class Market: def store_report(self): self.report.merge(Portfolio.report) - date = datetime.now() + date = datetime.datetime.now() if self.args.report_path is not None: self.store_file_report(date) if self.pg_config is not None and self.args.report_db: -- cgit v1.2.3