#!/usr/bin/env python
# encoding: utf-8

from factory import Factory

from ec2stack.models import *


def create_sql_model_function(class_to_create, *args, **kwargs):
    entity = class_to_create(**kwargs)
    DB.session.add(entity)
    DB.session.commit()
    return entity


Factory.set_creation_function(create_sql_model_function)


class UserFactory(Factory):
    FACTORY_FOR = User
    apikey = "ExampleAPIKey"
    secretkey = "ExampleSecretKey"