feat: add event_logger to test_connection and create_database commands (#13468)

Co-authored-by: Beto Dealmeida <roberto@dealmeida.net>
This commit is contained in:
Hugh A. Miles II
2021-03-09 08:17:13 -05:00
committed by GitHub
parent 9b8e2555bf
commit c91c45574b
6 changed files with 137 additions and 22 deletions

View File

@@ -104,7 +104,7 @@ class AbstractEventLogger(ABC):
def log_with_context( # pylint: disable=too-many-locals
self,
action: str,
duration: timedelta,
duration: Optional[timedelta] = None,
object_ref: Optional[str] = None,
log_to_statsd: bool = True,
**payload_override: Optional[Dict[str, Any]],
@@ -112,6 +112,9 @@ class AbstractEventLogger(ABC):
from superset.views.core import get_form_data
referrer = request.referrer[:1000] if request.referrer else None
duration_ms = int(duration.total_seconds() * 1000) if duration else None
try:
user_id = g.user.get_id()
except Exception as ex: # pylint: disable=broad-except
@@ -158,7 +161,7 @@ class AbstractEventLogger(ABC):
records=records,
dashboard_id=dashboard_id,
slice_id=slice_id,
duration_ms=int(duration.total_seconds() * 1000),
duration_ms=duration_ms,
referrer=referrer,
)