Add some acceptance tests

[perso/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Trader.git] / store.py

diff --git a/store.py b/store.py
index 467dd4b40cfd9ef05a9a0974c88e40f1560d8c1b..45a52339dc688bb9311451c476ec149b23f09a5f 100644 (file)
--- a/store.py
+++ b/store.py
@@ -409,6 +409,7 @@ class Portfolio:
     last_date = LockedVar(None)
     report = LockedVar(ReportStore(None, no_http_dup=True))
     worker = None
+    worker_tag = ""
     worker_started = False
     worker_notify = None
     callback = None
@@ -426,6 +427,7 @@ class Portfolio:
         cls.liquidities.start_lock()
         cls.report.start_lock()
 
+        cls.worker_tag = "[Worker] "
         cls.worker_started = True
         cls.worker.start()
 
@@ -445,7 +447,7 @@ class Portfolio:
             cls.worker_notify.wait()
             if cls.worker_started:
                 cls.worker_notify.clear()
-                cls.report.print_log("Fetching cryptoportfolio")
+                cls.report.print_log("[Worker] Fetching cryptoportfolio")
                 cls.get_cryptoportfolio(refetch=True)
                 cls.callback.set()
                 time.sleep(poll)
@@ -488,7 +490,7 @@ class Portfolio:
             cls.report.log_http_request(r.request.method,
                     r.request.url, r.request.body, r.request.headers, r)
         except Exception as e:
-            cls.report.log_error("get_cryptoportfolio", exception=e)
+            cls.report.log_error("{}get_cryptoportfolio".format(cls.worker_tag), exception=e)
             return
         try:
             cls.data.set(r.json(parse_int=D, parse_float=D))