@@ -360,6 +360,8 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
360360< span class ="sd "> dict containing the oidc information</ span >
361361< span class ="sd "> """</ span >
362362 < span class ="n "> oidc</ span > < span class ="o "> =</ span > < span class ="p "> {}</ span >
363+ < span class ="k "> if</ span > < span class ="ow "> not</ span > < span class ="n "> hostname</ span > < span class ="p "> :</ span >
364+ < span class ="k "> return</ span > < span class ="n "> oidc</ span >
363365 < span class ="k "> try</ span > < span class ="p "> :</ span >
364366 < span class ="n "> response</ span > < span class ="o "> =</ span > < span class ="n "> requests</ span > < span class ="o "> .</ span > < span class ="n "> get</ span > < span class ="p "> (</ span > < span class ="sa "> f</ span > < span class ="s2 "> "https://</ span > < span class ="si "> {</ span > < span class ="n "> hostname</ span > < span class ="si "> }</ span > < span class ="s2 "> /wts/external_oidc/"</ span > < span class ="p "> )</ span >
365367 < span class ="n "> response</ span > < span class ="o "> .</ span > < span class ="n "> raise_for_status</ span > < span class ="p "> ()</ span >
@@ -721,7 +723,6 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
721723 < 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 >
722724 < span class ="n "> resolved_drs_prefix_cache</ span > < span class ="p "> :</ span > < span class ="nb "> dict</ span > < span class ="p "> ,</ span >
723725 < 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 >
725726 < 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 >
726727< span class ="p "> )</ span > < span class ="o "> -></ span > < span class ="kc "> None</ span > < span class ="p "> :</ span >
727728< span class ="w "> </ span > < span class ="sd "> """Given a list of object_ids go through list and resolve + cache any unknown hosts</ span >
@@ -735,8 +736,6 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
735736 < 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 >
736737 < 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 >
737738 < 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 >
739- < span class ="n "> entry</ span > < span class ="o "> .</ span > < span class ="n "> hostname</ span > < span class ="o "> =</ span > < span class ="n "> endpoint</ span >
740739 < 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 >
741740 < span class ="c1 "> # if resolution fails the entry hostname will still be None</ span >
742741 < span class ="n "> entry</ span > < span class ="o "> .</ span > < span class ="n "> hostname</ span > < span class ="p "> ,</ span > < span class ="n "> nid</ span > < span class ="p "> ,</ span > < span class ="n "> drs_type</ span > < span class ="o "> =</ span > < span class ="n "> resolve_drs_hostname_from_id</ span > < span class ="p "> (</ span >
@@ -814,9 +813,6 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
814813
815814< span class ="sd "> Returns:</ span >
816815< span class ="sd "> The authz object from the user endpoint</ span >
817- < span class ="sd "> """</ span >
818- < span class ="w "> </ span > < span class ="sd "> """</ span >
819-
820816< span class ="sd "> """</ span >
821817 < span class ="n "> headers</ span > < span class ="o "> =</ span > < span class ="p "> {</ span >
822818 < span class ="s2 "> "Content-Type"</ span > < span class ="p "> :</ span > < span class ="s2 "> "application/json"</ span > < span class ="p "> ,</ span >
@@ -844,7 +840,7 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
844840< span class ="sd "> authz (str): dictionary of authz stringts</ span >
845841< span class ="sd "> """</ span >
846842 < span class ="nb "> print</ span > < span class ="p "> (</ span >
847- < span class ="sa " > f </ span > < span class =" s2 "> "───────────────────────────────────────────────────────────────────────────────────────────────────────"</ span >
843+ < span class ="s2 "> "───────────────────────────────────────────────────────────────────────────────────────────────────────"</ span >
848844 < span class ="p "> )</ span >
849845 < span class ="nb "> print</ span > < span class ="p "> (</ span > < span class ="sa "> f</ span > < span class ="s2 "> "Access for </ span > < span class ="si "> {</ span > < span class ="n "> hostname</ span > < span class ="si "> }</ span > < span class ="s2 "> :"</ span > < span class ="p "> )</ span >
850846 < span class ="k "> if</ span > < span class ="n "> authz</ span > < span class ="ow "> is</ span > < span class ="ow "> not</ span > < span class ="kc "> None</ span > < span class ="ow "> and</ span > < span class ="nb "> len</ span > < span class ="p "> (</ span > < span class ="n "> authz</ span > < span class ="p "> )</ span > < span class ="o "> ></ span > < span class ="mi "> 0</ span > < span class ="p "> :</ span >
@@ -856,6 +852,18 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
856852 < span class ="nb "> print</ span > < span class ="p "> (</ span > < span class ="s2 "> " No access"</ span > < span class ="p "> )</ span >
857853
858854
855+ < span class ="k "> def</ span > < span class ="w "> </ span > < span class ="nf "> get_hostname_from_endpoint</ span > < span class ="p "> (</ span > < span class ="n "> endpoint</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ):</ span >
856+ < span class ="w "> </ span > < span class ="sd "> """</ span >
857+ < span class ="sd "> Get hostname from an Gen3Auth endpoint value</ span >
858+ < span class ="sd "> Args:</ span >
859+ < span class ="sd "> endpoint (str): endpoint value form Gen3Auth</ span >
860+ < span class ="sd "> """</ span >
861+ < span class ="k "> if</ span > < span class ="ow "> not</ span > < span class ="n "> endpoint</ span > < span class ="p "> :</ span >
862+ < span class ="k "> return</ span > < span class ="kc "> None</ span >
863+ < span class ="n "> urlparts</ span > < span class ="o "> =</ span > < span class ="n "> urlparse</ span > < span class ="p "> (</ span > < span class ="n "> endpoint</ span > < span class ="p "> )</ span >
864+ < span class ="k "> return</ span > < span class ="n "> urlparts</ span > < span class ="o "> .</ span > < span class ="n "> hostname</ span >
865+
866+
859867< div class ="viewcode-block " id ="DownloadManager ">
860868< a class ="viewcode-back " href ="../../../../tools/drs_pull.html#gen3.tools.download.drs_download.DownloadManager "> [docs]</ a >
861869< span class ="k "> class</ span > < span class ="w "> </ span > < span class ="nc "> DownloadManager</ span > < span class ="p "> :</ span >
@@ -870,7 +878,6 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
870878 < span class ="n "> auth</ span > < span class ="p "> :</ span > < span class ="n "> Gen3Auth</ span > < span class ="p "> ,</ span >
871879 < 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 >
872880 < 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 >
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 >
874881 < 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 >
875882 < span class ="p "> ):</ span >
876883< span class ="w "> </ span > < span class ="sd "> """</ span >
@@ -884,8 +891,9 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
884891< span class ="sd "> download_list (List[Downloadable]): list of objects to download</ span >
885892< span class ="sd "> """</ span >
886893
887- < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> hostname</ span > < span class ="o "> =</ span > < span class ="n "> hostname</ span >
888- < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> endpoint</ span > < span class ="o "> =</ span > < span class ="n "> endpoint</ span >
894+ < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> hostname</ span > < span class ="o "> =</ span > < span class ="p "> (</ span >
895+ < span class ="n "> hostname</ span > < span class ="k "> if</ span > < span class ="n "> hostname</ span > < span class ="k "> else</ span > < span class ="n "> get_hostname_from_endpoint</ span > < span class ="p "> (</ span > < span class ="n "> auth</ span > < span class ="o "> .</ span > < span class ="n "> endpoint</ span > < span class ="p "> )</ span >
896+ < span class ="p "> )</ span >
889897 < 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 >
890898 < 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 >
891899 < 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 >
@@ -915,8 +923,9 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
915923 < span class ="n "> resolve_objects_drs_hostname</ span > < span class ="p "> (</ span >
916924 < span class ="n "> object_list</ span > < span class ="p "> ,</ span >
917925 < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> resolved_compact_drs</ span > < span class ="p "> ,</ span >
918- < 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 >
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 >
926+ < 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 >
927+ < span class ="k "> if</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> hostname</ span >
928+ < span class ="k "> else</ span > < span class ="kc "> None</ span > < span class ="p "> ,</ span >
920929 < 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 >
921930 < span class ="p "> )</ span >
922931 < span class ="n "> progress_bar</ span > < span class ="o "> =</ span > < span class ="p "> (</ span >
@@ -1212,7 +1221,6 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
12121221 < span class ="n "> auth</ span > < span class ="o "> =</ span > < span class ="n "> auth</ span > < span class ="p "> ,</ span >
12131222 < span class ="n "> download_list</ span > < span class ="o "> =</ span > < span class ="n "> object_list</ span > < span class ="p "> ,</ span >
12141223 < span class ="n "> show_progress</ span > < span class ="o "> =</ span > < span class ="n "> show_progress</ span > < span class ="p "> ,</ span >
1215- < span class ="n "> endpoint</ span > < span class ="o "> =</ span > < span class ="n "> hostname</ span > < span class ="p "> ,</ span >
12161224 < span class ="p "> )</ span >
12171225
12181226 < 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 >
@@ -1261,7 +1269,6 @@ <h1>Source code for gen3.tools.download.drs_download</h1><div class="highlight">
12611269 < span class ="n "> auth</ span > < span class ="o "> =</ span > < span class ="n "> auth</ span > < span class ="p "> ,</ span >
12621270 < span class ="n "> download_list</ span > < span class ="o "> =</ span > < span class ="n "> object_list</ span > < span class ="p "> ,</ span >
12631271 < span class ="n "> show_progress</ span > < span class ="o "> =</ span > < span class ="n "> show_progress</ span > < span class ="p "> ,</ span >
1264- < span class ="n "> endpoint</ span > < span class ="o "> =</ span > < span class ="n "> hostname</ span > < span class ="p "> ,</ span >
12651272 < span class ="n "> commons_url</ span > < span class ="o "> =</ span > < span class ="n "> commons_url</ span > < span class ="p "> ,</ span >
12661273 < span class ="p "> )</ span >
12671274
0 commit comments