class BasePage(object):
    def __init__(self,driver):
        self.driver = driver

    def find_element(self,*loc):
        return self.driver.find_element(*loc)

    def type_text(self,text,*loc):
        self.find_element(*loc).send_keys(text)

    def click(self,*loc):
        self.find_element(*loc).click()

    def clear(self,*loc):
        self.find_element(*loc).clear()

    def get_title(self):
        return self.driver.title

from time import sleep
from selenium.webdriver.common.by import By
from pages.basePage import BasePage

class UserLoginPage(BasePage):
    username_input = (By.NAME,'user')
    pwd_input = (By.NAME,'pwd')
    captcha_input = (By.NAME, 'captcha')
    login_btn = (By.CLASS_NAME,'btn')

    def __init__(self,driver):
        BasePage.__init__(self,driver)

    def goto_login_page(self):
        self.driver.get('http://192.168.10.184:8072/jpress-4.1.5/user/login')

    def input_username(self,username):
        self.clear(*self.username_input)
        self.type_text(username,*self.username_input)

    def input_pwd(self,pwd):
        self.clear(*self.pwd_input)
        self.type_text(pwd, *self.pwd_input)

    def input_captcha(self,captcha):
        self.clear(*self.captcha_input)
        self.type_text(captcha,*self.captcha_input)

    def click_login_btn(self):
        self.click(*self.login_btn)

# -*- coding: utf-8 -*-
import pytest
from selenium.webdriver.support.wait import WebDriverWait
from pages.userLoginPage import UserLoginPage
from selenium.webdriver.support import expected_conditions as EC
from selenium import webdriver
from time import sleep
from util import util
from selenium.webdriver.common.by import By

class TestUserLogin(object):
    login_data=[
        ('','admin','这是必填内容'),
        ('admin', '', '这是必填内容'),
        ('admin', 'admin', '用户中心')
    ]

    def setup_class(self):
        option = webdriver.ChromeOptions()
        # 无头模式
        option.add_argument('headless')
        # 沙盒模式运行
        option.add_argument('no-sandbox')
        # 渲染时写入/tmp
        option.add_argument('disable-dev-shm-usage')
        # 专门应对无头浏览器不能最大化屏幕的方案
        option.add_argument("--window-size=1920,1050")

        self.driver = webdriver.Chrome(options=option)
        self.loginPage = UserLoginPage(self.driver)
        self.loginPage.goto_login_page()
        self.logger = util.get_logger()


    # 测试用户登录
    @pytest.mark.parametrize('username,pwd,expected',login_data)
    def test_user_login_username(self,username,pwd,expected):
        # 输入用户名
        self.loginPage.input_username(username)
        # 输入密码
        self.loginPage.input_pwd(pwd)
        # 记录日志
        self.logger.debug('输入用户名称：%s,输入的用户密码：%s',username,pwd)

        if username=='':
            # 点击登录
            self.loginPage.click_login_btn()
            sleep(1)
            try:
                # assert self.driver.find_element_by_id('user-error').text == expected
                assert self.driver.find_element(By.ID,'user-error').text == expected
            except AssertionError as ae:
                self.logger.error("报错，报错：%s", "用户名为空登录", exc_info=1)
        elif pwd == '':
            # 点击登录
            self.loginPage.click_login_btn()
            sleep(1)
            try:
                # assert self.driver.find_element_by_id('pwd-error').text == expected
                assert self.driver.find_element(By.ID,'pwd-error').text == expected
            except AssertionError as ae:
                self.logger.error("报错，报错：%s", "密码为空登录", exc_info=1)
        else:
            # 自动识别验证码
            captcha = util.get_code(self.driver, 'captcha-img')
            # 输入验证码
            self.loginPage.input_captcha(captcha)
            # 点击登录
            self.loginPage.click_login_btn()
            # 等待提示框
            WebDriverWait(self.driver, 5).until(EC.title_is(expected))
            sleep(1)
            # 验证
            try:
                assert self.driver.title == expected
            except AssertionError as ae:
                self.logger.error("报错，报错：%s", "登录", exc_info=1)

if __name__ == '__main__':
    pytest.main(['testUserLogin.py'])

import base64
import json
import os
import pickle
import random
import string
import time
import requests
from PIL import Image
from selenium import webdriver
from selenium.webdriver.common.by import By
import logging.handlers
import datetime

_custom_url = "http://api.jfbym.com/api/YmServer/customApi"
_token = "Z-lvHdVz4hyuYT8_SmrG53VDH7EW4req4mpkN1rQIeY"
_headers = {
    'Content-Type': 'application/json'
}


def common_verify(image, verify_type="10110"):
    # 10110 通用数英1-4位
    payload = {
        "image": base64.b64encode(image).decode(),
        "token": _token,
        "type": verify_type
    }
    resp = requests.post(_custom_url, headers=_headers, data=json.dumps(payload))
    print(resp.text)
    return resp.json()['data']['data']

# id 是要截图的元素的id
def save_img(driver,id):
    # 获取验证码图片
    t = time.time()
    path = os.path.dirname(os.path.dirname(__file__)) + '\\screenshots'
    picture_name1 = path + '\\' + str(t) + '.png'

    driver.save_screenshot(picture_name1)
    # ce = driver.find_element_by_id(id)  # id = captchaimg
    ce = driver.find_element(By.ID,id)

    left = ce.location['x']
    top = ce.location['y']
    right = ce.size['width'] + left
    height = ce.size['height'] + top

    im = Image.open(picture_name1)
    # 抠图
    img = im.crop((left, top, right, height))
    t = time.time()
    picture_name2 = path + '\\' + str(t) + '.png'
    img.save(picture_name2)  # 把验证码的图保存下来
    return picture_name2

def get_code(driver,id):
    picture = save_img(driver,id)
    with open(picture, 'rb') as f:
        s = f.read()
    return common_verify(s)

def gen_random_str():
    rand_str = ''.join(random.sample(string.ascii_letters + string.digits, 8))
    return rand_str

def save_cookie(driver, path):
    with open(path, 'wb') as filehandler:
        cookies = driver.get_cookies()
        print(cookies)
        pickle.dump(cookies, filehandler)


def load_cookie(driver, path):
    with open(path, 'rb') as cookiesfile:
        cookies = pickle.load(cookiesfile)
        for cookie in cookies:
            driver.add_cookie(cookie)

def get_logger():
    path = os.path.dirname(os.path.dirname(__file__)) + '\\logs\\'
    logger = logging.getLogger('mylogger')
    logger.setLevel(logging.DEBUG)

    rf_handler = logging.handlers.TimedRotatingFileHandler(path+'all.log',when='midnight',interval=1,backupCount=7,atTime=datetime.time(0,0,0,0))
    rf_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s"'))
    f_handler = logging.FileHandler(path+'error.log')
    f_handler.setLevel(logging.ERROR)
    f_handler.setFormatter(logging.Formatter("%(asctime)s - %(levelname)s - %(filename)s[:%(lineno)d] - %(message)s"))

    logger.addHandler(rf_handler)
    logger.addHandler(f_handler)
    return logger

if __name__ == '__main__':
    driver = webdriver.Chrome()
    driver.get('http://192.168.10.184:8072/jpress-4.1.5/admin/login')
    id = "captcha-img"
    print(get_code(driver, id))

# -*- coding: utf-8 -*-
import pytest
import os
from testcases import *

if __name__ == '__main__':
    #批量执行文件夹下的测试用例
    pytest.main(['-s', '-v',
                 'testcases/testAdminLogin.py',
                 'testcases/testArticle.py',
                 'testcases/testUserLogin.py',
                 'testcases/testUserRegister.py',
                 '-q',
                 '--clean-alluredir',
                 '--alluredir', 'reports'])
    os.system("allure generate reports -c -o reports/allure_result/")