generated from justuser-31/mrl_v1_license
31 lines
799 B
Python
31 lines
799 B
Python
import os
|
|
|
|
from config import get_storage_root
|
|
from database import get_user_quota
|
|
|
|
|
|
def get_user_files(username):
|
|
user_dir = os.path.join(get_storage_root(), username)
|
|
if not os.path.exists(user_dir):
|
|
return []
|
|
files = []
|
|
for fname in os.listdir(user_dir):
|
|
fpath = os.path.join(user_dir, fname)
|
|
if os.path.isfile(fpath):
|
|
files.append(fname)
|
|
return files
|
|
|
|
|
|
def can_upload(username, file_size_bytes):
|
|
quota_mb, used_mb = get_user_quota(username)
|
|
if quota_mb is None:
|
|
return False
|
|
file_size_mb = file_size_bytes / (1024 * 1024)
|
|
return (used_mb + file_size_mb) <= quota_mb
|
|
|
|
|
|
def get_user_dir(username):
|
|
user_dir = os.path.join(get_storage_root(), username)
|
|
os.makedirs(user_dir, exist_ok=True)
|
|
return user_dir
|