Day 4 - 编写Model

有了ORM，我们就可以把Web App需要的3个表用Model表示出来：

  1. import time, uuid
  3. from transwarp.db import next_id
  4. from transwarp.orm import Model, StringField, BooleanField, FloatField, TextField
  6. class User(Model):
  7.     __table__ = 'users'
  9.     id = StringField(primary_key=True, default=next_id, ddl='varchar(50)')
  10.     email = StringField(updatable=False, ddl='varchar(50)')
  11.     password = StringField(ddl='varchar(50)')
  12.     admin = BooleanField()
  13.     name = StringField(ddl='varchar(50)')
  14.     image = StringField(ddl='varchar(500)')
  15.     created_at = FloatField(updatable=False, default=time.time)
  17. class Blog(Model):
  18.     __table__ = 'blogs'
  20.     id = StringField(primary_key=True, default=next_id, ddl='varchar(50)')
  21.     user_id = StringField(updatable=False, ddl='varchar(50)')
  22.     user_name = StringField(ddl='varchar(50)')
  23.     user_image = StringField(ddl='varchar(500)')
  24.     name = StringField(ddl='varchar(50)')
  25.     summary = StringField(ddl='varchar(200)')
  26.     content = TextField()
  27.     created_at = FloatField(updatable=False, default=time.time)
  29. class Comment(Model):
  30.     __table__ = 'comments'
  32.     id = StringField(primary_key=True, default=next_id, ddl='varchar(50)')
  33.     blog_id = StringField(updatable=False, ddl='varchar(50)')
  34.     user_id = StringField(updatable=False, ddl='varchar(50)')
  35.     user_name = StringField(ddl='varchar(50)')
  36.     user_image = StringField(ddl='varchar(500)')
  37.     content = TextField()
  38.     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脚本：

  1. -- schema.sql
  3. drop database if exists awesome;
  5. create database awesome;
  7. use awesome;
  9. grant select, insert, update, delete on awesome.* to 'www-data'@'localhost' identified by 'www-data';
  11. create table users (
  12.     `id` varchar(50) not null,
  13.     `email` varchar(50) not null,
  14.     `password` varchar(50) not null,
  15.     `admin` bool not null,
  16.     `name` varchar(50) not null,
  17.     `image` varchar(500) not null,
  18.     `created_at` real not null,
  19.     unique key `idx_email` (`email`),
  20.     key `idx_created_at` (`created_at`),
  21.     primary key (`id`)
  22. ) engine=innodb default charset=utf8;
  24. create table blogs (
  25.     `id` varchar(50) not null,
  26.     `user_id` varchar(50) not null,
  27.     `user_name` varchar(50) not null,
  28.     `user_image` varchar(500) not null,
  29.     `name` varchar(50) not null,
  30.     `summary` varchar(200) not null,
  31.     `content` mediumtext not null,
  32.     `created_at` real not null,
  33.     key `idx_created_at` (`created_at`),
  34.     primary key (`id`)
  35. ) engine=innodb default charset=utf8;
  37. create table comments (
  38.     `id` varchar(50) not null,
  39.     `blog_id` varchar(50) not null,
  40.     `user_id` varchar(50) not null,
  41.     `user_name` varchar(50) not null,
  42.     `user_image` varchar(500) not null,
  43.     `content` mediumtext not null,
  44.     `created_at` real not null,
  45.     key `idx_created_at` (`created_at`),
  46.     primary key (`id`)
  47. ) engine=innodb default charset=utf8;

如果表的数量很多，可以从Model对象直接通过脚本自动生成SQL脚本，使用更简单。

把SQL脚本放到MySQL命令行里执行：

  1. $ mysql -u root -p < schema.sql

我们就完成了数据库表的初始化。

编写数据访问代码

接下来，就可以真正开始编写代码操作对象了。比如，对于User对象，我们就可以做如下操作：

  1. # test_db.py
  3. from models import User, Blog, Comment
  5. from transwarp import db
  7. db.create_engine(user='www-data', password='www-data', database='awesome')
  9. u = User(name='Test', email='test@example.com', password='1234567890', image='about:blank')
  11. u.insert()
  13. print 'new user id:', u.id
  15. u1 = User.find_first('where email=?', 'test@example.com')
  16. print 'find user\'s name:', u1.name
  18. u1.delete()
  20. u2 = User.find_first('where email=?', 'test@example.com')
  21. print 'find user:', u2

可以在MySQL客户端命令行查询，看看数据是不是正常存储到MySQL里面了。

原文: https://wizardforcel.gitbooks.io/liaoxuefeng/content/py2/98.html