Added tracing for requests
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
@@ -2,6 +2,8 @@
|
||||
import os
|
||||
import logging
|
||||
import tempfile
|
||||
from requests_opentracing import SessionTracing
|
||||
import opentracing # ez kell ide hogy a session tracer jolegyen
|
||||
import requests
|
||||
import time
|
||||
from urllib.parse import urljoin
|
||||
@@ -13,6 +15,7 @@ from classifier_cache import ClassifierCache
|
||||
|
||||
class MagicDoer:
|
||||
classifier_cache = ClassifierCache(Config.MODEL_INFO_URL)
|
||||
requests_session = SessionTracing(propagate=True)
|
||||
|
||||
@classmethod
|
||||
def run_everything(cls, parameters: dict) -> dict:
|
||||
@@ -25,7 +28,7 @@ class MagicDoer:
|
||||
object_path = urljoin(Config.STORAGE_SERVICE_URL, f"object/{tag}")
|
||||
|
||||
logging.info(f"Downloading sample: {tag} from {object_path}")
|
||||
r = requests.get(object_path)
|
||||
r = cls.requests_session.get(object_path)
|
||||
with open(sample_file_handle, 'wb') as f:
|
||||
f.write(r.content)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user