Skip to content

Commit b1ca88b

Browse files
Merge pull request #248 from uc-cdis/chore/allow-commons-url-pull-object
Chore/allow commons-url pull object
2 parents 6b905a1 + 3093278 commit b1ca88b

File tree

13 files changed

+79
-34
lines changed

13 files changed

+79
-34
lines changed
337 Bytes
Binary file not shown.
1.61 KB
Binary file not shown.
0 Bytes
Binary file not shown.
5 Bytes
Binary file not shown.

docs/_build/html/_modules/gen3/tools/download/drs_download.html

Lines changed: 19 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -718,7 +718,11 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
718718

719719

720720
<span class="k">def</span><span class="w"> </span><span class="nf">resolve_objects_drs_hostname</span><span class="p">(</span>
721-
<span class="n">object_ids</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">Downloadable</span><span class="p">],</span> <span class="n">resolved_drs_prefix_cache</span><span class="p">:</span> <span class="nb">dict</span><span class="p">,</span> <span class="n">mds_url</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span> <span class="n">endpoint</span><span class="p">:</span> <span class="nb">str</span>
721+
<span class="n">object_ids</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">Downloadable</span><span class="p">],</span>
722+
<span class="n">resolved_drs_prefix_cache</span><span class="p">:</span> <span class="nb">dict</span><span class="p">,</span>
723+
<span class="n">mds_url</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
724+
<span class="n">endpoint</span><span class="p">:</span> <span class="nb">str</span><span class="p">,</span>
725+
<span class="n">commons_url</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
722726
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
723727
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Given a list of object_ids go through list and resolve + cache any unknown hosts</span>
724728

@@ -729,7 +733,9 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
729733
<span class="sd"> hostname (str): Hostname to main Gen3 environment</span>
730734
<span class="sd"> &quot;&quot;&quot;</span>
731735
<span class="k">for</span> <span class="n">entry</span> <span class="ow">in</span> <span class="n">object_ids</span><span class="p">:</span>
732-
<span class="k">if</span> <span class="n">endpoint</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span> <span class="ow">and</span> <span class="n">entry</span><span class="o">.</span><span class="n">hostname</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
736+
<span class="k">if</span> <span class="n">commons_url</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
737+
<span class="n">entry</span><span class="o">.</span><span class="n">hostname</span> <span class="o">=</span> <span class="n">commons_url</span>
738+
<span class="k">elif</span> <span class="n">endpoint</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span> <span class="ow">and</span> <span class="n">entry</span><span class="o">.</span><span class="n">hostname</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
733739
<span class="n">entry</span><span class="o">.</span><span class="n">hostname</span> <span class="o">=</span> <span class="n">endpoint</span>
734740
<span class="k">if</span> <span class="n">entry</span><span class="o">.</span><span class="n">hostname</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
735741
<span class="c1"># if resolution fails the entry hostname will still be None</span>
@@ -864,7 +870,8 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
864870
<span class="n">auth</span><span class="p">:</span> <span class="n">Gen3Auth</span><span class="p">,</span>
865871
<span class="n">download_list</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">Downloadable</span><span class="p">],</span>
866872
<span class="n">show_progress</span><span class="p">:</span> <span class="nb">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
867-
<span class="n">endpoint</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="kc">None</span>
873+
<span class="n">endpoint</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
874+
<span class="n">commons_url</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span>
868875
<span class="p">):</span>
869876
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
870877
<span class="sd"> Initialize the DownloadManager so that is ready to start downloading.</span>
@@ -879,6 +886,7 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
879886

880887
<span class="bp">self</span><span class="o">.</span><span class="n">hostname</span> <span class="o">=</span> <span class="n">hostname</span>
881888
<span class="bp">self</span><span class="o">.</span><span class="n">endpoint</span> <span class="o">=</span> <span class="n">endpoint</span>
889+
<span class="bp">self</span><span class="o">.</span><span class="n">commons_url</span> <span class="o">=</span> <span class="n">commons_url</span>
882890
<span class="bp">self</span><span class="o">.</span><span class="n">access_token</span> <span class="o">=</span> <span class="n">auth</span><span class="o">.</span><span class="n">get_access_token</span><span class="p">()</span>
883891
<span class="bp">self</span><span class="o">.</span><span class="n">metadata</span> <span class="o">=</span> <span class="n">Gen3Metadata</span><span class="p">(</span><span class="n">auth</span><span class="p">)</span>
884892
<span class="bp">self</span><span class="o">.</span><span class="n">wts_endpoints</span> <span class="o">=</span> <span class="n">wts_external_oidc</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
@@ -908,7 +916,8 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
908916
<span class="n">object_list</span><span class="p">,</span>
909917
<span class="bp">self</span><span class="o">.</span><span class="n">resolved_compact_drs</span><span class="p">,</span>
910918
<span class="n">mds_url</span><span class="o">=</span><span class="sa">f</span><span class="s2">&quot;http://</span><span class="si">{</span><span class="bp">self</span><span class="o">.</span><span class="n">hostname</span><span class="si">}</span><span class="s2">/mds/aggregate/info&quot;</span><span class="p">,</span>
911-
<span class="n">endpoint</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">endpoint</span>
919+
<span class="n">endpoint</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">endpoint</span><span class="p">,</span>
920+
<span class="n">commons_url</span><span class="o">=</span><span class="bp">self</span><span class="o">.</span><span class="n">commons_url</span><span class="p">,</span>
912921
<span class="p">)</span>
913922
<span class="n">progress_bar</span> <span class="o">=</span> <span class="p">(</span>
914923
<span class="n">tqdm</span><span class="p">(</span><span class="n">desc</span><span class="o">=</span><span class="sa">f</span><span class="s2">&quot;Resolving objects&quot;</span><span class="p">,</span> <span class="n">total</span><span class="o">=</span><span class="nb">len</span><span class="p">(</span><span class="n">object_list</span><span class="p">))</span>
@@ -1203,7 +1212,7 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
12031212
<span class="n">auth</span><span class="o">=</span><span class="n">auth</span><span class="p">,</span>
12041213
<span class="n">download_list</span><span class="o">=</span><span class="n">object_list</span><span class="p">,</span>
12051214
<span class="n">show_progress</span><span class="o">=</span><span class="n">show_progress</span><span class="p">,</span>
1206-
<span class="n">endpoint</span><span class="o">=</span><span class="n">hostname</span>
1215+
<span class="n">endpoint</span><span class="o">=</span><span class="n">hostname</span><span class="p">,</span>
12071216
<span class="p">)</span>
12081217

12091218
<span class="n">out_dir_path</span> <span class="o">=</span> <span class="n">ensure_dirpath_exists</span><span class="p">(</span><span class="n">Path</span><span class="p">(</span><span class="n">output_dir</span><span class="p">))</span>
@@ -1224,6 +1233,7 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
12241233
<span class="n">show_progress</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
12251234
<span class="n">unpack_packages</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
12261235
<span class="n">delete_unpacked_packages</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
1236+
<span class="n">commons_url</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span>
12271237
<span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Dict</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="n">Any</span><span class="p">]]:</span>
12281238
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
12291239
<span class="sd"> A convenience function used to download a single DRS object.</span>
@@ -1251,7 +1261,8 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
12511261
<span class="n">auth</span><span class="o">=</span><span class="n">auth</span><span class="p">,</span>
12521262
<span class="n">download_list</span><span class="o">=</span><span class="n">object_list</span><span class="p">,</span>
12531263
<span class="n">show_progress</span><span class="o">=</span><span class="n">show_progress</span><span class="p">,</span>
1254-
<span class="n">endpoint</span><span class="o">=</span><span class="n">hostname</span>
1264+
<span class="n">endpoint</span><span class="o">=</span><span class="n">hostname</span><span class="p">,</span>
1265+
<span class="n">commons_url</span><span class="o">=</span><span class="n">commons_url</span><span class="p">,</span>
12551266
<span class="p">)</span>
12561267

12571268
<span class="n">out_dir_path</span> <span class="o">=</span> <span class="n">ensure_dirpath_exists</span><span class="p">(</span><span class="n">Path</span><span class="p">(</span><span class="n">output_dir</span><span class="p">))</span>
@@ -1447,6 +1458,7 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
14471458
<span class="n">show_progress</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
14481459
<span class="n">unpack_packages</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span>
14491460
<span class="n">delete_unpacked_packages</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
1461+
<span class="n">commons_url</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span>
14501462
<span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
14511463
<span class="w"> </span><span class="sd">&quot;&quot;&quot;</span>
14521464
<span class="sd"> A convenience function used to download a single DRS object.</span>
@@ -1469,6 +1481,7 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
14691481
<span class="n">show_progress</span><span class="p">,</span>
14701482
<span class="n">unpack_packages</span><span class="p">,</span>
14711483
<span class="n">delete_unpacked_packages</span><span class="p">,</span>
1484+
<span class="n">commons_url</span><span class="p">,</span>
14721485
<span class="p">)</span>
14731486

14741487

docs/_build/html/searchindex.js

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

docs/_build/html/tools/drs_pull.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454
</div></blockquote>
5555
<dl class="py class">
5656
<dt class="sig sig-object py" id="gen3.tools.download.drs_download.DownloadManager">
57-
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">gen3.tools.download.drs_download.</span></span><span class="sig-name descname"><span class="pre">DownloadManager</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">hostname</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">auth</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference internal" href="../auth.html#gen3.auth.Gen3Auth" title="gen3.auth.Gen3Auth"><span class="pre">Gen3Auth</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">download_list</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">List</span><span class="p"><span class="pre">[</span></span><a class="reference internal" href="#gen3.tools.download.drs_download.Downloadable" title="gen3.tools.download.drs_download.Downloadable"><span class="pre">Downloadable</span></a><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">show_progress</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">endpoint</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span><span class="w"> </span><span class="p"><span class="pre">|</span></span><span class="w"> </span><span class="pre">None</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/gen3/tools/download/drs_download.html#DownloadManager"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#gen3.tools.download.drs_download.DownloadManager" title="Link to this definition"></a></dt>
57+
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">gen3.tools.download.drs_download.</span></span><span class="sig-name descname"><span class="pre">DownloadManager</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">hostname</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">auth</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference internal" href="../auth.html#gen3.auth.Gen3Auth" title="gen3.auth.Gen3Auth"><span class="pre">Gen3Auth</span></a></span></em>, <em class="sig-param"><span class="n"><span class="pre">download_list</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">List</span><span class="p"><span class="pre">[</span></span><a class="reference internal" href="#gen3.tools.download.drs_download.Downloadable" title="gen3.tools.download.drs_download.Downloadable"><span class="pre">Downloadable</span></a><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">show_progress</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">endpoint</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span><span class="w"> </span><span class="p"><span class="pre">|</span></span><span class="w"> </span><span class="pre">None</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">commons_url</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span><span class="w"> </span><span class="p"><span class="pre">|</span></span><span class="w"> </span><span class="pre">None</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference internal" href="../_modules/gen3/tools/download/drs_download.html#DownloadManager"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#gen3.tools.download.drs_download.DownloadManager" title="Link to this definition"></a></dt>
5858
<dd><p>Class to assist in downloading a list of Downloadable object which at a minimum is a json manifest
5959
of DRS object ids. The methods of interest are download and user_access.</p>
6060
<dl class="py method">

0 commit comments

Comments
 (0)