File size: 2,185 Bytes
c1a7abf | 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 | """
File handling utilities for SkillSync
"""
import os
import shutil
from datetime import datetime
class FileHandler:
"""Handles file operations for submissions"""
def __init__(self, upload_dir="data/user_data/submissions", max_size_mb=100):
self.upload_dir = upload_dir
self.max_size = max_size_mb * 1024 * 1024
self.allowed_extensions = [".zip", ".pdf", ".ipynb", ".py", ".txt", ".md"]
def ensure_upload_dir(self):
"""Ensure upload directory exists"""
os.makedirs(self.upload_dir, exist_ok=True)
def validate_file(self, file):
"""Validate uploaded file"""
if file is None:
return False, "No file provided"
file_name = getattr(file, 'name', 'unknown')
file_ext = os.path.splitext(file_name)[1].lower()
if file_ext not in self.allowed_extensions:
return False, f"File type {file_ext} not allowed"
return True, "File validated successfully"
def save_file(self, file, user_id, assignment_type):
"""Save uploaded file"""
try:
self.ensure_upload_dir()
original_name = getattr(file, 'name', 'uploaded_file')
original_name = os.path.basename(original_name)
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
new_name = f"{user_id}_{assignment_type}_{timestamp}_{original_name}"
destination = os.path.join(self.upload_dir, new_name)
if hasattr(file, 'name'):
shutil.copy2(file.name, destination)
return True, destination
except Exception as e:
return False, str(e)
def get_file_info(self, filepath):
"""Get information about a file"""
if not os.path.exists(filepath):
return None
stat = os.stat(filepath)
return {
"path": filepath,
"size": stat.st_size,
"created": datetime.fromtimestamp(stat.st_ctime).strftime("%Y-%m-%d %H:%M:%S"),
"modified": datetime.fromtimestamp(stat.st_mtime).strftime("%Y-%m-%d %H:%M:%S")
}
|