make update script os agnostic

michaelbabyn · michaelbabyn · commit 3918bf738a49 · 2018-08-31T16:35:50.000-04:00
diff --git a/scripts/update github.ipynb b/scripts/update github.ipynb
@@ -2,20 +2,33 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 35,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "1816 1783\n"
+     ]
+    }
+   ],
    "source": [
+    "import os\n",
     "import yaml\n",
     "\n",
     "#all_posts.txt is generated with the following command from the repo's root directory:\n",
     "# $ find `pwd`/_posts -type f  > all_posts.txt\n",
     "\n",
+    "cwd = os.getcwd()\n",
+    "docs_path = os.path.split(cwd)[0]\n",
+    "\n",
     "file_names = []\n",
-    "with open('../all_posts.txt') as f:\n",
+    "with open(os.path.join(docs_path,'all_posts.txt')) as f:\n",
     "    file_names = [line.split('\\n')[0] for line in f]\n",
     "\n",
-    "html = list(filter(lambda x: '.html' in x,file_names))\n",
+    "no_swp = list(filter(lambda x: '.swp' not in x,file_names))\n",
+    "html = list(filter(lambda x: '.html' in x,no_swp))\n",
     "\n",
     "def get_file(n):\n",
     "    py_file = []\n",
@@ -44,15 +57,15 @@
     "files = [get_file(n) for n in html]\n",
     "names = [(n,get_name(f)) for f,n in files if get_name(f)]\n",
     "\n",
-    "#make sure that most names are unique\n",
+    "#make sure that most names are unique so that we know we have few as 404s as possible\n",
     "c = [n[1] for n in names]\n",
     "print(len(c),len(set(c)))\n",
     "\n",
-    "get_git_data = lambda names: {n.split('name: ')[1]: 'https://github.com/plotly/documentation/tree/source-design-merge/' + '/'.join(loc.split('/home/michael/plotly/repos/documentation/')[1].split('/')[0:-1]) for loc,n in names}\n",
+    "get_git_data = lambda names: {n.split('name: ')[1]: 'https://github.com/plotly/documentation/tree/source-design-merge/' + '/'.join(loc.split(docs_path)[1].split('/')[0:-1]) for loc,n in names}\n",
     "\n",
     "git_data = get_git_data(names)\n",
     "\n",
-    "with open('../_data/git_url.yml', 'w') as outfile:\n",
+    "with open(os.path.join(docs_path,'_data','git_url.yml'), 'w') as outfile:\n",
     "    yaml.dump(git_data, outfile, default_flow_style=False)"
    ]
   }
