database: Made record id flexible

Author: Somerandomguy10111

opxrd/database/opxrd.py

@@ -6,6 +6,8 @@
 from xrdpattern.pattern import PatternDB
 from holytools.userIO import TrackedInt
 
+# -----------------------------
+
 
 class OpXRD(PatternDB):
     @classmethod
@@ -22,9 +24,9 @@ def load(cls, root_dirpath : str, download : bool = True, *args, **kwargs) -> Pa
         print(f'- Loading patterns from local files')
         return super().load(dirpath=root_dirpath, strict=True)
 
-    @staticmethod
-    def _download_zenodo_opxrd(output_fpath : str):
-        zenodo_url = f'https://zenodo.org/records/14278656'
+    @classmethod
+    def _download_zenodo_opxrd(cls, output_fpath : str):
+        zenodo_url = f'https://zenodo.org/records/{cls.get_record_id()}'
         file_response = requests.get(url=f'{zenodo_url}/files/opxrd.zip?download=1', stream=True)
 
         total_size = int(file_response.headers.get('content-length', 0))
@@ -48,6 +50,9 @@ def _unzip_file(zip_fpath : str, output_dir : str):
             zip_ref.extractall(output_dir)
         return f"Files extracted to {output_dir}"
 
+    @classmethod
+    def get_record_id(cls) -> int:
+        return 14279434
 
 if __name__ == "__main__":
     opxrd = OpXRD.load(root_dirpath='../data/opxrd')

opxrd/usage.ipynb

@@ -49,6 +49,53 @@
     "sys.stdout = stdout"
    ]
   },
+  {
+   "cell_type": "code",
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "- Loading patterns from local files\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\u001B[38;2;0;255;0m100%\u001B[39m \u001B[38;2;0;255;0m(84197 of 84197)\u001B[39m |##################| Elapsed Time: 0:02:16 Time:  0:02:160110\n"
+     ]
+    }
+   ],
+   "source": [
+    "from opxrd import OpXRD\n",
+    "\n",
+    "opxrd = OpXRD.load(root_dirpath='/home/daniel/aimat/data/opXRD/final/')"
+   ],
+   "metadata": {
+    "collapsed": false,
+    "ExecuteTime": {
+     "end_time": "2024-12-05T01:53:25.119145Z",
+     "start_time": "2024-12-05T01:51:03.405227Z"
+    }
+   },
+   "id": "ca2689f7b9f7dbe3",
+   "execution_count": 1
+  },
+  {
+   "cell_type": "code",
+   "outputs": [],
+   "source": [
+    "opxrd.plot_quantity(attr='primary_phase.spaecgroup')\n",
+    "opxrd.plot_quantity(attr='startval')\n",
+    "opxrd.plot_quantity(attr='endval')\n",
+    "opxrd.plot_quantity()"
+   ],
+   "metadata": {
+    "collapsed": false
+   },
+   "id": "ab9c2dc1b83bbc9c"
+  },
   {
    "cell_type": "code",
    "execution_count": 3,