Python編程：使用pythink查詢數(shù)據(jù)庫

CharlseShan 2019-04-16

展開全文

github：https://github.com/mouday/PyThink

說下這個pythink模塊的作用

背景：
我們公司的主要使用的后端語言有兩種，python和php

而且有一個專門的DBA管理數(shù)據(jù)庫，就是說，這個數(shù)據(jù)庫不是你一個人的，有同組的成員和其他組的成員在使用，好像大部分公司都一樣。

Python爬蟲部分的業(yè)務，主要操作是插入；而PHP業(yè)務主要是查詢

最開始我使用普通的模塊pymysql，這樣每次都要去連接，關(guān)閉，很麻煩；

import pymysql 

conn = pymysql.Connect(
    host='127.0.0.1', 
    port=3306, 
    user='root',
    passwd='123456',
    db='test'
)

cursor = conn.cursor()

rows = cursor.execute('insert into student(name, age, register_date, gender) values ('xiaoming', 23, '2018-12-30', 'M')')
cursor.close()
conn.close()

后來使用了peewee 來管理數(shù)據(jù)庫鏈接，使用Model來創(chuàng)建數(shù)據(jù)庫（這個時候還沒有人管理數(shù)據(jù)庫），簡直不要太爽，不寫一行SQL語句就實現(xiàn)了表創(chuàng)建，和數(shù)據(jù)操作；

后來我這邊的表太多了，業(yè)務發(fā)展也規(guī)范了，專門有人管理數(shù)據(jù)庫。這樣就不能使用Model直接建表，而是要寫成sql語句給數(shù)據(jù)庫管理人員，審核完后再進行建表操作

from peewee import *

# 設(shè)置數(shù)據(jù)庫
db = MySQLDatabase('mysql://root:123456@127.0.0.1:3306/demo')

class BaseModel(Model):
    class Meta:
        database = db

# 定義數(shù)據(jù)表
class User(BaseModel):
    name = CharField()

User.create(name='Tom')

在這種情況下，如果使用Model就必須先寫sql建表，再把所有字段定義寫出來才可以使用，如果涉及的表字段少還好，有時候涉及100多個字段（真的有過），那光是定義就一上午，Model的優(yōu)勢反而成了累贅，心累；

于是我開始不用Model了，直接使用peewee 提供execute_sql 這樣我們又回到了原生sql的時代；

db.execute_sql('insert into student(name, age) values ('Tom', 24)')

一般情況下，會將sql執(zhí)行的語句和結(jié)果封裝到一個函數(shù)中。寫的多了，發(fā)現(xiàn)很多重復代碼，除了sql不一樣以外，其他格式基本一致。于是需要抽象出來，將相同代碼抽象封裝，以便復用。

于是pythink 出現(xiàn)了，結(jié)合了peewee 的操作方式，也保留了一部分原生sql 的格式，比如，增(insert)、刪(delete)、改(update)、查(select)基本保留了sql的關(guān)鍵字，不需要額外記憶和區(qū)別，而且可以把對表的操作單獨卸載model中，進行統(tǒng)一管理，也可以重寫基類的方法，自己實現(xiàn)自己的業(yè)務邏輯（不建議，建議取一個不一樣的函數(shù)名字）。

一起感受下其如絲般順滑的操作

安裝

pip install pythink

代碼示例

mysql> use demo;
mysql> select * from student limit 2;
+----+------+------+------+
| id | name | age  | sex  |
+----+------+------+------+
|  1 | Tom  |   24 |    1 | 
|  2 | jack |   23 |    0 |
+----+------+------+------+
2 rows in set (0.00 sec)

# -*- coding: utf-8 -*-

from pythink import ThinkModel
from playhouse.db_url import connect

# 1、使用peewee進行數(shù)據(jù)庫連接
db = connect('mysql://root:123456@127.0.0.1:3306/demo')

# 2、繼承 ThinkModel 實現(xiàn)自己的類
class StudentThinkModel(ThinkModel):
    table_name = 'student'
    database = db

# 3、數(shù)據(jù)插入insert
data = {
    'name': 'Tom',
    'age': 24
}

result = StudentThinkModel.insert(data)
print(result)
# 1

# 4、刪除
result = StudentThinkModel.delete(13)
print(result)
# 1

# 5、修改
data = {
    'name': 'Tom',
    'age': 24
}
result = StudentThinkModel.update(1, data)
print(result)
# 1

# 6、查詢數(shù)據(jù)庫中 id=1 的 name 和 age 兩個字段
students = StudentThinkModel.select(fields=['name', 'age'],  where='id=1', limit=1)
for student in students:
    print('name: {}'.format(student.name))
    print('age: {}'.format(student.age))

'''
name: Tom
age: 24
'''

# 7、記得關(guān)閉數(shù)據(jù)庫連接
db.close()