Skip to content

Commit 0b05cfb

Browse files
committed
feat(toggl): get last name from google
lookup in user info cache before querying google save in user info cache after query
1 parent e6449e8 commit 0b05cfb

File tree

1 file changed

+25
-0
lines changed

1 file changed

+25
-0
lines changed

notebooks/toggl-to-harvest.ipynb

+25
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
"metadata": {},
77
"outputs": [],
88
"source": [
9+
"from compiler_admin.services.google import user_info as google_user_info\n",
910
"import json\n",
1011
"import os\n",
1112
"from pathlib import Path\n",
@@ -124,6 +125,30 @@
124125
"source[\"First Name\"] = source[\"First Name\"].astype(\"category\")\n",
125126
"source.dtypes"
126127
]
128+
},
129+
{
130+
"cell_type": "code",
131+
"execution_count": null,
132+
"metadata": {},
133+
"outputs": [],
134+
"source": [
135+
"# Get cached last name or query from Google\n",
136+
"def get_last_name(email: str):\n",
137+
" user = USER_INFO.get(email)\n",
138+
" last_name = user.get(\"Last Name\") if user else None\n",
139+
" if last_name is None:\n",
140+
" user = google_user_info(email)\n",
141+
" last_name = user.get(\"Last Name\") if user else None\n",
142+
" if email in USER_INFO:\n",
143+
" USER_INFO[email].update(user)\n",
144+
" else:\n",
145+
" USER_INFO[email] = user\n",
146+
" return last_name\n",
147+
"\n",
148+
"source[\"Last Name\"] = source[\"Email\"].apply(get_last_name)\n",
149+
"source[\"Last Name\"] = source[\"Last Name\"].astype(\"category\")\n",
150+
"source.dtypes"
151+
]
127152
}
128153
],
129154
"metadata": {

0 commit comments

Comments
 (0)