Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions docassemble/EFSPIntegration/efm_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
DADateTime,
as_datetime,
reconsider,
current_context,
)
from docassemble.AssemblyLine.al_document import ALDocumentBundle
from docassemble.AssemblyLine.al_general import ALIndividual
Expand Down Expand Up @@ -183,10 +184,15 @@ def __init__(

self.credentials_code_block = credentials_code_block

interview_name = f"{current_context().filename}"
interview_name = (
interview_name.replace(".", "-").replace(":", "-").replace("/", "-")[:72]
)
super().__init__(
url=url,
api_key=api_key,
default_jurisdiction=default_jurisdiction,
interview_name=interview_name,
logger=DALogger(logging.getLogger("docassemble")),
)

Expand Down
16 changes: 15 additions & 1 deletion docassemble/EFSPIntegration/py_efsp_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,13 @@ class EfspConnection:
"""A python client that communicates with the E-file proxy server."""

def __init__(
self, *, url: str, api_key: str, default_jurisdiction: str = None, logger=None
self,
*,
url: str,
api_key: str,
default_jurisdiction: str = None,
interview_name: str = None,
logger=None,
):
"""
Args:
Expand All @@ -133,6 +139,7 @@ def __init__(
self.active_token = None
# Keep one uuid for the whole of this class's life, i.e. the session
self.session_id = str(uuid4())
self.interview_name = interview_name
if logger is None:
logger = logging.getLogger()
self.logger = LoggerWithContext(logger, {"session_id": self.session_id})
Expand Down Expand Up @@ -170,6 +177,7 @@ def _send(self, to_send: Request, *, req_id: Optional[UUID] = None) -> ApiRespon
req_id = uuid4()
to_send.headers["efsp-request-id"] = str(req_id)
to_send.headers["efsp-session-id"] = self.get_session_id()
to_send.headers["efsp-interview-name"] = self.get_interview_name()
self.get_logger().info(
f"Calling {to_send.method} on {to_send.url}", extra={"req-id": str(req_id)}
)
Expand All @@ -181,6 +189,12 @@ def get_session_id(self):
self.session_id = str(uuid4())
return self.session_id

def get_interview_name(self):
if hasattr(self, "interview_name"):
return self.interview_name
else:
return None

def get_logger(self):
if not hasattr(self, "logger"):
self.logger = LoggerWithContext(
Expand Down
Loading