excel.py

from openpyxl import Workbook
import sys

import download

if __name__ == '__main__':
    cookie_data=download.user_data()

    if len(sys.argv) > 2:
        url = sys.argv[1]
        cookie_data.set_cookies(sys.argv[2])
    elif len(sys.argv) > 1:
        url = sys.argv[1]
    else:
        url = input("url: ")
    title, tabs, opendoc_params = download.initial_fetch(url,cookie_data=cookie_data)
    print("文档名称: %s" % title)
    wb = Workbook()
    for tab in tabs:
        tab_id = tab["id"]
        name = tab["name"]
        print("正在下载: %s" % name)
        sheet_content, max_col = download.read_sheet(url, tab_id, opendoc_params,cookie_data)
        row = []
        ws = wb.create_sheet(name)
        for k, v in sheet_content.items():
            if (int(k) % max_col == 0 and k != '0'):
                ws.append(row)
                row=[]
            if '2' in v:
                row.append(v['2'][1])
            else:
                row.append("")
    empty_ws = wb["Sheet"]
    wb.remove(empty_ws)
    wb.save('%s.xlsx' % title)
    print("下载完成，已保存与根目录")