#!/bin/bash


# 檢查apt安裝的包是否成功
function apt_check(){
    dpkg -l | grep -i $1 > /dev/null
    if [ $? -eq 0 ];then
        echo "======$1 已安裝！======="
    else
        echo "======$1 安裝失敗！======="
    fi    
}

# 檢查pip安裝的包是否成功
function pip_check(){
    pip3 list | grep -i $1 > /dev/null
    if [ $? -eq 0 ];then
        echo 
 
"======$1 已安裝！======="
    else
        echo "======$1 安裝失敗！======="
    fi    
}

# apt安裝的方法
function apt_get(){
    sudo apt-get -y install $1
}
# pip安裝
function pip3_install(){
    sudo pip3 install $1
}

#更新源
# cd /etc/apt && sudo rm -rf sources.list
# cd -
# sudo cp sources.list /etc/apt && sudo apt-get update
 

# 替換源
#cd /etc/apt 
old_list=$(head -1 /etc/apt/sources.list)
new_list=$(head -1 sources.list)

if [ ${old_list} == ${new_list} ];then
    echo "源已更新"
else
    cd /etc/apt && sudo rm -rf sources.list
    cd -
    sudo cp sources.list /etc/apt && sudo apt-get update
fi

# 切回初始目錄
cd -


# 安裝pip3
 

apt_get python3-pip

# 安裝pyautogui和依賴包
pip3_install pillow
pip3_install python3-xlib
apt_get scrot
apt_get python3-tk
apt_get python3-dev
pip3_install pyautogui -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

# 安裝dogtail依賴包
apt_get python3-pyatspi
pip3_install dogtail -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com


# 安裝其他依賴包
pip3_install ddt
pip3_install BeautifulReport
pip3_install xlrd
pip3_install PyYAML

# 安裝pytest
# pip3_install pytest
# pip3_install allure-pytest
# pip3_install timeout-decorator
pip3_install opencv-python -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

# 安裝d-feet
apt_get d-feet

# 檢查apt-get命令是否安裝成功
for i in python3-pip scrot python3-tk python3-dev python3-pyatspi d-feet
do
    apt_check $i
done

# 檢查pip3命令是否安裝成功
for i in pillow python3-xlib pyautogui dogtail ddt BeautifulReport xlrd PyYAML pytest allure-pytest timeout-decorator opencv-python
do
    pip_check $i
done

sleep 15