websites/Flask-Files
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

working folder creation

Browse source
This commit is contained in:
pika 2025-03-23 01:31:21 +01:00
parent ea3e92b8b7
commit b9a82af12f
11 changed files with 2791 additions and 1552 deletions
Download patch file Download diff file Expand all files Collapse all files

49
app/models.py
View file

@ -21,6 +21,10 @@ class User(UserMixin, db.Model):
# Relationships
files = db.relationship('File', backref='owner', lazy='dynamic',
foreign_keys='File.user_id', cascade='all, delete-orphan')
shares = db.relationship('Share', backref='owner', lazy='dynamic',
foreign_keys='Share.user_id', cascade='all, delete-orphan')
downloads = db.relationship('Download', backref='user', lazy='dynamic',
foreign_keys='Download.user_id', cascade='all, delete-orphan')
def __repr__(self):
return f'<User {self.username}>'
@ -48,6 +52,8 @@ class File(db.Model):
# Relationships
children = db.relationship('File', backref=db.backref('parent', remote_side=[id]),
lazy='dynamic', cascade='all, delete-orphan')
shares = db.relationship('Share', backref='file', lazy='dynamic', cascade='all, delete-orphan')
downloads = db.relationship('Download', backref='file', lazy='dynamic', cascade='all, delete-orphan')
def __repr__(self):
return f'<File {self.name} {"(Folder)" if self.is_folder else ""}>'
@ -62,17 +68,42 @@ class File(db.Model):
return f"{uuid.uuid4().hex}.{ext}" if ext else f"{uuid.uuid4().hex}"
class Share(db.Model):
__tablename__ = 'share'
id = db.Column(db.Integer, primary_key=True)
file_id = db.Column(db.Integer, db.ForeignKey('file.id'))
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
token = db.Column(db.String(64), unique=True)
file_id = db.Column(db.Integer, db.ForeignKey('file.id', ondelete='CASCADE'), nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
share_token = db.Column(db.String(64), unique=True)
is_public = db.Column(db.Boolean, default=False)
is_password_protected = db.Column(db.Boolean, default=False)
password_hash = db.Column(db.String(128))
expiry_date = db.Column(db.DateTime, nullable=True)
created_at = db.Column(db.DateTime, default=datetime.utcnow)
expires_at = db.Column(db.DateTime, nullable=True)
downloads = db.relationship('Download', backref='share', lazy='dynamic')
def __repr__(self):
return f'<Share {self.share_token}>'
def set_password(self, password):
self.password_hash = generate_password_hash(password)
self.is_password_protected = True
def check_password(self, password):
return check_password_hash(self.password_hash, password)
def generate_token(self):
"""Generate a unique token for sharing"""
return uuid.uuid4().hex
class Download(db.Model):
__tablename__ = 'download'
id = db.Column(db.Integer, primary_key=True)
file_id = db.Column(db.Integer, db.ForeignKey('file.id'))
share_id = db.Column(db.Integer, db.ForeignKey('share.id'))
ip_address = db.Column(db.String(45))
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
file_id = db.Column(db.Integer, db.ForeignKey('file.id', ondelete='CASCADE'), nullable=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=True) # Nullable for anonymous downloads
ip_address = db.Column(db.String(64))
user_agent = db.Column(db.String(255))
share_id = db.Column(db.Integer, db.ForeignKey('share.id', ondelete='SET NULL'), nullable=True)
downloaded_at = db.Column(db.DateTime, default=datetime.utcnow)
def __repr__(self):
return f'<Download of File {self.file_id} at {self.downloaded_at}>'