3 Selenium Python 数据库及文件

1 MySQL

1.1 安装

下载：MySQL-python-1.2.3.win-amd64-py2.7直接安装，要求Python2.7（Python version 2.7 required）

验证：import MySQLdb 不报错就可以了

1.2 基础

1 连接数据库：MySQLdb.connect(host='',port='',user='',passwd='',db='')

class Connection(_mysql.connection):

    """MySQL Database Connection Object"""

    default_cursor = cursors.Cursor
    
    def __init__(self, *args, **kwargs):
        """
        Create a connection to the database. It is strongly recommended
        that you only use keyword parameters. Consult the MySQL C API
        documentation for more information.
        host
          string, host to connect  
        user
          string, user to connect as
        passwd
          string, password to use
        db
          string, database to use
        port
          integer, TCP/IP port to connect to
        charset
          If supplied, the connection character set will be changed
          to this character set (MySQL-4.1 and newer). This implies
          use_unicode=True.
        """

2 操作数据库：首先需要获得一个cursor对象， 然后使用cursor的方法执行SQL

• execute(sql, args)：执行单条sql语句，接收的参数为sql语句和参数列表,返回值为受影响的行数

    def execute(self, query, args=None):

        """Execute a query.
   
        query -- string, query to execute on server
        args -- optional sequence or mapping, parameters to use with query.

        Note: If args is a sequence, then %s must be used as the
        parameter placeholder in the query. If a mapping is used,
        %(key)s must be used as the placeholder.

        Returns long integer rows affected, if any

        """

• callproc( procname, args)：执行存储过程，接收的参数为存储过程名和参数列表，返回值为受影响的行数
• executemany(sql, args)：执行单挑sql语句，但是重复执行参数列表里的参数，返回值为受影响的行数
• nextset()：移动到下一个结果集

3 接收返回值：也是使用cursor对象的方法进行接收

• fetchone()：返回一条结果
• fetchall()：返回全部结果
• fetchmany(size=None)：返回size条结果。如果size的值大于返回的结果行的数量，则会返回cursor.arraysize条数据
• scroll(value, mode='relative')：移动指针到某一行.
• mode='relative',则表示从当前所在行移动value条
• mode='absolute',则表示从结果集的第一 行移动value条.

4 关闭数据库：需要关闭cursor对象和connect对象

1.3 举例

#配置文件=============================
#DB配置文件
[db]
host = 127.0.0.1
port = 3306
user = root
passwd =
dbname = test

#配置文件解析类=============================
#!/usr/bin/env python
# coding=utf-8

import ConfigParser

class Configer(object):
    
    #构造函数，初始化ConfigParser类，并读取config文件
    def __init__(self , filePath):
        self.conf = ConfigParser.ConfigParser()
        self.conf.read(filePath)
    
    #获取key的value 
    def getConf(self , section , key):
        result = self.conf.get(section, key)
        return result

#解析配置文件=============================
#!/usr/bin/env python
# coding=utf-8

from testdb.Configer import *

class db(object):
    conf = Configer("./db.conf")
    host = conf.getConf('db', 'host')
    port = conf.getConf('db', 'port')
    user = conf.getConf('db', 'user')
    passwd = conf.getConf('db', 'passwd')
    dbname = conf.getConf('db', 'dbname')

#数据库封装及测试=============================
#!/usr/bin/env python
#encoding=UTF-8

from MySQLdb import *
import sys
from testdb.db import *

# print db.host
class MySQLEngine(object):
     
    def __init__(self):
        self.conn = None
        self.cur = None
         
    '''
            打开数据库
    '''
    def open(self,dbHost,dbPort,dbUser,dbPasswd,dbName):
        self.conn = connect(host=dbHost,port=dbPort,user=dbUser,passwd=dbPasswd,db=dbName,charset='utf8')
        self.cur = self.conn.cursor()   #拿到游标，sql语句需要游标执行
    '''
        查询一条
    '''     
    def searchOne(self,sql):
        affRows = self.cur.execute(sql) #通过游标执行查询操作,返回影响行数
        res = self.cur.fetchone()   
        return res,affRows
    '''
        查询指定条数
    '''     
    def searchMany(self,sql,size):
        affRows = self.cur.execute(sql) #通过游标执行查询操作,返回影响行数
        res = self.cur.fetchmany(size)  #每条结果都是一个tuple,所有tuple最终组成一个tuple
        return res,affRows
    '''
        查询所有
    '''
    def searchAll(self,sql):
        affRows = self.cur.execute(sql) #通过游标执行查询操作,返回影响行数
        res = self.cur.fetchall()
        return res,affRows
    '''
        该改记录，包括：增、删、改
    '''
    def modify(self,sql,params):
        affRows = self.cur.execute(sql,params)
        self.conn.commit()  #不commit不会修改到数据库中
        return affRows
    '''
        关闭数据库
    '''
    def close(self):
        self.cur.close()
        self.conn.close()
         
#实例化
mysql = MySQLEngine()
#连接数据库
mysql.open(db.host,int(db.port),db.user,db.passwd,db.dbname)
#单个查询
(result,affRows) = mysql.searchOne('SELECT * FROM t_redis WHERE id=1;')
print result[2]
print affRows
#指定个数查询
(result,affRows) = mysql.searchMany('SELECT * FROM t_redis WHERE id>1;' , 3)
print result
print affRows
#插入
sql = "INSERT INTO t_redis(test_id , test_info) VALUES(%s,%s);"
params = ('00019','豆腐')
print mysql.modify(sql, params)
mysql.close()

#测试结果=============================
张飞
1

((2, u'2', u'info_2'), (5, u'0005', u'u7a7au7075'), (15, u'00015', u'u5c06u519b'))
6

1

 2 xlutils Excel

2.1 简介

python处理excel文件主要是第三方模块库xlrd、xlwt、xluntils

• xlrd读取excel但是不能对其进行操作
• xlwt生成excel文件，不能在已有的excel文件基础上进行修改
• 如需要修改文件就要使用xluntils模块

2.2 安装

• xlutils 依赖xlrd和xlwt，pip install xlutils 安装会自动安装xlrd和xlwt
• xlutils 只能处理xls文件

 2.3 举例

#!/usr/bin/env python
# coding=utf-8

from xlrd import open_workbook
from xlutils.copy import copy
#
rbook = open_workbook(u'C:\测试数据2.xls')
rsheet = rbook.sheet_by_name('测试用例')
wbook = copy(rbook)
#读操作
print rsheet.cell(1,1).value
#写操作，通过get_sheet()获取的sheet有write()方法
wsheet = wbook.get_sheet(0)
wsheet.write(2, 4, 'changed!')
 
wbook.save(u'C:\测试数据2.xls')

 3 openpyxl Excel

3.1 简介

A Python library to read/write Excel 2007 xlsx/xlsm files

openpyxl是Python处理表格的库之一，但是它是处理excel2007/2010的格式，也就是xlsx系列，如果要处理以前的2003的表格（xls），那么则要用另外的库

如果只是要进行表格数据读取和进行简单的写数据的话，推荐使用openpyxl

3.2 安装

openpyxl依赖jdcal ，pip install openpyxl安装会自动安装jdcal 

3.3 举例

#!/usr/bin/env python
# coding=utf-8
from openpyxl.workbook import Workbook
from openpyxl.reader.excel import load_workbook

#指定excel
book = load_workbook(filename=u'C:\测试数据3.xlsx')  
#指定sheet页
sheet = book.get_sheet_by_name('xml字段')
#最大行列数
print sheet.max_row
print sheet.max_column
# column=len(sheet.column)
#获取指定cell的值
print sheet.cell(row=2, column=1).value
#指定cell写入值
sheet.cell(row=6,column=6).value = "value3333"
#保存
book.save(u'C:\测试数据3.xlsx')