Merge pull request #13 from rday/master

Mixer/Faker examples

Author: rday

Bizness/models.py

@@ -21,6 +21,10 @@ class Task(db.Model):
     description = db.Column(db.Text)
     due_date = db.Column(db.TIMESTAMP)
 
+    @staticmethod
+    def paginate(page, page_size):
+        return Task.query.paginate(page, page_size)
+
     @staticmethod
     def from_request(request):
         parser = RequestParser()
@@ -227,3 +231,20 @@ def to_dict(self):
 
     def __repr__(self):
         return u"Item {} <#{}> due {}".format(self.name, self.id, self.due_date.isoformat())
+
+
+class User(db.Model):
+    __tablename__ = 'users'
+
+    id = db.Column(db.Integer(), primary_key=True)
+    first_name = db.Column(db.String(64), nullable=False)
+    last_name = db.Column(db.String(64), nullable=False)
+    email_address = db.Column(db.String(64), nullable=False)
+    password = db.Column(db.String(64), nullable=False)
+    phone_number = db.Column(db.String(64), nullable=False)
+    address = db.Column(db.String(64))
+    city = db.Column(db.String(64))
+    street = db.Column(db.String(64))
+    state = db.Column(db.String(64))
+    title = db.Column(db.String(64), nullable=False)
+    active = db.Column(db.Boolean, nullable=False)

BiznessWeb/__init__.py

@@ -10,8 +10,9 @@
 from flask.ext.admin.contrib.sqla import ModelView
 from flask.ext.login import LoginManager
 from .items import items_bp
+from .tasks import tasks_bp
 from Bizness import db
-from Bizness.models import Item
+from Bizness.models import Item, Task
 
 
 app = Flask(__name__)
@@ -23,6 +24,7 @@
 # Setup admin
 admin = Admin(app)
 admin.add_view(ModelView(Item, db.session))
+admin.add_view(ModelView(Task, db.session))
 
 # Setup the login manager
 # For authentication
@@ -33,3 +35,4 @@
 
 # Setup our route blueprints
 app.register_blueprint(items_bp)
+app.register_blueprint(tasks_bp)

BiznessWeb/tasks.py

@@ -0,0 +1,78 @@
+from flask import Blueprint, jsonify, request, abort
+from werkzeug.exceptions import BadRequest
+from Bizness import db
+from Bizness.models import Task
+from .api import BizResponse
+from config import PAGE_SIZE
+import iso8601
+
+
+tasks_bp = Blueprint('tasks_bp', __name__, url_prefix='/tasks')
+
+
+@tasks_bp.route('/')
+def index():
+    """
+    request.args inherits from TypeConversionDict !
+    """
+    page_size = request.args.get('page_size', PAGE_SIZE, type=int)
+    page = request.args.get('page', 1, type=int)
+    last_date = request.args.get('last_date', None, type=lambda d: iso8601.parse_date(d))
+
+    if last_date is None:
+        tasks_pagination = Task.paginate(page, page_size)
+    else:
+        tasks_pagination = Task.paginate_by_due_date(last_date, page_size)
+
+    tasks = {
+        'tasks': [task.to_dict() for task in tasks_pagination.items]
+    }
+
+    response = BizResponse(
+        success=True,
+        data=tasks,
+        total=tasks_pagination.total,
+        page=tasks_pagination.page
+    )
+
+    return jsonify(response.to_dict())
+
+
+@tasks_bp.route('/', methods=['POST'])
+def create_task():
+    try:
+        new_task = Task.from_request(request)
+        db.session.add(new_task)
+        db.session.commit()
+
+        response = BizResponse(
+            success=True,
+            data=new_task.to_dict(),
+            total=1
+        )
+    except BadRequest as e:
+        response = BizResponse(
+            success=False,
+            data=e.description
+        )
+    except Exception:
+        response = BizResponse(
+            success=False,
+            data='An error occurred saving the task'
+        )
+
+    return jsonify(response.to_dict())
+
+
+@tasks_bp.route('/<int:task_id>', methods=['GET'])
+def get_task(task_id):
+    task = Task.query.filter(Task.id == task_id).first()
+    if task is None:
+        abort(404)
+
+    response = BizResponse(
+        success=True,
+        data=task.to_dict()
+    )
+
+    return jsonify(response.to_dict())

tools/factory.py

@@ -0,0 +1,26 @@
+from random import randint
+from Bizness import models
+from BiznessWeb import app, db
+from mixer.backend.flask import Mixer
+from mixer.backend.sqlalchemy import GenFactory
+
+
+def fake_active():
+    return randint(0, 10) > 2
+
+
+class MyFactory(GenFactory):
+    fakers = GenFactory.fakers
+    fakers[('active', bool)] = fake_active
+
+
+mixer = Mixer(locale='en_US', commit=False, factory=MyFactory)
+
+with app.app_context():
+    db.drop_all()
+    db.create_all()
+
+    mixer.init_app(app)
+
+    for user in mixer.cycle(20).blend(models.User):
+        print("{} {}: {}".format(user.first_name, user.last_name, user.active))

tools/middleware.py

@@ -0,0 +1,25 @@
+from Bizness import models
+from BiznessWeb import app, db
+from mixer.backend.flask import Mixer
+from flask.ext.bcrypt import Bcrypt
+
+
+mixer = Mixer(locale='en_US', commit=False)
+bcrypt = Bcrypt(app)
+
+
+@mixer.middleware(models.User)
+def encrypt_password(user):
+    user.password = bcrypt.generate_password_hash('def4ult')
+    return user
+
+
+with app.app_context():
+    db.drop_all()
+    db.create_all()
+
+    mixer.init_app(app)
+    mixer.register(models.User, email_address=mixer.faker.email_address)
+
+    for user in mixer.cycle(20).blend(models.User):
+        print(u"{} {}: {}({})".format(user.first_name, user.last_name, user.email_address, user.password))

tools/populate_db.py

@@ -3,18 +3,24 @@
 from Bizness import models
 from BiznessWeb import app, db
 from mixer.backend.flask import mixer
+from mixer.main import TypeMixer
+
+@mixer.middleware(models.Task)
+def encrypt_password(task):
+    task.due_date = datetime.datetime.utcnow() + datetime.timedelta(days=randint(5, 10))
+    return task
 
 
 with app.app_context():
     db.drop_all()
     db.create_all()
     mixer.init_app(app)
+    mixer.register(models.Task, description=lambda: 'LALAAL')
 
     for task in mixer.cycle(10).blend(models.Task,
                                       name=mixer.faker.name,
-                                      description=mixer.faker.text):
-        task.due_date = datetime.datetime.utcnow() + datetime.timedelta(days=randint(5, 10))
-        db.session.add(task)
+                                      description=mixer.FAKE):
+        print(task.due_date)
 
     db.session.flush()
 

tools/providers.py

@@ -0,0 +1,60 @@
+from Bizness import models
+from BiznessWeb import app, db
+from mixer.backend.flask import Mixer
+from faker.providers.phone_number import Provider as BaseProvider
+
+
+class FixedPhoneNumberProvider(BaseProvider):
+    formats = (
+        '###-###-####',
+        '(###)###-####',
+        '1-###-###-####',
+        '###.###.####',
+        '###-###-####x###',
+        '(###)###-####x###',
+        '1-###-###-####x###',
+        '###.###.####x###',
+        '###-###-####x####',
+        '(###)###-####x####',
+        '1-###-###-####x####',
+        '###.###.####x####',
+        '###-###-####x#####',
+        '(###)###-####x#####',
+        '1-###-###-####x#####',
+        '###.###.####x#####'
+    )
+
+
+class JobTitleProvider(BaseProvider):
+    formats = ['{{experience_level}} {{job_type}}', ]
+
+    job_types = ['Developer', 'QA Analyst', 'UX Designer']
+
+    experience_levels = ['', 'Senior', 'Junior', 'Intern']
+
+    def title(self):
+        pattern = self.random_element(self.formats)
+        return self.generator.parse(pattern)
+
+    @classmethod
+    def job_type(cls):
+        return cls.random_element(cls.job_types)
+
+    @classmethod
+    def experience_level(cls):
+        return cls.random_element(cls.experience_levels)
+
+
+mixer = Mixer(locale='en_US', commit=False)
+
+with app.app_context():
+    db.drop_all()
+    db.create_all()
+
+    mixer.init_app(app)
+    mixer.faker.add_provider(JobTitleProvider)
+    mixer.faker.add_provider(FixedPhoneNumberProvider)
+    mixer.register(models.User, phone_number=mixer.faker.phone_number)
+
+    for user in mixer.cycle(20).blend(models.User):
+        print("{} {}: {} {}".format(user.first_name, user.last_name, user.title, user.phone_number))