@@ -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 "> -></ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
723727< span class ="w "> </ span > < span class ="sd "> """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 "> """</ 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 "> """</ 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 "> "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"</ 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 "> "Resolving objects"</ 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 "> -></ 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 "> """</ 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 "> -></ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
14511463< span class ="w "> </ span > < span class ="sd "> """</ 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
0 commit comments