74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
from pydantic import BaseModel
|
|
from datetime import datetime, timedelta
|
|
from typing import Optional
|
|
|
|
class User(BaseModel):
|
|
username: str
|
|
status: str
|
|
quota: str
|
|
traffic_used: str
|
|
expiry_date: str
|
|
expiry_days: str
|
|
enable: bool
|
|
|
|
@staticmethod
|
|
def from_dict(username: str, user_data: dict):
|
|
user_data = {'username': username, **user_data}
|
|
user_data = User.__parse_user_data(user_data)
|
|
return User(**user_data)
|
|
|
|
@staticmethod
|
|
def __parse_user_data(user_data: dict) -> dict:
|
|
expiration_days = user_data.get('expiration_days', 0)
|
|
|
|
if expiration_days > 0:
|
|
creation_date_str = user_data.get("account_creation_date")
|
|
display_expiry_days = str(expiration_days)
|
|
|
|
if isinstance(creation_date_str, str):
|
|
try:
|
|
creation_date = datetime.strptime(creation_date_str, "%Y-%m-%d")
|
|
expiry_dt_obj = creation_date + timedelta(days=expiration_days)
|
|
display_expiry_date = expiry_dt_obj.strftime("%Y-%m-%d")
|
|
except ValueError:
|
|
display_expiry_date = "Error"
|
|
else:
|
|
display_expiry_date = "Error"
|
|
else:
|
|
display_expiry_days = "Unlimited"
|
|
display_expiry_date = "Unlimited"
|
|
|
|
used_bytes = user_data.get("download_bytes", 0) + user_data.get("upload_bytes", 0)
|
|
quota_bytes = user_data.get('max_download_bytes', 0)
|
|
|
|
used_formatted = User.__format_traffic(used_bytes)
|
|
quota_formatted = "Unlimited" if quota_bytes == 0 else User.__format_traffic(quota_bytes)
|
|
|
|
percentage = 0
|
|
if quota_bytes > 0:
|
|
percentage = (used_bytes / quota_bytes) * 100
|
|
|
|
traffic_used_display = f"{used_formatted}/{quota_formatted} ({percentage:.1f}%)"
|
|
|
|
return {
|
|
'username': user_data['username'],
|
|
'status': user_data.get('status', 'Not Active'),
|
|
'quota': quota_formatted,
|
|
'traffic_used': traffic_used_display,
|
|
'expiry_date': display_expiry_date,
|
|
'expiry_days': display_expiry_days,
|
|
'enable': not user_data.get('blocked', False),
|
|
}
|
|
|
|
@staticmethod
|
|
def __format_traffic(traffic_bytes) -> str:
|
|
if traffic_bytes == 0:
|
|
return "0 B"
|
|
if traffic_bytes < 1024:
|
|
return f'{traffic_bytes} B'
|
|
elif traffic_bytes < 1024**2:
|
|
return f'{traffic_bytes / 1024:.2f} KB'
|
|
elif traffic_bytes < 1024**3:
|
|
return f'{traffic_bytes / 1024**2:.2f} MB'
|
|
else:
|
|
return f'{traffic_bytes / 1024**3:.2f} GB' |