python自動化--模塊操作之re、MySQL、Excel

一、python自有模塊正則

import re

# re.match只匹配字符串的開始，如果字符串開始不符合正則表達式，則匹配失敗，函數返回None
print(re.match("www","wwwwccc").group())   #在起始位置匹配
print(re.match("www","wcccwww"))           #不在起始位置匹配，返回None

# re.search掃描整個字符串並返回第一個成功的匹配
print(re.search("haha","woshihahanishishui").group())

# re.findall從左到右掃描字符串，按順序返回匹配，如果無匹配結果則返回空列表
print(re.findall("\d","one1two2three3four4"))
print(re.findall("\d","onetwothreefour"))

# sub用於替換字符串中的匹配項
print(re.sub("g..t","good","goot geet up"))
# split返回切割后的列表
print(re.split("\+","123+456*789+abcd"))

group及groups區別：

 re.compile將正則字符串編譯成正則表達式對象：

 

# ['AD123453', 'AC342123', 'AR343212']
# [A-Z]{2}\d{6}

# ^(13|14|15|17|18)\d{9}
# ^1[34578]\d{9}

推薦一個正則表達式的小工具：https://pan.baidu.com/s/1EXrSF5pS8hJ7050BOOtL9w

 

二、python第三方模塊操作MySQL

連接數據庫：

import mysql.connector
config = {
    "host":"127.0.0.1",
    "user":"root",
    "passwd":"vertrigo",
    "port":3306,
    "db":"test",
    "charset":"utf8",
　　　　"buffered"='True'

}

try:
    db = mysql.connector.connect(**config)
except mysql.connector.Error as e:
    print("連接數據庫失敗！",str(e))

插入數據：

 2 cursor = db.cursor(buffered=True)   #buffered=True會把結果集保存到本地並一次性返回，這樣可以提高性能
try:
    #第一種：直接字符串插入方式
    # sql_insert1="insert into student (name, age) values ('xiao', 27)"
    # cursor.execute(sql_insert1)

    #第二種：元組連接插入方式
    sql_insert2="insert into student (name, age) values (%s, %s)"
    #此處的%s為占位符，而不是格式化字符串，所以age用%s
    # data=('xiaoqiang',18)
    # cursor.execute(sql_insert2,data)
    data = [("xiao",20),
            ("xian",25),
            ("rourou",27),
            ("juju",28)]
    cursor.executemany(sql_insert2,data)

    #如果表引擎為Innodb，執行完成后需執行commit進行事務提交
    db.commit()
    #cursor.execute('commit')
except mysql.connector.Error as e:
    print('插入失敗！', str(e))
finally:
    cursor.close()
    db.close()

刪除數據：

cursor = db.cursor(buffered=True)
try:
    sql_del = "delete from student where name=%s and age=%s"
    data_del = [
        ("cui",28),
        ("hao",27)]
    cursor.executemany(sql_del,data_del)
    db.commit()
except mysql.connector.Error as e:
    print("刪除數據失敗！",str(e))
finally:
    cursor.close()
    db.close()

修改數據：

cursor = db.cursor(buffered=True)
try:
    sql_update = "update student set age = 28 where name='kai'"
    cursor.execute(sql_update)
    db.commit()
except mysql.connector.Error as e:
    print('修改數據失敗',str(e))
finally:
    cursor.close()
    db.close()

查詢數據：

cursor = db.cursor(buffered=True)
try:
    # sql_select1 = "select * from student where age>%s"
    # cursor.execute(sql_select1,(1,))

    sql_select2 = "select * from student where age>%s"
    cursor.execute(sql_select2,(26,))
    datas1 = cursor.fetchall()         #如果在后加上[1]代表是取第一條數據

    cursor.execute(sql_select2,(20,))
    datas2 = cursor.fetchone()[1]      #如果在后加上[1]代表是取第一個字段
    datas3 = cursor.fetchmany(5)       #返回5條數據

    print(datas1)
    print(datas2)
    print(datas3)
except mysql.connector.Error as e:
    print("查詢數據失敗！",str(e))
finally:
    cursor.close()
    db.close()

三、python第三方模塊操作Excel

注意高能：openpyxl只能操作xlsx文件而不能操作xls文件！所以在創建的時候一定要新建.xlsx格式的Excel！！！

import openpyxl
#打開文件
path_file = "D:/PycharmProjects/open.xlsx"
wp = openpyxl.load_workbook(path_file)

print("獲取所有工作表名：",wp.get_sheet_names())
# sheet = wp.get_sheet_by_name("Sheet1")   #獲取指定的工作表sheet
# print("獲取指定的工作表名:",sheet.title)
sheet2 = wp.get_active_sheet()             #獲取活動的工作表sheet,一般是當前打開的sheet頁
print("獲取活動的工作表名:",sheet2.title)


# 操作單元格:數據的讀取與寫入
# 獲取單元格數據
print(sheet2['A1'].value)
print(sheet2.cell(row=2,column=1).value)    #獲取第二行第一列的值

# 數據的寫入
sheet2.cell(row=4,column=1).value = "工作總結"
sheet2['C3']='cs'

print("最大列數",sheet2.max_column)
print("最大行數",sheet2.max_row)

# wp.save('D:/PycharmProjects/open1.xlsx')     #另存為
wp.save("open.xlsx")                           #保存,默認保存在當前目錄下
wp.save(path_file)                             #覆蓋保存

小練習：

1、Python里面match()和search()的區別？

python正則中的match是在字符串的起始位置進行匹配，如果起始位置匹配不成功就會報錯。如下例子：
import re
print(re.match("www","wwccc").group())
python正則中的search是在字符串中匹配第一次匹配成功的字符串並返回。如下例子：
import re
print(re.search("ws","wwcccwss").group())

2、以<div><span>test</span></div>進行匹配

<.*>意思為：匹配以<為開始的所有的字符，遇到回車換行中斷匹配
 <.*>結果為：<div><span>test</span></div>
<.*?>意思為：匹配以<為開始的字符串，出現一次就返回
 <.*?>結果為：
 <div>
 <span>
 <span>
 <div>

3、獲取URL的后綴名

abc = ("http://www.baidu.cn/archive/6688431.html")
print(abc.split('.')[-1])