generated from justuser-31/mrl_v1_license
Разделение на файлы, мелкие доработки.
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
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
|
||||
Reference in New Issue
Block a user