# ---------------- User discovery ----------------
def get_current_job_instance_id():
ws_id = fabric.get_notebook_workspace_id()
nb_id = fabric.get_artifact_id()
client = FabricRestClient()
resp = client.get(f"/v1/workspaces/{ws_id}/items/{nb_id}/jobs/instances").json()
runs = resp.get("value", [])
if not runs:
return None, None
active = [r for r in runs if str(r.get("status","")).lower() in {"inprogress","notstarted"}]
pick = max(active or runs, key=lambda r: r.get("startTimeUtc",""))
return pick.get("id"), pick
def _pluck_userish(obj):
"""
Try common shapes:
{createdBy: {displayName, userPrincipalName, id}}
{requestedBy: {...}}, {submittedBy: {...}}, {initiatedBy: {...}}
{owner: {...}}, {principal: {...}}
"""
if not isinstance(obj, dict):
return None
candidates = ["createdBy", "requestedBy", "submittedBy", "initiatedBy", "owner", "principal"]
for k in candidates:
if k in obj and isinstance(obj[k], dict):
u = obj[k]
return {
"displayName": u.get("displayName"),
"userPrincipalName": u.get("userPrincipalName") or u.get("upn"),
"id": u.get("id") or u.get("objectId")
}
return None
def find_user_from_job(details):
# check top-level common fields
user = _pluck_userish(details)
if user:
return user
# scan nested dicts one level down
for v in details.values():
if isinstance(v, dict):
user = _pluck_userish(v)
if user:
return user
return None
def get_current_user_via_graph():
"""
Resolve the current interactive user via Microsoft Graph /me.
Requires that the Fabric runtime can mint a token for Graph.
"""
try:
from notebookutils import mssparkutils # available in Fabric runtime
if not token:
return None
headers = {"Authorization": f"Bearer {token}"}
if r.status_code == 200:
me = r.json()
return {
"displayName": me.get("displayName"),
"userPrincipalName": me.get("userPrincipalName"),
"id": me.get("id"),
"_source": "graph"
}
else:
print("Graph /me call failed:", r.status_code, r.text[:300])
except Exception as e:
print("Graph lookup failed:", e)
return None
def get_user_from_runtime_context():
"""
Fallback for manual runs: read runner from Fabric runtime context.
Uses keys observed in your tenant: userName (display) and userId (object id).
"""
try:
from notebookutils import mssparkutils
ctx = mssparkutils.runtime.context or {}
display_name = ctx.get("userName")
object_id = ctx.get("userId")
if display_name or object_id:
return {"displayName": display_name, "userPrincipalName": None, "id": object_id, "_source": "runtime_context"}
except Exception:
pass
return None
# ---------------- Run & print ----------------
job_instance_id, details = get_current_job_instance_id()
print("Job Instance ID:", job_instance_id)
run_started_by_name = None # capture for logging later
if details:
invoke_type = details.get("invokeType")
print("InvokeType:", invoke_type)
# -------- Try to resolve the user who ran it --------
user = find_user_from_job(details)
if user and (user.get("displayName") or user.get("userPrincipalName")):
run_started_by_name = (user.get('displayName') or user.get('userPrincipalName'))
print("Run started by (from job):",
f"{user.get('displayName') or ''}".strip(),
f"({user.get('userPrincipalName')})" if user.get("userPrincipalName") else "")
else:
# Fallback 1: runtime context (works in your tenant)
user = get_user_from_runtime_context()
if user and (user.get("displayName") or user.get("userPrincipalName")):
run_started_by_name = (user.get('displayName') or user.get('userPrincipalName'))
print("Run started by (from context):",
f"{user.get('displayName') or ''}".strip(),
f"({user.get('userPrincipalName')})" if user.get("userPrincipalName") else "")
else:
# Fallback 2: ask Graph who is running this notebook interactively
user = get_current_user_via_graph()
if user:
run_started_by_name = (user.get('displayName') or user.get('userPrincipalName'))
print("Run started by (from Graph):",
f"{user.get('displayName') or ''}".strip(),
f"({user.get('userPrincipalName')})" if user.get("userPrincipalName") else "")
else:
print("Run started by: Unknown (no user info in job metadata, runtime context, or Graph)")
######################################################################################
# ----------- Prepare Log Parameters (safe) -----------
if trigger_type.lower() == "pipelineactivity":
trigger_type = "Orchestrator Pipeline"
trigger_name = trigger_name
elif trigger_type.lower() == "1":
trigger_type = "Scheduled"
trigger_name = "Scheduling Module"
elif trigger_type.lower() == "manual":
trigger_type = "Manual"
trigger_name = run_started_by_name
else:
trigger_type = trigger_type
trigger_name = "UNKNOWN"
