From 4f5967f41a893c40d8d2b33ea16e8b3a968c89ca Mon Sep 17 00:00:00 2001
From: Rantaharju Jarno <jarno.rantaharju@aalto.fi>
Date: Mon, 2 Dec 2024 10:51:27 +0200
Subject: [PATCH] Add a function to list sections in MyActivity

---
 niimpy/reading/google_takeout.py | 44 ++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)

diff --git a/niimpy/reading/google_takeout.py b/niimpy/reading/google_takeout.py
index 76e21b4a..ba869292 100644
--- a/niimpy/reading/google_takeout.py
+++ b/niimpy/reading/google_takeout.py
@@ -1084,12 +1084,30 @@ def myactivity(zip_filename, section, start_date=None, end_date=None):
                 data.append({"timestamp": timestamp, "description": text})
     
     df = pd.DataFrame(data)
+    if len(df) == 0:
+        return df
     df["timestamp"] = pd.to_datetime(df["timestamp"], format='%b %d, %Y, %I:%M:%S %p %Z', utc=True)
     df["timestamp"] = df["timestamp"].dt.tz_convert('EET')
     df.set_index("timestamp", inplace=True)
     return df
 
 
+def list_myactivity_sections(zip_filename):
+    """ List sections in the My Activity in a given Google Takeout zip file.
+    """
+    data_path = os.path.join("Takeout", "My Activity")
+
+    with ZipFile(zip_filename) as zip_file:
+        filenames = zip_file.namelist()
+        sections = set()
+        for filename in filenames:
+            if not filename.startswith(data_path):
+                continue
+            sections.add(filename.split("/")[2])
+    return sections
+
+
+
 def YouTube(zip_filename, start_date=None, end_date=None):
     df = myactivity(zip_filename, "YouTube", start_date, end_date)
     activity_strings ={
@@ -1196,3 +1214,29 @@ def Search(zip_filename, start_date=None, end_date=None):
     return df
 
 
+def Maps(zip_filename, start_date=None, end_date=None):
+    """ Read Google Maps history from Google Takeout zip file.
+    """
+    df = myactivity(zip_filename, "Maps", start_date, end_date)
+    activity_strings ={
+        "Viewed ": "Viewed",
+        "Used ": "Used",
+        "Directions to ": "Directions",
+        "Searched for ": "Searched",
+        "Explored ": "Explored",
+    }
+
+    for activity_string, activity_type in activity_strings.items():
+        rows = df["description"].str.startswith(activity_string)
+        df.loc[rows, "activity_type"] = activity_type
+        df.loc[rows, "description"] = df.loc[rows, "description"].str.replace(activity_string, "")
+
+    df.loc[df["activity_type"].isna(), "activity_type"] = "Viewed"
+    
+    description_lines = df["description"].str.split("\n")
+    df[""] = description_lines.str[0].str.strip()
+
+    del df["description"]
+
+    return df
+