File size: 3,364 Bytes
e1b5e25 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 | # Models MongoDB pour la gestion des fichiers et des accès
from mongoengine import Document, fields, signals
from datetime import datetime
import hashlib
import os
class FileAccessLog(Document):
"""Journal d'accès aux fichiers"""
meta = {'collection': 'file_access_logs'}
file_id = fields.ObjectIdField(required=True)
user_id = fields.StringField(required=True)
access_type = fields.StringField(choices=['view', 'download', 'upload', 'delete'])
ip_address = fields.StringField()
user_agent = fields.StringField()
timestamp = fields.DateTimeField(default=datetime.now)
success = fields.BooleanField(default=True)
class FileMetadata(Document):
"""Métadonnées des fichiers stockés"""
meta = {'collection': 'file_metadata'}
# Informations de base
filename = fields.StringField(required=True, max_length=255)
original_filename = fields.StringField(required=True, max_length=255)
file_path = fields.StringField(required=True)
file_size = fields.IntField(required=True)
file_hash = fields.StringField(required=True, unique=True)
mime_type = fields.StringField()
# Catégorisation
category = fields.StringField(choices=[
'document', 'image', 'video', 'audio', 'archive', 'other'
])
tags = fields.ListField(fields.StringField())
# Propriétaire et permissions
owner_id = fields.StringField(required=True)
owner_name = fields.StringField()
is_public = fields.BooleanField(default=False)
allowed_users = fields.ListField(fields.StringField()) # IDs des utilisateurs autorisés
# Timestamps
created_at = fields.DateTimeField(default=datetime.now)
updated_at = fields.DateTimeField(default=datetime.now)
last_accessed = fields.DateTimeField()
# Statistiques
download_count = fields.IntField(default=0)
view_count = fields.IntField(default=0)
def calculate_hash(self):
"""Calculer le hash SHA256 du fichier"""
hash_sha256 = hashlib.sha256()
try:
with open(self.file_path, "rb") as f:
for chunk in iter(lambda: f.read(4096), b""):
hash_sha256.update(chunk)
return hash_sha256.hexdigest()
except:
return None
class UserPermission(Document):
"""Gestion des permissions utilisateur"""
meta = {'collection': 'user_permissions'}
user_id = fields.StringField(required=True, unique=True)
username = fields.StringField(required=True)
email = fields.EmailField()
# Permissions globales
is_admin = fields.BooleanField(default=False)
can_upload = fields.BooleanField(default=True)
can_download = fields.BooleanField(default=True)
can_delete_own = fields.BooleanField(default=True)
can_delete_others = fields.BooleanField(default=False)
# Limites de stockage (en octets)
storage_limit = fields.IntField(default=1073741824) # 1GB par défaut
current_storage_used = fields.IntField(default=0)
# Statuts
is_active = fields.BooleanField(default=True)
last_login = fields.DateTimeField()
created_at = fields.DateTimeField(default=datetime.now)
# Signaux pour la mise à jour automatique
signals.pre_save.connect(lambda sender, document: setattr(document, 'updated_at', datetime.now()), sender=FileMetadata) |