Close #30, update stand-alone script

Author: rjzupkoii

talkmap.py

@@ -1,47 +1,45 @@
-
-
-# # Leaflet cluster map of talk locations
-#
-# (c) 2016-2017 R. Stuart Geiger, released under the MIT license
+# Leaflet cluster map of talk locations
 #
-# Run this from the _talks/ directory, which contains .md files of all your talks. 
-# This scrapes the location YAML field from each .md file, geolocates it with
-# geopy/Nominatim, and uses the getorg library to output data, HTML,
-# and Javascript for a standalone cluster map.
-#
-# Requires: glob, getorg, geopy
-
+# Run this from the _talks/ directory, which contains .md files of all your
+# talks. This scrapes the location YAML field from each .md file, geolocates it
+# with geopy/Nominatim, and uses the getorg library to output data, HTML, and
+# Javascript for a standalone cluster map. This is functionally the same as the
+# #talkmap Jupyter notebook.
+import frontmatter
 import glob
 import getorg
 from geopy import Nominatim
 
-g = glob.glob("*.md")
-
+# Collect the Markdown files
+g = glob.glob("_talks/*.md")
 
-geocoder = Nominatim()
+# Prepare to geolocate
+geocoder = Nominatim(user_agent="academicpages.github.io")
 location_dict = {}
 location = ""
 permalink = ""
 title = ""
 
-
+# Perform geolocation
 for file in g:
-    with open(file, 'r') as f:
-        lines = f.read()
-        if lines.find('location: "') > 1:
-            loc_start = lines.find('location: "') + 11
-            lines_trim = lines[loc_start:]
-            loc_end = lines_trim.find('"')
-            location = lines_trim[:loc_end]
-                            
-           
-        location_dict[location] = geocoder.geocode(location)
-        print(location, "\n", location_dict[location])
-
-
-m = getorg.orgmap.create_map_obj()
-getorg.orgmap.output_html_cluster_map(location_dict, folder_name="../talkmap", hashed_usernames=False)
+    # Read the file
+    data = frontmatter.load(file)
+    data = data.to_dict()
 
+    # Press on if the location is not present
+    if 'location' not in data:
+        continue
 
+    # Prepare the description
+    title = data['title'].strip()
+    venue = data['venue'].strip()
+    location = data['location'].strip()
+    description = f"{title}<br />{venue}; {location}"
 
+    # Geocode the location and report the status
+    location_dict[description] = geocoder.geocode(location)
+    print(description, location_dict[description])
 
+# Save the map
+m = getorg.orgmap.create_map_obj()
+getorg.orgmap.output_html_cluster_map(location_dict, folder_name="talkmap", hashed_usernames=False)