Skip to content

Commit 5678b37

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 5de0955 commit 5678b37

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",
@@ -126,6 +127,30 @@
126127
"source[\"First Name\"] = source[\"First Name\"].astype(\"category\")\n",
127128
"source.dtypes"
128129
]
130+
},
131+
{
132+
"cell_type": "code",
133+
"execution_count": null,
134+
"metadata": {},
135+
"outputs": [],
136+
"source": [
137+
"# Get cached last name or query from Google\n",
138+
"def get_last_name(email: str):\n",
139+
" user = USER_INFO.get(email)\n",
140+
" last_name = user.get(\"Last Name\") if user else None\n",
141+
" if last_name is None:\n",
142+
" user = google_user_info(email)\n",
143+
" last_name = user.get(\"Last Name\") if user else None\n",
144+
" if email in USER_INFO:\n",
145+
" USER_INFO[email].update(user)\n",
146+
" else:\n",
147+
" USER_INFO[email] = user\n",
148+
" return last_name\n",
149+
"\n",
150+
"source[\"Last Name\"] = source[\"Email\"].apply(get_last_name)\n",
151+
"source[\"Last Name\"] = source[\"Last Name\"].astype(\"category\")\n",
152+
"source.dtypes"
153+
]
129154
}
130155
],
131156
"metadata": {

0 commit comments

Comments
 (0)