Tkinter和selenium结合实现登录UC后台,最后打包成exe

主要实现的功能:小号模式自动登录UC阿里汇川广告后台,屏蔽账号密码输入
主要用的技术:用Tkinter展示所有的广告账号界面,使用selenium控制谷歌浏览器,打开阿里汇川登录页,登录汇川后台。
第一次写,遇到的坑比较多,三天,搞定。给自己一个棒棒~☺️

import Tkinter as tk  
import os
import sys
import requests
from selenium import webdriver
import base64
from PIL import Image, ImageTk
import math


path = os.path.split(os.path.realpath(__file__))[0] + os.sep

class showCode(tk.Toplevel):
  def __init__(self):
    # super().__init__()
    tk.Toplevel.__init__(self)
    self.title('请输入验证码')
    self.wm_attributes('-topmost',1)
    # 弹窗界面
    self.setup_UI()
  def setup_UI(self):
    # from PIL import Image
    # im = Image.open(path + 'code/code.gif')
    # im.show()
    self._inputCode = tk.StringVar() # 将此处的验证码放置到类中,直接获取该值无法获取
    frame = tk.Frame(self)  #初始化一个框架
    frame.pack()

    im=Image.open(path  + "/code/code.gif")  
    self.photo=ImageTk.PhotoImage(im)  
    cv = tk.Canvas(frame)  
    cv.pack(side='top',fill='both', expand='false',padx=20,pady=20)  
    cv.create_image(0, 0, image=self.photo, anchor='nw')  

    # im=Image.open(path + 'code/code.gif')
    # img=ImageTk.PhotoImage(im)
    # imLabel=tk.Label(frame,image=img).pack()# 显示图片

    e1 = tk.Entry(frame,width=10,textvariable=self._inputCode).pack(anchor='w')
    tk.Button(frame,text='确定',command=self.setCode).pack(anchor='w')

  #点击确定将验证码赋值给主窗口  
  def setCode(self):
      self.inputCode = self._inputCode.get()
      self.destroy() # 销毁窗口



# 主窗
class MyApp(tk.Tk):
  def __init__(self):
    global path
    # super().__init__()
    tk.Tk.__init__(self)
    #self.pack() # 若继承 tk.Frame ,此句必须有!
    self.title('UC账号管家')
    self.geometry('1000x600')
    self.iconbitmap(default='code/logo.ico')  # 加图标

    # 获取所有账号
    self.getAccountsFirst()    
    # 调起程序界面
    self.setupUI()

  # 获取所有的投放账号
  def getAccountsFirst(self):
    f = open(path + 'code/account.log')
    content = f.read()
    f.close()
    #解密账号
    _account = base64.b64decode(content).replace(' ', '').split("\n")
    __account = []
    for i in range(len(_account)):
      __account.append(_account[i].split(","))
      # self.account[i] = _account[i].split(",")
    #将账号赋值给账号界面
    self.accounts = __account

  #拉起账号界面
  def setupUI(self):
    def callback():
        argAccount = var.get()
        for account in self.accounts:
            if(account[0].decode('utf-8') == argAccount):
                # 此时做登录操作
                self.login(account[0],account[1])
    row1 = tk.LabelFrame(self)
    var = tk.StringVar()
    group=tk.LabelFrame(row1,text='选择要登录的账号').pack(padx=100)   # 基于root 制定一个框架 . 

    count = 0
    row = 1
    for name,pwd in self.accounts:
        column = int(math.fmod(count,5))
        if(row * 5 == count):
          row = row + 1
        b = tk.Radiobutton(group,text=name,variable=var,value=name,indicatoron=0,padx=30,pady=3,command=callback)
        b.grid(row=row,column=column)
        count = count + 1
        # b.pack(anchor="w")
    label = tk.Label(row1)
    label.pack()

  def login(self,username,pwd):
    #开始登陆
    login_url = 'https://e.uc.cn/'
    headers = {
        'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_12_5) AppleWebKit/537.36 (KHTML, like Gecko) '
                      'Chrome/58.0.3029.110 Safari/537.36'
    }
    options = webdriver.ChromeOptions()
    options.add_experimental_option("excludeSwitches", ["ignore-certificate-errors"])
    options.add_argument('disable-infobars')
    driver = webdriver.Chrome(chrome_options=options)
    # options.binary_location = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
    # driver.chrome_driver_binary = "C:\Program Files (x86)\Google\Chrome\Application\chrome.exe"
    # driver = webdriver.Chrome()
    driver.delete_all_cookies()
    driver.get("https://e.uc.cn/sso/static/pages/hcsso.html?authorizeUrl=http%3A%2F%2Fe.uc.cn%2Fsso%2Fv1%2Fauthorize%3Fresponse_type%3Dcode%26scope%3Dopenid%26client_id%3DeyJjdHkiOiJKV1QiLCJlbmMiOiJBMTI4Q0JDLUhTMjU2IiwiYWxnIjoiQTEyOEtXIn0.PYe1Zfthswo5iITZVGBkn5hCfdzyuQAkXWPE7VAO0eAuRNEzikoQgQ.3g8Cf2JZ2D8bhKdKlgaHGw.-56UrnfHvZPoryCii7MPSvoBUlnacryxYxBqQBLUQe3k_XJ1j3Trv-Io9JK9Pg3CyFbslM281pe8MsNGjvIfPUT7Z_Vc0N_0t_bCp3Lp4H0eJ3YODXapXgMX2qYROw1uWCbb8FcWpa37HyySgjsaOHIiDVkwmb3I_flywaJu5Tavbp09nwirOg-AJ8unie4nQXt9XqA5Dpq9F5qv36TFGVKRUbTMzvA8XGjUpE1ExL5oodpxfh8FfUfBLvF69PgjHwJiijrC702X5VHx0Og949xnmIX4znCbJ3MPUDjPwtS92IxGYypjWDTi4bfWYZYl8iPVoMbZCCz--QrmH1FUjznu6YmVavmKXasvHg6qI9Gpn5hnv--9rPp0CJgpJg_7DLyxr4D7jOj2Ju7gXGNPKqsoqP_GT1zsa1mkUZbeMSLYQVZJPwO6HLcpyZG-zH1_fz70IgIxACHBxky2V3P9WgUaExKys-e_8zawHxWpbyTl1DZlg7v1razzzrwgteHmpoJxyRUt_Gs7WQyYlCrgLrhVWEF7JKTy9MJFeepuGg8GWyzUtu86yRkIkDDL726B4J_sa779JzfutyZtVBAj36BuSj_EnvFsNau13vJ23Nyc3wm8TT8QS6xD6vrKRHWYCp7trDHcInlIczi7gnVth2c_RstjKeE8wcIPN1a1-4wkwIiUAkrnx_y2EaUECHp5vHITjbpBD1rPzZwKZT8Mlw.9hgjcRYoM3FgvXVF7YOgIw%26redirect_uri%3Dhttps%3A%2F%2Fe.uc.cn%2Fuc%2Fsso_auth_code%26target_url%3Dhttp%3A%2F%2Fe.uc.cn%2Fuc%2F%3FisFromPortal%3D0")

    # 将所有的cookie清除
    driver.delete_all_cookies()

    #重新获取验证码并且将cookie写入浏览器
    cookies = self.getCode()
    for cok in cookies:
        driver.add_cookie({'name':cok,'value':cookies[cok]})


    # 弹窗
    code = self.ask_inputCode()
    userName = driver.find_element_by_id('username')
    password = driver.find_element_by_id('password')
    captchaCode = driver.find_element_by_name('captchaCode')

    userName.clear()
    userName.send_keys(username.decode('utf-8'))
    password.clear()
    password.send_keys(pwd.decode('utf-8'))
    captchaCode.send_keys(code.decode('utf-8'))

    submit = driver.find_element_by_class_name('js-hc-loginBtn')
    submit.click()

  def ask_inputCode(self):
    inputOjb = showCode()
    self.wait_window(inputOjb) # 这一句很重要!!!
    return inputOjb.inputCode

  def getCode(self):
    global path
    img_src = "https://e.uc.cn/sso/auth.jpg"
    r = requests.get(img_src)
    r.raise_for_status()
    with open(path + 'code/code.gif',"wb") as f: #开始写文件,wb代表写二进制文件
        f.write(r.content)
        f.close()
    f.close()
    cookies = requests.utils.dict_from_cookiejar(r.cookies)
    return cookies

if __name__ == '__main__':
  app = MyApp()
  app.mainloop()

关于环境:

我是在mac里面先写完,然后最后因为要编译成exe,所以必须在window里面重新再搭建环境。
遇到的问题:
问题一、
同样是使用Tkinter,环境py27
mac中 import tkinter as tk 是没问题的

在window中使用    import tkinter as tk  报 ImportError:No module named tkinter

解决办法:原来一直以为是没有装tkinter,因为在mac里面使用(
mac安装Tkinter库brew install homebrew/dupes/tcl-tk)重新过tkinter,以为window也要重新搞一遍
后来发现用法用错了。在window中py2.7 要使用

import Tkinter as tk

问题二、

关于代码

问题一、最开始没有使用类的方式,所有的方法都放在一个文件中,然后在弹窗将输入的值返回给主界面,一直返回不回去。如果要设置的参数个数超过两个,那么tkinter提供的标准窗口就处理不了了。怎样将自定义窗口中的数据传回主窗口?
解决办法:原来一直以为是对python的类理解不深刻,然后是Tkinter中的使用就是返回不回去
参照:https://www.jb51.net/article/119817.htm 修改成了类的方式来做。我用的第一种,把值返回给主界面

1)松耦合
说明:
主窗类,继承了 tk.Tk
弹窗类,继承了 tk.Toplevel
要点:
弹窗,将多个数据,打包,放入一个名为 username 的私有 list 对象,销毁弹窗
主窗,待弹窗运行后,通过wait_window方法,取得弹窗的名为 username 私有变量

import tkinter as tk
'''松耦合'''
# 弹窗
class MyDialog(tk.Toplevel):
  def __init__(self):
    super().__init__()
    self.title('设置用户信息')
    # 弹窗界面
    self.setup_UI()
  def setup_UI(self):
    # 第一行(两列)
    row1 = tk.Frame(self)
    row1.pack(fill="x")
    tk.Label(row1, text='姓名:', width=8).pack(side=tk.LEFT)
    self.name = tk.StringVar()
    tk.Entry(row1, textvariable=self.name, width=20).pack(side=tk.LEFT)
    # 第二行
    row2 = tk.Frame(self)
    row2.pack(fill="x", ipadx=1, ipady=1)
    tk.Label(row2, text='年龄:', width=8).pack(side=tk.LEFT)
    self.age = tk.IntVar()
    tk.Entry(row2, textvariable=self.age, width=20).pack(side=tk.LEFT)
    # 第三行
    row3 = tk.Frame(self)
    row3.pack(fill="x")
    tk.Button(row3, text="取消", command=self.cancel).pack(side=tk.RIGHT)
    tk.Button(row3, text="确定", command=self.ok).pack(side=tk.RIGHT)
  def ok(self):
    self.userinfo = [self.name.get(), self.age.get()] # 设置数据
    self.destroy() # 销毁窗口
  def cancel(self):
    self.userinfo = None # 空!
    self.destroy()
# 主窗
class MyApp(tk.Tk):
  def __init__(self):
    super().__init__()
    #self.pack() # 若继承 tk.Frame ,此句必须有!
    self.title('用户信息')
    # 程序参数/数据
    self.name = '张三'
    self.age = 30
    # 程序界面
    self.setupUI()
  def setupUI(self):
    # 第一行(两列)
    row1 = tk.Frame(self)
    row1.pack(fill="x")
    tk.Label(row1, text='姓名:', width=8).pack(side=tk.LEFT)
    self.l1 = tk.Label(row1, text=self.name, width=20)
    self.l1.pack(side=tk.LEFT)
    # 第二行
    row2 = tk.Frame(self)
    row2.pack(fill="x")
    tk.Label(row2, text='年龄:', width=8).pack(side=tk.LEFT)
    self.l2 = tk.Label(row2, text=self.age, width=20)
    self.l2.pack(side=tk.LEFT)
    # 第三行
    row3 = tk.Frame(self)
    row3.pack(fill="x")
    tk.Button(row3, text="设置", command=self.setup_config).pack(side=tk.RIGHT)
  # 设置参数
  def setup_config(self):
    # 接收弹窗的数据
    res = self.ask_userinfo()
    #print(res)
    if res is None: return
    # 更改参数
    self.name, self.age = res
    # 更新界面
    self.l1.config(text=self.name)
    self.l2.config(text=self.age)
  # 弹窗
  def ask_userinfo(self):
    inputDialog = MyDialog()
    self.wait_window(inputDialog) # 这一句很重要!!!
    return inputDialog.userinfo
if __name__ == '__main__':
  app = MyApp()
  app.mainloop()

2)紧耦合
说明:
主窗类,继承了 tk.Tk
弹窗类,继承了 tk.Toplevel
要点:
弹窗,显式地保存父窗口,显式地修改父窗口数据,显式地更新父窗口部件,最后销毁弹窗
主窗,待弹窗运行后,通过wait_window方法,返回 None


import tkinter as tk
'''紧耦合'''
# 弹窗
class PopupDialog(tk.Toplevel):
  def __init__(self, parent):
    super().__init__()
    self.title('设置用户信息')
    self.parent = parent # 显式地保留父窗口
    # 第一行(两列)
    row1 = tk.Frame(self)
    row1.pack(fill="x")
    tk.Label(row1, text='姓名:', width=8).pack(side=tk.LEFT)
    self.name = tk.StringVar()
    tk.Entry(row1, textvariable=self.name, width=20).pack(side=tk.LEFT)
    # 第二行
    row2 = tk.Frame(self)
    row2.pack(fill="x", ipadx=1, ipady=1)
    tk.Label(row2, text='年龄:', width=8).pack(side=tk.LEFT)
    self.age = tk.IntVar()
    tk.Entry(row2, textvariable=self.age, width=20).pack(side=tk.LEFT)
    # 第三行
    row3 = tk.Frame(self)
    row3.pack(fill="x")
    tk.Button(row3, text="取消", command=self.cancel).pack(side=tk.RIGHT)
    tk.Button(row3, text="确定", command=self.ok).pack(side=tk.RIGHT)
  def ok(self):
    # 显式地更改父窗口参数
    self.parent.name = self.name.get()
    self.parent.age = self.age.get()
    # 显式地更新父窗口界面
    self.parent.l1.config(text=self.parent.name)
    self.parent.l2.config(text=self.parent.age)
    self.destroy() # 销毁窗口
  def cancel(self):
    self.destroy()
# 主窗
class MyApp(tk.Tk):
  def __init__(self):
    super().__init__()
    # self.pack() # 若继承 tk.Frame,此句必须有!!!
    self.title('用户信息')
    # 程序参数
    self.name = '张三'
    self.age = 30
    # 程序界面
    self.setupUI()
  def setupUI(self):
    # 第一行(两列)
    row1 = tk.Frame(self)
    row1.pack(fill="x")
    tk.Label(row1, text='姓名:', width=8).pack(side=tk.LEFT)
    self.l1 = tk.Label(row1, text=self.name, width=20)
    self.l1.pack(side=tk.LEFT)
    # 第二行
    row2 = tk.Frame(self)
    row2.pack(fill="x")
    tk.Label(row2, text='年龄:', width=8).pack(side=tk.LEFT)
    self.l2 = tk.Label(row2, text=self.age, width=20)
    self.l2.pack(side=tk.LEFT)
    # 第三行
    row3 = tk.Frame(self)
    row3.pack(fill="x")
    tk.Button(row3, text="设置", command=self.setup_config).pack(side=tk.RIGHT)
  # 设置参数
  def setup_config(self):
    pw = PopupDialog(self)
    self.wait_window(pw) # 这一句很重要!!!
    return
if __name__ == '__main__':
  app = MyApp()
  app.mainloop()

问题二、
“Use of super on an old style class” py27环境里面super不支持使用,需要使用老方式来显示
Python类的继承有两种写法,一是老样式(old style),一是新样式(new style),都能解决上述的错误。
解决办法:
参照我的代码。
好的博客:这个涉及到关于py的类和继承,这个有时间再看下,因为直接做项目,蒙头做,回过头再研究吧
https://blog.csdn.net/ws_cs_dn/article/details/37699195
https://blog.csdn.net/yiifaa/article/details/78068970

问题三、
下载验证码到本地,然后使用Tkinter,打开该文件一直打不开。不显示该图片。最开始使用Lab方式来显示图片,反正死活没显示出来,后来使用tk.Canvas 然后显示出来图片……
解释:首先,查了下Tkinter的Lab只支持gif格式的图片显示,如果想支持多格式,需要下PIL包,然后用PIL的TkinterImage来打开图片(参照我代码),最开始我理解的,可能我下载的图片格式不对,我直接把验证码文件下载下来在保存的时候,直接给了后缀,可能文件格式有问题~这是一个
另一个是Lab的用法,可能在类里面的使用方式不对。

另外,我使用 tk.Canvas显示图片,也要格外注意self的使用,不然可能会出现弹窗和主界面主键混乱的情况。。。虽然我实现了功能,很多我也要重新再看下,我只给出解决的方式,至于为什么…………以后慢慢研究吧……

问题四、
在使用chromeDriver时,chromeDriver会有一个弹窗,如何隐藏chromeDriver.exe 黑色弹窗?
修改selenium包文件中
参照:https://stackoverflow.com/questions/33983860/hide-chromedriver-console-in-python
*\selenium\webdriver\common\services.py文件
首先引入 from win32process import CREATE_NO_WINDOW
然后修改start 方法(如下例子),增加creationflags=CREATE_NO_WINDOW

def start(self):
    """
    Starts the Service.

    :Exceptions:
     - WebDriverException : Raised either when it can't start the service
       or when it can't connect to the service
    """
    try:
        cmd = [self.path]
        cmd.extend(self.command_line_args())
        self.process = subprocess.Popen(cmd, env=self.env,
                                        close_fds=platform.system() != 'Windows',
                                        stdout=self.log_file, stderr=self.log_file, creationflags=CREATE_NO_WINDOW)
    except TypeError:
        raise

这样子,就不会显示黑色chromeDriver.exe 窗口啦,嘻嘻嘻嘻~

屏蔽,运行自动化脚本
option = webdriver.ChromeOptions()
option.add_argument(‘disable-infobars’)
webdriver.Chrome(chrome_options = option,desired_capabilities = None)

关于打包:
python
打包使用了pyinstaller 来打包。
pip install pyinstaller
pyinstaller -F uc-end.py –noconsole

你可能感兴趣的:(python)