Add Speaker Model (#10)

app/database/models.py

@@ -1,9 +1,18 @@
+import enum
+from datetime import datetime
+
+from pytz import timezone
 from sqlalchemy import Text
 from sqlalchemy.dialects.mysql import TINYINT, INTEGER
 from sqlalchemy.orm import validates
 
 from app.utils.errors import InvalidArgumentError
 from app.database import db
+from app.utils.validators import is_valid_email, has_valid_length, \
+    is_valid_phone
+
+PASSWORD_LENGTH_MINIMUM = 4
+PASSWORD_LENGTH_MAXIMUM = 8
 
 
 class Library(db.Model):
@@ -41,3 +50,76 @@ def validate_not_empty(self, key, field):
 
     def __repr__(self):
         return '<Library %r>' % self.name
+
+
+class SessionEnum(enum.Enum):
+    first = "09:00"
+    second = "10:00"
+
+
+def now_at_seoul():
+    return datetime.now(tz=timezone('Asia/Seoul'))
+
+
+class TimestampMixin(object):
+    created_at = db.Column(db.DateTime,
+                           nullable=False,
+                           default=now_at_seoul)
+    updated_at = db.Column(db.DateTime,
+                           nullable=True,
+                           onupdate=now_at_seoul)
+
+
+class Speaker(db.Model, TimestampMixin):
+    _id = db.Column('id', db.Integer, primary_key=True,
+                    autoincrement=True)
+    name = db.Column(db.String(30), nullable=False)
+    email = db.Column(db.String(30), nullable=False, unique=True)
+    phone = db.Column(db.String(30), nullable=False)
+
+    password = db.Column(db.String(30), nullable=False)
+    is_email_verified = db.Column(db.Boolean, default=False)
+    email_sended_at = db.Column(db.DateTime, nullable=True)
+    session_time = db.Column(db.Enum(SessionEnum), nullable=False)
+    library_id = db.Column(INTEGER(11, unsigned=True),
+                           db.ForeignKey('Libraries.id'),
+                           nullable=False)
+    library = db.relationship('Library', lazy=True,
+                              backref=db.backref('speakers', lazy=True))
+
+    @validates('name')
+    def validate_not_empty(self, key, field):
+        if not field:
+            raise InvalidArgumentError("{} must be not empty.".format(key))
+
+        return field
+
+    @validates('email')
+    def validate_email_format(self, key, field):
+        if not is_valid_email(field):
+            raise InvalidArgumentError(
+                "{} must be fitted in email format.".format(key))
+
+        return field
+
+    @validates('password')
+    def validate_password_length(self, key, field):
+        if not has_valid_length(field,
+                                PASSWORD_LENGTH_MINIMUM,
+                                PASSWORD_LENGTH_MAXIMUM):
+            raise InvalidArgumentError(
+                "{}'s length must be {0} ~ {0}.".format(
+                    key, PASSWORD_LENGTH_MINIMUM, PASSWORD_LENGTH_MAXIMUM))
+
+        return field
+
+    @validates('phone')
+    def validate_phone_format(self, key, field):
+        if not is_valid_phone(field):
+            raise InvalidArgumentError(
+                "{} must be fitted in phone number format.".format(key))
+
+        return field
+
+    def __repr__(self):
+        return '<%r %r>' % (self.__class__.__name__, self.name)

app/utils/validators.py

@@ -0,0 +1,10 @@
+def is_valid_email(field):
+    return True
+
+
+def has_valid_length(field, length_min, length_max):
+    return True
+
+
+def is_valid_phone(field):
+    return True

migrations/versions/6c0fbb516dfd_.py

@@ -0,0 +1,51 @@
+"""empty message
+
+Revision ID: 6c0fbb516dfd
+Revises: ad3c95a8a551
+Create Date: 2018-04-11 23:34:31.356913
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy.dialects import mysql
+
+# revision identifiers, used by Alembic.
+revision = '6c0fbb516dfd'
+down_revision = 'ad3c95a8a551'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_table('speaker',
+    sa.Column('created_at', sa.DateTime(), nullable=False),
+    sa.Column('updated_at', sa.DateTime(), nullable=True),
+    sa.Column('id', sa.Integer(), nullable=False),
+    sa.Column('name', sa.String(length=30), nullable=False),
+    sa.Column('email', sa.String(length=30), nullable=False),
+    sa.Column('phone', sa.String(length=30), nullable=False),
+    sa.Column('password', sa.String(length=30), nullable=False),
+    sa.Column('is_email_verified', sa.Boolean(), nullable=True),
+    sa.Column('email_sended_at', sa.DateTime(), nullable=True),
+    sa.Column('session_time', sa.Enum('first', 'second', name='sessionenum'), nullable=False),
+    sa.Column('library_id', mysql.INTEGER(display_width=11, unsigned=True), nullable=False),
+    sa.ForeignKeyConstraint(['library_id'], ['Libraries.id'], ),
+    sa.PrimaryKeyConstraint('id'),
+    sa.UniqueConstraint('email')
+    )
+    op.alter_column('Libraries', 'name',
+               existing_type=mysql.VARCHAR(length=20),
+               nullable=True)
+    op.drop_index('name', table_name='Libraries')
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_index('name', 'Libraries', ['name'], unique=True)
+    op.alter_column('Libraries', 'name',
+               existing_type=mysql.VARCHAR(length=20),
+               nullable=False)
+    op.drop_table('speaker')
+    # ### end Alembic commands ###