目錄
- 1、連接PG庫
- 2、連接MySQL
- 2.1 連接數據庫
- 2.2 創建數據庫和表
- 2.3 插入數據
- 2.4 數據庫查詢操作
- 2.5 數據庫更新操作
- 2.6 刪除數據操作
- 3、連接Mongo庫
- 3.1 判讀庫是否存在
- 3.2 創建集合（表）
- 3.3 插入集合
- 3.4 返回 _id 字段
- 3.5 插入多個文檔
- 3.6 插入指定 _id 的多個文檔
- 3.7 查詢一條數據
- 3.8 查詢集合中所有數據
- 3.9 查詢指定字段的數據
- 3.10 根據指定條件查詢
- 3.11 高級查詢
- 3.12 使用正則表達式查詢
- 3.13 返回指定條數記錄
- 3.14 修改數據
- 3.15 排序
- 3.16 刪除數據
- 3.17 刪除集合中的所有文檔
- 3.17 刪除集合
Postgres/Mysql/Mongo是本人在工作中最常用到的數據庫。現羅列了python操作這三種數據庫的基本操作，只要掌握了基本的操作，再多加訓練，其實可以應用的得心應手。
1、連接PG庫
## 導入psycopg2包
import psycopg2
## 連接到一個給定的數據庫
conn = psycopg2.connect(database="zabbix", user="zabbix",password="zabbix", host="127.0.0.1", port="5432")
## 建立游標，用來執行數據庫操作
cursor = conn.cursor()
## 執行SQL命令
#cursor.execute("CREATE TABLE test_conn(id int, name text)")
#cursor.execute("INSERT INTO test_conn values(1,'haha')")
## 提交SQL命令
#conn.commit()
## 執行SQL SELECT命令
cursor.execute("select * from drules;")
## 獲取SELECT返回的元組
rows = cursor.fetchall()
print('druleid|proxy_hostid|name|iprange| delay|nextcheck|status')
for row in rows:
#print(row)
print(row[0],row[1],row[2],row[3],row[4],row[5])
## 關閉游標
cursor.close()
## 關閉數據庫連接
conn.close()
2、連接MySQL
2.1 連接數據庫
連接數據庫前，請先確認以下事項：
- 您已經創建了數據庫 TESTDB.
- 在TESTDB數據庫中您已經創建了表 EMPLOYEE
- EMPLOYEE表字段為FIRST_NAME, LAST_NAME, AGE, SEX 和 INCOME。
- 連接數據庫TESTDB使用的用戶名為"testuser" ，密碼為 “test123”,你可以可以自己設定或者直接使用root用戶名及其密碼，Mysql數據庫用戶授權請使用Grant命令。
- 在你的機子上已經安裝了 Python MySQLdb 模塊。 如果您對sql語句不熟悉，可以訪問我們的 SQL基礎教程
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import MySQLdb
# 打開數據庫連接
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' )
# 使用cursor()方法獲取操作游標
cursor = db.cursor()
# 使用execute方法執行SQL語句
cursor.execute("SELECT VERSION()")
# 使用 fetchone() 方法獲取一條數據
data = cursor.fetchone()
print "Database version : %s " % data
# 關閉數據庫連接
db.close()
2.2 創建數據庫和表
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import MySQLdb
# 打開數據庫連接
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' )
# 使用cursor()方法獲取操作游標
cursor = db.cursor()
# 如果數據表已經存在使用 execute() 方法刪除表。
cursor.execute("DROP TABLE IF EXISTS EMPLOYEE")
# 創建數據表SQL語句
sql = """CREATE TABLE EMPLOYEE (
FIRST_NAME CHAR(20) NOT NULL,
LAST_NAME CHAR(20),
AGE INT,
SEX CHAR(1),
INCOME FLOAT )"""
cursor.execute(sql)
# 關閉數據庫連接
db.close()
2.3 插入數據
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import MySQLdb
# 打開數據庫連接
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' )
# 使用cursor()方法獲取操作游標
cursor = db.cursor()
# SQL 插入語句
sql = """INSERT INTO EMPLOYEE(FIRST_NAME,
LAST_NAME, AGE, SEX, INCOME)
VALUES ('Mac', 'Mohan', 20, 'M', 2000)"""
try:
# 執行sql語句
cursor.execute(sql)
# 提交到數據庫執行
db.commit()
except:
# Rollback in case there is any error
db.rollback()
# 關閉數據庫連接
db.close()
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import MySQLdb
# 打開數據庫連接
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' )
# 使用cursor()方法獲取操作游標
cursor = db.cursor()
# SQL 插入語句
sql = "INSERT INTO EMPLOYEE(FIRST_NAME, \
LAST_NAME, AGE, SEX, INCOME) \
VALUES (%s, %s, %s, %s, %s )" % \
('Mac', 'Mohan', 20, 'M', 2000)
try:
# 執行sql語句
cursor.execute(sql)
# 提交到數據庫執行
db.commit()
except:
# 發生錯誤時回滾
db.rollback()
# 關閉數據庫連接
db.close()
2.4 數據庫查詢操作
Python查詢Mysql使用 fetchone() 方法獲取單條數據, 使用fetchall() 方法獲取多條數據。
- fetchone(): 該方法獲取下一個查詢結果集。結果集是一個對象
- fetchall():接收全部的返回結果行.
- rowcount: 這是一個只讀屬性，并返回執行execute()方法后影響的行數。
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import MySQLdb
# 打開數據庫連接
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' )
# 使用cursor()方法獲取操作游標
cursor = db.cursor()
# SQL 查詢語句
sql = "SELECT * FROM EMPLOYEE \
WHERE INCOME > %s" % (1000)
try:
# 執行SQL語句
cursor.execute(sql)
# 獲取所有記錄列表
results = cursor.fetchall()
for row in results:
fname = row[0]
lname = row[1]
age = row[2]
sex = row[3]
income = row[4]
# 打印結果
print "fname=%s,lname=%s,age=%s,sex=%s,income=%s" % \
(fname, lname, age, sex, income )
except:
print "Error: unable to fecth data"
# 關閉數據庫連接
db.close()
2.5 數據庫更新操作
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import MySQLdb
# 打開數據庫連接
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' )
# 使用cursor()方法獲取操作游標
cursor = db.cursor()
# SQL 更新語句
sql = "UPDATE EMPLOYEE SET AGE = AGE + 1 WHERE SEX = '%c'" % ('M')
try:
# 執行SQL語句
cursor.execute(sql)
# 提交到數據庫執行
db.commit()
except:
# 發生錯誤時回滾
db.rollback()
# 關閉數據庫連接
db.close()
2.6 刪除數據操作
#!/usr/bin/python
# -*- coding: UTF-8 -*-
import MySQLdb
# 打開數據庫連接
db = MySQLdb.connect("localhost", "testuser", "test123", "TESTDB", charset='utf8' )
# 使用cursor()方法獲取操作游標
cursor = db.cursor()
# SQL 刪除語句
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
try:
# 執行SQL語句
cursor.execute(sql)
# 提交修改
db.commit()
except:
# 發生錯誤時回滾
db.rollback()
# 關閉連接
db.close()
3、連接Mongo庫
3.1 判讀庫是否存在
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
dblist = myclient.list_database_names()
# dblist = myclient.database_names()
if "runoobdb" in dblist:
print("數據庫已存在！")
3.2 創建集合（表）
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
collist = mydb. list_collection_names()
# collist = mydb.collection_names()
if "sites" in collist: # 判斷 sites 集合是否存在
print("集合已存在！")
else:
mycol = mydb["sites"]
3.3 插入集合
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
mydict = { "name": "RUNOOB", "alexa": "10000", "url": "https://www.runoob.com" }
x = mycol.insert_one(mydict)
print(x)
print(x)
3.4 返回 _id 字段
insert_one() 方法返回 InsertOneResult 對象，該對象包含 inserted_id 屬性，它是插入文檔的 id 值。
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient('mongodb://localhost:27017/')
mydb = myclient['runoobdb']
mycol = mydb["sites"]
mydict = { "name": "Google", "alexa": "1", "url": "https://www.google.com" }
x = mycol.insert_one(mydict)
print(x.inserted_id)
3.5 插入多個文檔
集合中插入多個文檔使用 insert_many() 方法，該方法的第一參數是字典列表。
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
mylist = [
{ "name": "Taobao", "alexa": "100", "url": "https://www.taobao.com" },
{ "name": "QQ", "alexa": "101", "url": "https://www.qq.com" },
{ "name": "Facebook", "alexa": "10", "url": "https://www.facebook.com" },
{ "name": "知乎", "alexa": "103", "url": "https://www.zhihu.com" },
{ "name": "Github", "alexa": "109", "url": "https://www.github.com" }
]
x = mycol.insert_many(mylist)
# 輸出插入的所有文檔對應的 _id 值
print(x.inserted_ids)
3.6 插入指定 _id 的多個文檔
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["site2"]
mylist = [
{ "_id": 1, "name": "RUNOOB", "cn_name": "菜鳥教程"},
{ "_id": 2, "name": "Google", "address": "Google 搜索"},
{ "_id": 3, "name": "Facebook", "address": "臉書"},
{ "_id": 4, "name": "Taobao", "address": "淘寶"},
{ "_id": 5, "name": "Zhihu", "address": "知乎"}
]
x = mycol.insert_many(mylist)
# 輸出插入的所有文檔對應的 _id 值
print(x.inserted_ids)
3.7 查詢一條數據
使用 find_one() 方法來查詢集合中的一條數據。
查詢 sites 文檔中的第一條數據：
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
x = mycol.find_one()
print(x)
3.8 查詢集合中所有數據
find() 方法可以查詢集合中的所有數據，類似 SQL 中的 SELECT * 操作。
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
for x in mycol.find():
print(x)
3.9 查詢指定字段的數據
可以使用 find() 方法來查詢指定字段的數據，將要返回的字段對應值設置為 1。
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
for x in mycol.find({},{ "_id": 0, "name": 1, "alexa": 1 }):
print(x)
除了 _id 你不能在一個對象中同時指定 0 和 1，如果你設置了一個字段為 0，則其他都為 1，反之亦然。
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
for x in mycol.find({},{ "alexa": 0 }):
print(x)
3.10 根據指定條件查詢
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
myquery = { "name": "RUNOOB" }
mydoc = mycol.find(myquery)
for x in mydoc:
print(x)
3.11 高級查詢
以下實例用于讀取 name 字段中第一個字母 ASCII 值大于 “H” 的數據，大于的修飾符條件為 {"$gt": “H”} :
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
myquery = { "name": { "$gt": "H" } }
mydoc = mycol.find(myquery)
for x in mydoc:
print(x)
3.12 使用正則表達式查詢
正則表達式修飾符只用于搜索字符串的字段。
以下實例用于讀取 name 字段中第一個字母為 “R” 的數據，正則表達式修飾符條件為 {"$regex": “^R”} :
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
myquery = { "name": { "$regex": "^R" } }
mydoc = mycol.find(myquery)
for x in mydoc:
print(x)
3.13 返回指定條數記錄
如果我們要對查詢結果設置指定條數的記錄可以使用 limit() 方法，該方法只接受一個數字參數。
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
myresult = mycol.find().limit(3)
# 輸出結果
for x in myresult:
print(x)
3.14 修改數據
以在 MongoDB 中使用 update_one() 方法修改文檔中的記錄。該方法第一個參數為查詢的條件，第二個參數為要修改的字段。
如果查找到的匹配數據多于一條，則只會修改第一條。
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
myquery = { "alexa": "10000" }
newvalues = { "$set": { "alexa": "12345" } }
mycol.update_one(myquery, newvalues)
# 輸出修改后的 "sites" 集合
for x in mycol.find():
print(x)
update_one() 方法只能修匹配到的第一條記錄，如果要修改所有匹配到的記錄，可以使用 update_many()。
以下實例將查找所有以 F 開頭的 name 字段，并將匹配到所有記錄的 alexa 字段修改為 123：
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
myquery = { "name": { "$regex": "^F" } }
newvalues = { "$set": { "alexa": "123" } }
x = mycol.update_many(myquery, newvalues)
print(x.modified_count, "文檔已修改")
3.15 排序
sort() 方法可以指定升序或降序排序。
sort() 方法第一個參數為要排序的字段，第二個字段指定排序規則，1 為升序，-1 為降序，默認為升序。
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
mydoc = mycol.find().sort("alexa")
for x in mydoc:
print(x)
3.16 刪除數據
使用 delete_one() 方法來刪除一個文檔，該方法第一個參數為查詢對象，指定要刪除哪些數據。
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
myquery = { "name": {"$regex": "^F"} }
x = mycol.delete_many(myquery)
print(x.deleted_count, "個文檔已刪除")
刪除多個文檔
delete_many() 方法來刪除多個文檔，該方法第一個參數為查詢對象，指定要刪除哪些數據。
刪除所有 name 字段中以 F 開頭的文檔:
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
x = mycol.delete_many({})
print(x.deleted_count, "個文檔已刪除")
3.17 刪除集合中的所有文檔
delete_many() 方法如果傳入的是一個空的查詢對象，則會刪除集合中的所有文檔：
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
x = mycol.delete_many({})
print(x.deleted_count, "個文檔已刪除")
3.17 刪除集合
使用 drop() 方法來刪除一個集合。
以下實例刪除了 customers 集合：
#!/usr/bin/python3
import pymongo
myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["runoobdb"]
mycol = mydb["sites"]
mycol.drop()
到此這篇關于Python連接Postgres/Mysql/Mongo數據庫基本操作的文章就介紹到這了,更多相關Python連接數據庫內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持腳本之家！
您可能感興趣的文章:- 將Python腳本打包成MACOSAPP程序過程
- [項目布局配置]Nosql與PythonWeb-Flask框架組合
- Python使用signal定時結束AsyncIOScheduler任務的問題
- python生成可執行exe控制Microsip自動填寫號碼并撥打功能
- Python之os模塊案例詳解
