python+selenium连接数据库，从数据库中获取想要的数据

        

      python+selenium连接数据库，从数据库中获取想要的数据

           将下面的脚本放在一个参数的模块当中(parameter.vaccine)，需要用时直接调用就可以了              

# -*- coding:utf-8 -*-             # 声明使用编码
import pymssql                     #引入数据库
import random                      #因为是随机生成的数据，所以必须要引入random
def sql_server(abc="select Barcode from CDC.Vaccines "): #定义一个函数，abc是默认的参数
    server = "192.168.1.197:42317" #服务器名称
    user = "sa"                    #登录名
    password = "jdhsb"             #登录密码
    database = "CDC_DBTest"        #数据库名
    conn = pymssql.connect(server, user, password, database, charset="utf8")#打开数据库连接
    cur = conn.cursor()            # 使用cursor()方法获取操作游标
    sql = abc
    cur.execute(sql)               # 使用execute方法执行SQL语句
    rows = cur.fetchall()          #使用 fetchone() 方法获取一条数据
    #print(rows)
    print(type(rows))
    conn.close()                   #关闭数据库
    return rows
sql1 = sql_server(r"select Barcode from CDC.Vaccines where"
                  r" LocationId=(select id from CDC.Locations where"
                  r" Name=N'大浪社康接种点') and VaccineTypeId=(select id from CDC.VaccineTypes "
                  r"where name like N'%辽宁成大%冻干狂犬疫苗%') and VaccineStatus=N'1' and PackagingType='3' "
                  r"and Barcode is not null")  #数据库查询语句，前面加r的意思是格式不变
barcode = r''.join(random.choice(sql1))   #用join 方法获取随机的数据
#print(barcode)                           #打印出数据

 在放函数的模块中调用上面的参数barcode

# -*- coding:utf-8 -*-                          #声明使用编码
from time import sleep                          #引入时间
from selenium import webdriver                  #引入web驱动
from function.login import username,password    #引入登录模块的参数username,password
from parameter.vaccine import barcode           #引入数据库的参数barcode
def login(username,password):
       global driver
       driver = webdriver.Chrome()
       driver.get('http://sdsf/')
       driver.find_element_by_xpath('//*[@id="UserName"]').send_keys(username[2])#输入用户名
       driver.find_element_by_xpath('//*[@id="Password"]').send_keys(password)   #输入密码
       driver.find_element_by_xpath('//*[@id="loginForm"]/div[4]/input').click() #点击【登录】按钮
        sleep(1)
def yimiao(barcode):
        driver.find_element_by_xpath('//label[text()="疫苗条码"]//../input').send_keys(barcode)  #将参数barcode传入yimiao这个函数中
yimiao(barcode) #调用这个yimiao函数