added group functionality

This commit is contained in:
2026-05-14 14:13:41 +02:00
parent 23631c6d24
commit a8554905ce
3 changed files with 56 additions and 21 deletions
+13 -1
View File
@@ -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))