Spaces:
Sleeping
Sleeping
from typing import List, Dict, Optional, Set, Tuple | |
from pydantic import BaseModel | |
class User(BaseModel): | |
username: str | |
email: Optional[str] = None | |
youtube_credentials: Optional[Dict] = None | |
class UserDatabase: | |
""" | |
In-memory user database. In a production app, | |
replace with a proper database like SQLAlchemy | |
""" | |
users = {} | |
def create_user(cls, username: str, credentials: Dict): | |
user = User(username=username, youtube_credentials=credentials) | |
cls.users[username] = user | |
return user | |
def get_user(cls, username: str): | |
return cls.users.get(username) |