# -*- coding utf-8 -*-

from flask.ext.sqlalchemy import SQLAlchemy

from datetime import datetime

db = SQLAlchemy()

class Membership(db.Model):

    id = db.Column(db.Integer, primary_key=True)

    uid = db.Column(db.Integer, db.ForeignKey('user.id'))

    gid = db.Column(db.Integer, db.ForeignKey('group.id'))

    #timecreated = db.Column(db.DateTime)

    def __init__(self, uid, gid):

        self.uid = uid

        self.gid = gid

        #self.timecreated = timecreated

    def createrow(self, tocreate):        #To create record

        self.tocreate = tocreate

        db.session.add(tocreate)

        db.session.commit()

    def delrow(self, id):                 # To delete a record

        todel = User.query.get(id)

        db.session.delete(todel)

        db.session.commit()

    def __repr__(self):

        return'<Membership %d>' % self.uid