models.py

from fakk import db, bcrypt
#from flask_login import UserMixin
from sqlalchemy import ForeignKey, func, DateTime
from datetime import datetime, timedelta
from sqlalchemy.orm import relationship, backref

# create user 
#class User(db.Model, UserMixin):
class User(db.Model):

    __tablename__ = "users"

    userid = db.Column(db.Integer, primary_key=True)
    phone = db.Column(db.String)
    firstname = db.Column(db.String)
    lastname = db.Column(db.String)
    username = db.Column(db.String)
    email = db.Column(db.String)
    password = db.Column(db.String)
    #posts = relationship("BlogPost", backref="author")
    friend_requester = db.relationship('Relationship', foreign_keys='Relationship.userid')
    friend_receiver = db.relationship('Relationship', foreign_keys='Relationship.frienduserid')
    invoice_sender = db.relationship('Invoice', foreign_keys='Invoice.userid')
    invoice_receiver = db.relationship('Invoice', foreign_keys='Invoice.frienduserid')
    #db.UniqueConstraint(userid)
    #__table_args__ = (db.UniqueConstraint('userid'), )

    def __init__(self, username, password):
        self.username = username
        #self.email = email
        self.password = bcrypt.generate_password_hash(password).decode('utf-8')
        #self.password = password

    def is_authenticated(self):
        return True

    def is_active(self):
        return True

    def is_anonymous(self):
        return False

    def get_id(self):
        return self.userid

    def get_friends(self):
        result = []
        for i in self.friend_requester:
            if i.statusid == 3:
                result.append(i)
        #return self.friend_requester.query.filter_by(statusid=3).all()
        return result
        #return self.query.(friend_requester).filter(statusid==3).all()


    def is_friend(self, frienduserid):
        for i in self.friend_requester:
            if i.frienduserid == frienduserid and i.statusid == 3:
                return True
            else:
                return False

    def get_username(self):
        return self.username

    def update(self, username, email, password, phone):
        if(username):
            self.username = username
        if(email):
            self.email = email
        if(phone):
            self.phone = phone
        if(password):
            self.password = bcrypt.generate_password_hash(password).decode('utf-8')
        db.session.commit()

    def serialize(self):
        return {
        'userid' : self.userid,
        'username' : self.username,
        'email' : self.email
        }
    def delete(self):
        db.session.delete(self)
        db.session.commit()

   # def __repr__(self):
    #	return '<name - {}>'.format(self.username)

class Relationship(db.Model):

    __tablename__ = "friends"

    friendshipid = db.Column(db.Integer, primary_key=True)
    userid = db.Column(db.Integer, db.ForeignKey("users.userid"), nullable=False)
    frienduserid = db.Column(db.Integer, db.ForeignKey("users.userid"), nullable=False)
    statusid = db.Column(db.Integer)
    rel_sender = relationship('User', foreign_keys='Relationship.userid')
    rel_receiver = relationship('User', foreign_keys='Relationship.frienduserid')
    #__table_args__ = (db.UniqueConstraint('frienduserid', 'userid', name='_frienduserid_uc'), )
    #__table_args__ = (db.UniqueConstraint('receiving_user', 'requesting_user', name='_receiving_user_uc'), )
    def __init__(self, user, receiving_user, status):
        self.userid = user
        self.frienduserid = receiving_user
        self.statusid = status


    def get_id(self):
        return self.friendshipid

    def accept_req(self):
        self.statusid = 3
        db.session.commit()

    def delete(self):
        db.session.delete(self)
        db.session.commit()

    def __repr__(self):
        #return '<rel-friendshipid - {}>'.format(self.friendshipid)
        return 'Relationship(Id: %s, Sender: %s, Receiver: %s, Statusid: %s)' % (self.friendshipid, self.userid, self.frienduserid, self.statusid)

class Invoice(db.Model):

    __tablename__ = "invoices"

    invoiceid = db.Column(db.Integer, primary_key=True)
    amount = db.Column(db.Integer)
    description = db.Column(db.String)
    userid = db.Column(db.Integer, db.ForeignKey("users.userid"), nullable=False)
    frienduserid = db.Column(db.Integer, db.ForeignKey("users.userid"), nullable=False)
    statusid = db.Column(db.Integer)
    #invoice_date = db.Column(DateTime(timezone=True), server_default=func.now())
    date_created = db.Column(DateTime(timezone=True), server_default=func.now())
    date_updated = db.Column(DateTime(timezone=True))
    date_due = db.Column(DateTime(timezone=True))
    #duedate = db.Column(db.Date)
    message = db.Column(db.String)
    invoice_version = db.Column(db.Integer)
    sender = relationship('User', foreign_keys='Invoice.userid')
    receiver = relationship('User', foreign_keys='Invoice.frienduserid')
    #__table_args__ = (db.UniqueConstraint('frienduserid', 'userid', name='_frienduserid_uc'), )
    #__table_args__ = (db.UniqueConstraint('receiving_user', 'requesting_user', name='_receiving_user_uc'), )
    def __init__(self, userid, receiving_user, amount, description):
        self.userid = userid
        self.frienduserid = receiving_user
        self.statusid = 1
        self.amount = amount
        self.invoice_version = 1
        self.description = description
        self.date_due = func.now() + timedelta(days=7)

    def update(self, description, amount):
        if(amount):
            self.amount = amount
        if(description):
            self.description = description
        self.invoice_version += 1
        self.statusid = 1
        self.message = None
        self.date_updated = func.now()
        db.session.commit()

    def get_id(self):
        return self.invoiceid

    def get_amount(self):
        return self.amount

    def delete(self):
        db.session.delete(self)
        db.session.commit()

    def reject(self, message):
        self.statusid = 3
        self.message = message
        db.session.commit()

    def change_status(self, status):
        self.statusid = status
        db.session.commit()


    def __repr__(self):
        #return '<rel-invoiceid - {}>'.format(self.invoiceid)
        return 'Invoice(Id: %s, Receiver: %s, amount: %s, description: %s, sender: %s, receiver: %s)' % (self.invoiceid, self.frienduserid, self.amount, self.description, self.sender, self.receiver)