py导出时段报表

py导出时段报表

　　昨天做的任务，一天下来各种杂事，白天也时不时摸鱼，搞到晚上回家继续弄。今早弄些细节基本算完成~~~

　　这个脚本跟之前的最大区别，进去报表导出页面后，条件选择需要先点击下拉框，再选择地市或导出格式；另外一点，选择时间查询（前一天）的时候，先要抹掉默认值（用clear方法）再填入昨天的时间。

　　因为正值上班时间，赶紧发完博，继续搬砖。。

1、报表页面原来默认进去显示这样：

现在需求：开始和结束时间选前一天日期的，地市选择“顺德”，点查看后再导出报表。

# -*- coding: utf-8 -*- 
# @Time : 2022/4/19 8:57
# @Author : ljy 
# @File : period-20220419.py
#

'''
   此脚本利用获取登陆系统的cookie，进去xx市销量页面，导出数据

'''

# coding:utf-8
import os
from requests_toolbelt.multipart.encoder import MultipartEncoder
from selenium.webdriver.chrome.options import Options
from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
import time
import shutil
import datetime
import re
import requests
import json
from tkinter import *
import tkinter as tk
import tkinter.messagebox

from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.keys import Keys
import os
import requests
import sys
import shutil


#########################################################
# 1、发送消息的函数
def send_msg(msg):
    """
    :param msg: 要发送的消息
    :return: 200 or False
    """
    url = 'xxx'
    program = {"msgtype": "text", "text": {"content": msg}, "at": {"atMobiles": ['我的手机号'], "isAtAll": "false"}}
    headers = {'Content-Type': 'application/json'}
    try:
        f = requests.post(url, data=json.dumps(program), headers=headers)
    except Exception as e:
        return False
    return f.status_code


#########################################################
# 2、获取上个工作日时间
def last_workday():
    for i in range(1, 7):
        now_time = datetime.datetime.now()
        yes_time = now_time + datetime.timedelta(days=-i)
        yes_time_week = yes_time.strftime('%Y-%m-%d')
        global last_workdays
        last_workdays = yes_time_week
        return last_workdays


#########################################################
# 3、 初始化浏览器
def browser_page():
    path_get = 'E:\\table'
    print("当前路径: %s" % path_get)
    time.sleep(3)
    options = webdriver.ChromeOptions()
    prefs = {'download.default_directory': path_get,
             #  prefs = {'download.default_directory': 'E:\\table',
             'download.prompt_for_download': False,
             "download.directory_upgrade": False,
             "safebrowsing.enabled": False
             }
    options.add_experimental_option('prefs', prefs)
    try:
        global browser
        browser = webdriver.Chrome(r"chromedriver", chrome_options=options)
    except:
        print("browser faild!")
        send_msg("test: 浏览器初始化错误!")
        browser = webdriver.Chrome(r"chromedriver")

##################################################################
# 4、登录主页
def browser_project(browser):
    # 设置cookies前必须访问一次页面
    browser.get("某系统url")
    try:
        with open("cookies.txt", "r") as fp:
            cookies = json.load(fp)
            for cookie in cookies:
                browser.add_cookie(cookie)
    except:
        print("获取cookie失败")
        send_msg("test: cookie已过期，获取cookie失败")
    # 拿cookie进去主页
    browser.get("某系统url")


##############################################################
# 5、进入主页跳转
def get_bo(browser):
    wait = WebDriverWait(browser, 10, 0.5)
    time.sleep(2)
    # 5.1 点击“某关键字”
    input_user = browser.find_elements_by_xpath(r'//*[@id="bodyer"]/div/div[2]/div/div[2]')

    try:
        input_user[0].click()
    except IndexError:
        send_msg("test：cookie文件失效，请重新执行!")
        tk.Tk().withdraw()
        tk.messagebox.showinfo("提示", "请先运行: 获取cookie脚本")
        browser.quit()
        # exit()

    time.sleep(5)
    browser.switch_to.window(browser.window_handles[1])
    # 5.2 选择“数据报表”
    input_user = browser.find_elements_by_xpath(r'//*[@id="mainNav"]/li[3]/a')

    time.sleep(5)
    input_user[0].click()

　　     ## 新增，该页面有时候很久才加载出来　　　　# 等到出现xxx元素

　　　　 element = WebDriverWait(browser, 20).until(
EC.element_to_be_clickable((By.ID, "ctl00_ctl00_content_ContentPlaceHolder2_uc_report1_divReportContent")))
        time.sleep(2)

    # 5.3 选择“xxx时段报表”
    input_user = browser.find_elements_by_xpath(
        r'//*[@id="ctl00_ctl00_content_ContentPlaceHolder2_uc_report1_divReportContent"]/div[6]/ul/li[2]/p/a')
    time.sleep(2)
    input_user[0].click()
    browser.switch_to.window(browser.window_handles[2])

    ##################################################################

# 6、xxxx报表页面，导出数据

    # 6.1 选择时间:
    # 报错
    # browser.find_element_by_id(r'//*[@id="reportViewer_ctl04_ctl03_txtValue"]').clear()
    # browser.find_elements_by_xpath(r'//*[@id="reportViewer_ctl04_ctl03_txtValue"]').clear()

    ## 6.1.1 左边日期框
    last_workdays = last_workday()
    print(last_workdays)
    time.sleep(20)

    browser.find_element_by_id("reportViewer_ctl04_ctl03_txtValue").clear()
    browser.find_element_by_id('reportViewer_ctl04_ctl03_txtValue').send_keys(last_workdays)

    ## 6.1.2 右边日期框
    time.sleep(2)
    browser.find_element_by_id("reportViewer_ctl04_ctl05_txtValue").clear()
    browser.find_element_by_id('reportViewer_ctl04_ctl05_txtValue').send_keys(last_workdays)

    # 6.2 选择地市：  ---- 如果选择所有地市，把6.2全部注释掉，貌似需求后来改成所有地市
    ## 6.2.1 先点下拉
    DownButton = browser.find_elements_by_xpath(r'//*[@id="reportViewer_ctl04_ctl07_ddDropDownButton"]')
    time.sleep(2)
    DownButton[0].click()

    ## 6.2.2 点“全选”
    QuanXuan = browser.find_elements_by_xpath(r'//*[@id="reportViewer_ctl04_ctl07_divDropDown_ctl00"]')
    time.sleep(2)
    QuanXuan[0].click()

    ## 6.2.3 点“顺德”
    ShunDe = browser.find_elements_by_xpath(r'//*[@id="reportViewer_ctl04_ctl07_divDropDown_ctl23"]')
    time.sleep(2)
    ShunDe[0].click()

    # 6.3 点击“查看报表”
    time.sleep(5)
    input_user = browser.find_elements_by_xpath(r'//*[@id="reportViewer_ctl04_ctl00"]')
    input_user[0].click()

    time.sleep(10)
    ## 6.3.1 先点下拉
    buttonLink = browser.find_element_by_xpath(r'//*[@id="reportViewer_ctl06_ctl04_ctl00_ButtonLink"]')
    time.sleep(2)
    buttonLink.click()

    ## 6.3.2 导出为excel
    menu = browser.find_elements_by_xpath(r'//*[@id="reportViewer_ctl06_ctl04_ctl00_Menu"]/div[5]/a')
    time.sleep(2)
    menu[0].click()

    # 6.4 报表改名
    path_get = os.getcwd()
    filepath2 = change_name(path_get, "period tb")

    browser.quit()
    print("exit")

###########################
# 7、 修改导出报表的名字
def change_name(url, add):
    url = 'E:\\table'
    print("改名，当前路径: %s" % url)
    time.sleep(3)
    print(r"RF_0202%s" % last_workdays)
    patt = re.compile(r"RF_0202")
    lists = os.listdir(url)
    print(lists)
    while True:
        lists.sort(key=lambda fn: os.path.getmtime(url + '\\' + fn))
        filepath = os.path.join(url, lists[-1])
        try:
            context = patt.search(filepath).group()
            if context:
                break
        except:
            try:
                lists.pop()
            except:
                send_msg("test: %s -- 报表导出失败，没有文件！" % add)
    # ok
    filepath2 = url + "\\" + add + "-" + last_workdays.replace("-", "") + ".xls"

    print(filepath2)
    time.sleep(5)

    shutil.move(filepath, filepath2)
    return filepath2
#
#
# ###################### 导入报表 
  后续等公司开发弄完再做。。。

#################################################################
def start_project_bo():
  #  last_workday()
    browser_page()
    browser_project(browser)
    get_bo(browser)

if __name__ == '__main__':
    start_project_bo()

　　大家主要看 “6、XX时段报表页面，导出数据”，前面都是各种页面跳转到报表页面。

　　这个模板，以后可以继续参考用。

2、成果图：

 运行脚本，自动选好查询条件，查看报表后，最后导出：先下拉，再选 Excel 格式下载

 

　　公司前端后来发了这个给我看，鼠标定位的：https://m.yisu.com/zixun/174401.html

　　说爬虫用这种方法实现，比较简单，毕竟xpath定位元素容易报错。