modify to use file in data dir

Author: MelissaGraham

MPC/RFL_Solar_System_MPC_Retrieval.ipynb

@@ -105,11 +105,16 @@
    "outputs": [],
    "source": [
     "targets = []\n",
-    "with open('Rubin_First_Look_Solar_System_Discovery_Designations.csv', 'r') as f:\n",
+    "path = '/rubin/cst_repos/tutorial-notebooks-data/data/'\n",
+    "input_file = path + 'Rubin_First_Look_Solar_System_Discovery_Designations.csv'\n",
+    "with open(input_file, 'r') as f:\n",
     "    for line in f.readlines():\n",
     "        desig = line.strip().split(',')\n",
-    "        targets.append(desig[0])\n",
-    "#print(targets[1:])"
+    "        if desig[0][0:1] != 'u':\n",
+    "            targets.append(desig[0])\n",
+    "assert len(targets) == 2098\n",
+    "print('Number of targets: ', len(targets))\n",
+    "print('Example target names: ', targets[0], targets[1], targets[2])"
    ]
   },
   {
@@ -161,11 +166,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "column_names = ['designation', 'absolute_magnitude', 'semimajor_axis', 'eccentricity', 'inclination', 'argument_of_perihelion', 'ascending_node', 'peri_time', 'nobs_total']\n",
+    "column_names = ['designation', 'absolute_magnitude', 'semimajor_axis',\n",
+    "                'eccentricity', 'inclination', 'argument_of_perihelion',\n",
+    "                'ascending_node', 'peri_time', 'nobs_total']\n",
     "\n",
     "discovery_orbits = pd.DataFrame(columns=column_names)\n",
     "\n",
-    "for obj in targets[1:]:\n",
+    "for i, obj in enumerate(targets[1:]):\n",
     "    response = requests.get(\"https://data.minorplanetcenter.net/api/get-orb\", json={\"desig\": obj})\n",
     "    if response.ok:\n",
     "        mpc_orb = response.json()[0]['mpc_orb']\n",
@@ -184,7 +191,19 @@
     "    else:\n",
     "        print(\"Error: \", response.status_code, response.content)\n",
     "    discovery_orbits = pd.concat([discovery_orbits, next_orb_df], ignore_index=True)\n",
-    "discovery_orbits['orbit_type_int'] = discovery_orbits['orbit_type_int'].apply(np.int64)\n",
+    "    if (i == 10) or (i == 100) or (i == 1000):\n",
+    "        print(i, ' out of ', len(targets), ' done')\n",
+    "\n",
+    "discovery_orbits['orbit_type_int'] = discovery_orbits['orbit_type_int'].apply(np.int64)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7ed39608-4486-4b07-95ef-cc217aaa7595",
+   "metadata": {},
+   "outputs": [],
+   "source": [
     "discovery_orbits"
    ]
   },
@@ -232,7 +251,7 @@
     "a, e, inc = discovery_orbits[\"semimajor_axis\"], discovery_orbits[\"eccentricity\"], discovery_orbits[\"inclination\"]\n",
     "inner_orbits = discovery_orbits[a < 7]\n",
     "outer_orbits = discovery_orbits[a >= 7]\n",
-    "    \n",
+    "\n",
     "orbit_type_map = {\n",
     "    0: \"Atira\",\n",
     "    1: \"Aten\",\n",
@@ -354,8 +373,9 @@
     "\n",
     "discovery_observations = pd.DataFrame(columns=column_names_trunc)\n",
     "\n",
-    "for obj in targets[1:]:\n",
-    "    response = requests.get(\"https://data.minorplanetcenter.net/api/get-obs\", json={\"desigs\": [obj], \"output_format\":[\"ADES_DF\"]})\n",
+    "for i, obj in enumerate(targets[1:]):\n",
+    "    response = requests.get(\"https://data.minorplanetcenter.net/api/get-obs\",\n",
+    "                            json={\"desigs\": [obj], \"output_format\":[\"ADES_DF\"]})\n",
     "    if response.ok:\n",
     "        ades_df = pd.DataFrame(response.json()[0]['ADES_DF'])\n",
     "    else:\n",
@@ -382,6 +402,9 @@
     "    discovery_observations['rmsra'] = pd.to_numeric(discovery_observations['rmsra'])\n",
     "    discovery_observations['mag'] = pd.to_numeric(discovery_observations['mag'])\n",
     "    discovery_observations['rmsmag'] = pd.to_numeric(discovery_observations['rmsmag'])\n",
+    "    if (i == 10) or (i == 100) or (i == 1000):\n",
+    "        print(i, ' out of ', len(targets), ' done')\n",
+    "\n",
     "len(discovery_observations)"
    ]
   },