@bp.route('/add/organisation', methods=['GET', 'POST'])
@UserPermission()
def add_organisation():
    """Add organisation"""

    db.create_all()
    form = AddOrganisationForm(request.form)
    if request.method == 'POST':
        if form.validate():
            contact = Organisation(**form.data, user=g.user)
            db.session.add(contact)
            db.session.commit()
            return redirect(url_for('site.index'))
    return render_template('account/add/add_organisation.html', form=form)

# coding: utf-8
from datetime import datetime
from werkzeug.security import generate_password_hash, check_password_hash
from ._base import db
from .base import Base
from sqlalchemy.orm import relationship

class User(Base):

    name = db.Column(db.String(50), unique=True)
    email = db.Column(db.String(50), unique=True)
    avatar = db.Column(db.String(200), default='default.png')
    password = db.Column(db.String(200))
    is_admin = db.Column(db.Boolean, default=False)
    organisations = db.relationship('Organisation')


    def __setattr__(self, name, value):
        if name == 'password':
            value = generate_password_hash(value)
        super(User, self).__setattr__(name, value)

    def check_password(self, password):
        return check_password_hash(self.password, password)

    def __repr__(self):
        return '<User %s>' % self.name

# coding: utf-8
from datetime import datetime
from ._base import db
from .base import Base
from sqlalchemy import ForeignKey

class Organisation(Base):

    name = db.Column(db.String(50), unique=True)
    created_by = db.Column(db.Integer, db.ForeignKey('user.id'))

    @staticmethod
    def create(**kwargs):
        o = Organisation(**kwargs)
        db.session.add(o)
        try:
            db.session.commit()
        except IntegrityError:
            db.session.rollback()
        return o

    def __repr__(self):
        return '<Organisation %s>' % self.name