forked from notahat/machinist
-
Notifications
You must be signed in to change notification settings - Fork 0
/
sequel_environment.rb
73 lines (58 loc) · 1.25 KB
/
sequel_environment.rb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
require 'sequel'
require 'sequel/extensions/migration'
require 'machinist/sequel'
DB = Sequel.mysql(:database => 'machinist')
Sequel.migration do
change do
create_table! :users do
primary_key :id
String :username
end
create_table! :posts do
primary_key :id
String :title
Integer :author_id
String :body, :text => true
end
create_table! :comments do
primary_key :id
Integer :post_id
String :body, :text => true
end
create_table! :tags do
primary_key :id
String :name
end
create_table! :posts_tags do
Integer :post_id
Integer :tag_id
end
end
end
module SequelEnvironment
class User < Sequel::Model
plugin :validation_helpers
def validate
super
validates_presence :username
validates_unique :username
end
end
class Post < Sequel::Model
one_to_many :comments
many_to_one :author, :class_name => "SequelEnvironment::User"
many_to_many :tags
end
class Comment < Sequel::Model
many_to_one :post
end
class Tag < Sequel::Model
many_to_many :posts
end
def empty_database!
[User, Post, Comment].each do |klass|
klass.delete
klass.clear_blueprints!
end
end
end