added group functionality
This commit is contained in:
@@ -8,6 +8,7 @@ db = SQLAlchemy() # Create the extension object
|
||||
class User(UserMixin, db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
updated_at = db.Column(db.TIMESTAMP, default=datetime.now(), onupdate=datetime.now(), nullable=False)
|
||||
group_id = db.Column(db.Integer, default=0, nullable=False)
|
||||
name = db.Column(db.String(40), unique=True, nullable=False)
|
||||
password_hash = db.Column(db.String(256))
|
||||
is_admin = db.Column(db.Boolean, default=False)
|
||||
@@ -51,7 +52,12 @@ class User(UserMixin, db.Model):
|
||||
shared = Shared.query.filter_by(user_id=self.id, listofitem_id=listofitem_id).first()
|
||||
return True if shared else False
|
||||
|
||||
|
||||
def set_group_id(self, secret):
|
||||
group = Group.query.filter_by(secret=secret).first()
|
||||
if group:
|
||||
return group.id
|
||||
else:
|
||||
return 0
|
||||
|
||||
class ListOfItems(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
@@ -97,3 +103,9 @@ class Item(db.Model):
|
||||
class Shared(db.Model):
|
||||
user_id = db.Column(db.Integer, primary_key=True)
|
||||
listofitem_id = db.Column(db.Integer, primary_key=True)
|
||||
|
||||
class Group(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
updated_at = db.Column(db.TIMESTAMP, default=datetime.now(), onupdate=datetime.now(), nullable=False)
|
||||
secret = db.Column(db.String(128))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user