#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time  : 2019-12-25 11:33
# @Author : Anthony
# @Email  : ianghont7@163.com
# @File  : get_video_audio_file.py

import xlrd as xl
import requests



# 創(chuàng)建文件夾
def mkdir_floder(path):
  import os
  isExists = os.path.exists(path)
  if not isExists:
    os.makedirs(path)
    # print(path + ' 創(chuàng)建成功')
    return True
  else:
    # 如果目錄存在則不創(chuàng)建，并提示目錄已存在
    # print(path + ' 目錄已存在')
    return False



def request_floder(floder01,floder02,filename,url,types):
  # 拼接全路徑
  all_path = basedirpath+floder01+'/'+floder02+'/'
  res = requests.get(url.strip())
  music = res.content
  with open(all_path+filename+'.'+types, 'ab') as file: # 保存到本地的文件名
    file.write(music)
    file.flush()



def get_xls(filename):
  # 打開文件
  xls_file=xl.open_workbook(filename)
  # 獲取第一個(gè)sheet內(nèi)容
  xls_sheet=xls_file.sheets()[0]
  # 總行數(shù)
  line_nums = xls_sheet.nrows
  for i in range(0,line_nums):
    # 獲取每一行的內(nèi)容
    row_value = xls_sheet.row_values(i)
    # 獲取第一級(jí)目錄名稱
    folder1 = row_value[0]
    # 獲取第二級(jí)目錄名稱
    folder2 = row_value[1]
    # 獲取文件名
    file_name = row_value[2]
    # 獲取url內(nèi)容
    url = row_value[3]
    # 創(chuàng)建目錄
    mkdir_floder(basedirpath+folder1+'/'+folder2)
    # 轉(zhuǎn)換url的類型為dict
    url = eval(url)

    # 獲取視頻內(nèi)容
    if "audio" in url:
      if url["audioF"].strip() != "":
        audio_url = url["audioF"]
        audio_url_end = audio_url.split('.')[-1]
        print(audio_url)
        request_floder(folder1, folder2, file_name, audio_url, audio_url_end)

      elif url["audio"].strip() != "":
        audio_url = url["audio"]
        audio_url_end = audio_url.split('.')[-1]
        print(audio_url)
        request_floder(folder1, folder2, file_name, audio_url, audio_url_end)

    # 獲取音頻內(nèi)容
    elif "video" in url:
      if url["video"].strip() != "":
        video_file = url["video"]
        video_file_end = video_file.split('.')[-1]
        print(video_file)
        request_floder(folder1, folder2, file_name, video_file, video_file_end)

if __name__ == "__main__":
  # 文件存放路徑
  basedirpath = "/Users/ianthony/Desktop/Devops/"
  # 讀取的xlsx文件
  get_xls("001.xlsx")