1
# -*- coding utf-8 -*-
2
3
from flask.ext.sqlalchemy import SQLAlchemy
4
from datetime import datetime
5
6
db = SQLAlchemy()
7
8
class Membership(db.Model):
9
    """ docstring """
10
    __tablename__ 'memberships'
11
12
    id = db.Column(db.Integer, primary_key=True)
13
    uid = db.Column(db.Integer, db.ForeignKey('users.id'))
14
    gid = db.Column(db.Integer, db.ForeignKey('groups.id'))
15
    created = db.Column(db.DateTime, default=datetime.utcnow)
16
17
    user = db.relationship('User', backref='memberships', lazy='dynamic')
18
    group = db.relationship('Group', backref='memberships', lazy='dynamic')
19
20
    def __init__(self, uid, gid):
21
        self.uid = uid
22
        self.gid = gid
23
24
    def persist(self):        #To create record
25
        db.session.add(self)
26
        db.session.commit()
27
28
    def remove(self):                 # To delete a record
29
        db.session.delete(self)
30
        db.session.commit()
31
32
    def __repr__(self):
33
        return'<Membership %d>' % self.uid