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','這是必填內(nèi)容'),
        ('admin', '', '這是必填內(nèi)容'),
        ('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')
        # 專門應(yīng)對無頭瀏覽器不能最大化屏幕的方案
        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 通用數(shù)英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__':
    #批量執(zhí)行文件夾下的測試用例
    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/")