下载google drive文件代码

最新推荐文章于 2023-10-09 13:05:27 发布

BeKnown

最新推荐文章于 2023-10-09 13:05:27 发布

阅读量773

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/m0_37953759/article/details/110383078

版权

Python 专栏收录该内容

8 篇文章 0 订阅

订阅专栏

该代码示例提供了一个Python脚本，用于从Google Drive稳定下载大文件。通过使用`requests`库创建会话并处理确认令牌，可以避免因网络中断导致的重新下载。在下载过程中，还定义了一个进度条类`ProgressBar`来显示下载进度。在命令行中运行此脚本，需要提供Google Drive文件ID和目标保存路径作为参数。

摘要由CSDN通过智能技术生成

通过chrome下载如果中断了，有可能需要重新下载。如果是几个G的文件，重新下载成本很高，使用下面方法下载会稳定一些。

下载google drive的文件

import requests

def download_file_from_google_drive(id, destination):
    def get_confirm_token(response):
        for key, value in response.cookies.items():
            if key.startswith('download_warning'):
                return value

        return None

    def save_response_content(response, destination):
        CHUNK_SIZE = 32768
        
        # content_size = int(response.headers['Content-Length'])
        # progress = ProgressBar(destination, total=1024*1024*6.5,
        #                             unit="KB", chunk_size=CHUNK_SIZE, run_status="正在下载", fin_status="下载完成")

        with open(destination, "wb") as f:
            for chunk in response.iter_content(CHUNK_SIZE):
                if chunk: # filter out keep-alive new chunks
                    f.write(chunk)
                    # import pdb; pdb.set_trace()
        #            progress.refresh(count=len(chunk))

    URL = "https://docs.google.com/uc?export=download"

    session = requests.Session()

    response = session.get(URL, params = { 'id' : id }, stream = True)
    token = get_confirm_token(response)

    if token:
        params = { 'id' : id, 'confirm' : token }
        response = session.get(URL, params = params, stream = True)

    save_response_content(response, destination)    


class ProgressBar(object):

    def __init__(self, title,
                 count=0.0,
                 run_status=None,
                 fin_status=None,
                 total=100.0,
                 unit='', sep='/',
                 chunk_size=1.0):
        super(ProgressBar, self).__init__()
        self.info = "【%s】%s %.2f %s %s %.2f %s"
        self.title = title
        self.total = total
        self.count = count
        self.chunk_size = chunk_size
        self.status = run_status or ""
        self.fin_status = fin_status or " " * len(self.status)
        self.unit = unit
        self.seq = sep

    def __get_info(self):
        # 【名称】状态 进度 单位 分割线 总数 单位
        _info = self.info % (self.title, self.status,
                             self.count/self.chunk_size, self.unit, self.seq, self.total/self.chunk_size, self.unit)
        return _info

    def refresh(self, count=1, status=None):
        self.count += count
        # if status is not None:
        self.status = status or self.status
        end_str = "\r"
        if self.count >= self.total:
            end_str = '\n'
            self.status = status or self.fin_status
        print(self.__get_info(), end=end_str)

if __name__ == "__main__":
    import sys
    if len(sys.argv) is not 3:
        print("Usage: python google_drive.py drive_file_id destination_file_path")
    else:
        # TAKE ID FROM SHAREABLE LINK
        file_id = sys.argv[1]
        # DESTINATION FILE ON YOUR DISK
        destination = sys.argv[2]
        download_file_from_google_drive(file_id, destination)