UniqueSoft/WGDashboard
2
0
Fork 0
mirror of https://github.com/donaldzou/WGDashboard synced 2025-02-26 05:58:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #619 from akhepcat/v4.2-dev

Browse Source 
Update wgd.sh - install pypi mirror can auto select by rtt after delay
This commit is contained in:
Donald Zou 2025-02-18 02:46:19 +08:00 committed by GitHub
commit 8e5cf14ebc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 37 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
src/wgd.sh
View File

@ -15,6 +15,8 @@ heavy_checkmark=$(printf "\xE2\x9C\x85")
heavy_crossmark=$(printf "\xE2\x9D\x8C")
install=$(printf "\xF0\x9F\x92\xBF")
msleep=15
PID_FILE=./gunicorn.pid
environment=$(if [[ $ENVIRONMENT ]]; then echo $ENVIRONMENT; else echo 'develop'; fi)
if [[ $CONFIGURATION_PATH ]]; then
@ -256,16 +258,16 @@ _determinePypiMirror(){
printf "[WGDashboard] %s Pinging list of recommended Python Package Index mirror\n" "$install"
urls=(
"https://pypi.org/simple/"
"https://pypi.tuna.tsinghua.edu.cn/simple/"
"https://pypi.mirrors.ustc.edu.cn/simple/"
"https://mirrors.aliyun.com/pypi/simple/"
"https://pypi.douban.com/simple/"
)
# Function to extract hostname and ping it
index=1
printf " ---------------------------------------------------------\n"
for url in "${urls[@]}"; do
"https://pypi.tuna.tsinghua.edu.cn/simple/"
"https://pypi.mirrors.ustc.edu.cn/simple/"
"https://mirrors.aliyun.com/pypi/simple/"
"https://pypi.douban.com/simple/"
)
# Function to extract hostname and ping it
index=1
printf " ---------------------------------------------------------\n"
for url in "${urls[@]}"; do
# Extract the hostname from the URL
hostname=$(echo "$url" | awk -F/ '{print $3}')
# Ping the hostname once and extract the RTT
@ -275,24 +277,38 @@ _determinePypiMirror(){
rtt="9999"
printf " [%i] [FAILED] %s\n" "$index" "$url"
else
rtt=${rtt//.*/}
printf " [%i] %sms %s\n" "$index" "$rtt" "$url"
fi
rtthost[$index]=$rtt
index=$((index+1))
done
for i in "${!rtthost[@]}"; do
[[ -z ${rtthost[i]} ]] && continue # Skip unset or empty values
if [[ -z $min_val || ${rtthost[i]} -lt $min_val ]]; then
min_val=${rtthost[i]}
min_idx=$i
fi
done
min_idx=$((min_idx - 1))
printf "\n"
printf " Which mirror you would like to use (Hit enter to use default): "
read -r choice
printf " Which mirror you would like to use (Hit enter or wait ${msleep} seconds to use default: ${urls[$min_idx]}): "
read -t ${msleep} -r choice
printf "\n"
if [[ -z "$choice" ]]; then
choice=${min_dix}
fi
if [[ "$choice" =~ ^[0-9]+$ ]] && (( choice >= 1 && choice <= ${#urls[@]} )); then
selected_url="${urls[choice-1]}"
printf "[WGDashboard] %s Will download Python packages from %s\n" "$heavy_checkmark" "$selected_url"
else
selected_url="${urls[0]}"
printf "[WGDashboard] %s Will download Python packages from %s\n" "$heavy_checkmark" "${urls[0]}"
fi
selected_url="${urls[choice-1]}"
printf "[WGDashboard] %s Will download Python packages from %s\n" "$heavy_checkmark" "$selected_url"
else
selected_url="${urls[0]}"
printf "[WGDashboard] %s Will download Python packages from %s\n" "$heavy_checkmark" "${urls[0]}"
fi
}
install_wgd(){