users.py

import json
from datetime import datetime
from dataclasses import dataclass
from enum import Enum
from typing import List
from warnings import warn

from .config import USERS_DATASET_PATH


@dataclass
class User:

    class Status(str, Enum):
        NEW = 'NEW'
        PAID = "PAID"
        TRIAL = "TRIAL"

    id: int
    enabled: bool
    status: Status
    custom_properties: dict
    first_name: str
    last_name: str
    address: str
    company: str
    credit_card: str
    currency: str
    created_at: datetime
    bio: str
    budget: float
    labels: List[str]


if USERS_DATASET_PATH.exists():
    _cached_users: List[User] = [
        User(**row)
        for row
        in json.load(USERS_DATASET_PATH.open())
    ]
else:
    warn("Missing test users dataset.")


async def get_users() -> List[User]:
    return _cached_users


async def get_user() -> User:
    return _cached_users[0]