有了ORM,我們就可以把Web App需要的3個表用Model表示出來:
import time, uuidfrom transwarp.db import next_idfrom transwarp.orm import Model, StringField, BooleanField, FloatField, TextFieldclass User(Model): __table__ = 'users' id = StringField(primary_key=True, default=next_id, ddl='varchar(50)') email = StringField(updatable=False, ddl='varchar(50)') password = StringField(ddl='varchar(50)') admin = BooleanField() name = StringField(ddl='varchar(50)') image = StringField(ddl='varchar(500)') created_at = FloatField(updatable=False, default=time.time)class Blog(Model): __table__ = 'blogs' id = StringField(primary_key=True, default=next_id, ddl='varchar(50)') user_id = StringField(updatable=False, ddl='varchar(50)') user_name = StringField(ddl='varchar(50)') user_image = StringField(ddl='varchar(500)') name = StringField(ddl='varchar(50)') summary = StringField(ddl='varchar(200)') content = TextField() created_at = FloatField(updatable=False, default=time.time)class Comment(Model): __table__ = 'comments' id = StringField(primary_key=True, default=next_id, ddl='varchar(50)') blog_id = StringField(updatable=False, ddl='varchar(50)') user_id = StringField(updatable=False, ddl='varchar(50)') user_name = StringField(ddl='varchar(50)') user_image = StringField(ddl='varchar(500)') content = TextField() created_at = FloatField(updatable=False, default=time.time)
在編寫ORM時,給一個Field增加一個default參數可以讓ORM自己填入預設值,非常方便。並且,預設值可以作為函數對象傳入,在調用insert()時自動計算。
例如,主鍵id的預設值是函數next_id,建立時間created_at的預設值是函數time.time,可以自動化佈建當前日期和時間。
日期和時間用float類型儲存在資料庫中,而不是datetime類型,這麼做的好處是不必關心資料庫的時區以及時區轉換問題,排序非常簡單,顯示的時候,只需要做一個float到str的轉換,也非常容易。
初始化資料庫表
如果表的數量很少,可以手寫建立表的SQL指令碼:
-- schema.sqldrop database if exists awesome;create database awesome;use awesome;grant select, insert, update, delete on awesome.* to 'www-data'@'localhost' identified by 'www-data';create table users ( `id` varchar(50) not null, `email` varchar(50) not null, `password` varchar(50) not null, `admin` bool not null, `name` varchar(50) not null, `image` varchar(500) not null, `created_at` real not null, unique key `idx_email` (`email`), key `idx_created_at` (`created_at`), primary key (`id`)) engine=innodb default charset=utf8;create table blogs ( `id` varchar(50) not null, `user_id` varchar(50) not null, `user_name` varchar(50) not null, `user_image` varchar(500) not null, `name` varchar(50) not null, `summary` varchar(200) not null, `content` mediumtext not null, `created_at` real not null, key `idx_created_at` (`created_at`), primary key (`id`)) engine=innodb default charset=utf8;create table comments ( `id` varchar(50) not null, `blog_id` varchar(50) not null, `user_id` varchar(50) not null, `user_name` varchar(50) not null, `user_image` varchar(500) not null, `content` mediumtext not null, `created_at` real not null, key `idx_created_at` (`created_at`), primary key (`id`)) engine=innodb default charset=utf8;
如果表的數量很多,可以從Model對象直接通過指令碼自動產生SQL指令碼,使用更簡單。
把SQL指令碼放到MySQL命令列裡執行:
$ mysql -u root -p < schema.sql
我們就完成了資料庫表的初始化。
編寫資料存取碼
接下來,就可以真正開始編寫代碼操作對象了。比如,對於User對象,我們就可以做如下操作:
# test_db.pyfrom models import User, Blog, Commentfrom transwarp import dbdb.create_engine(user='www-data', password='www-data', database='awesome')u = User(name='Test', email='test@example.com', password='1234567890', image='about:blank')u.insert()print 'new user id:', u.idu1 = User.find_first('where email=?', 'test@example.com')print 'find user\'s name:', u1.nameu1.delete()u2 = User.find_first('where email=?', 'test@example.com')print 'find user:', u2
可以在MySQL用戶端命令列查詢,看看資料是不是正常儲存到MySQL裡面了。