blob: b04ac50b2d69e7c19ae8de9b952f9bed64faea3e (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
from datetime import datetime
from typing import Optional
from uuid import UUID
from pydantic import BaseModel, Field
from shared.abstractions import R2RSerializable
from ..utils import generate_default_user_collection_id
class Collection(BaseModel):
id: UUID
name: str
description: Optional[str] = None
created_at: datetime = Field(
default_factory=datetime.utcnow,
)
updated_at: datetime = Field(
default_factory=datetime.utcnow,
)
class Config:
populate_by_name = True
from_attributes = True
def __init__(self, **data):
super().__init__(**data)
if self.id is None:
self.id = generate_default_user_collection_id(self.name)
class Token(BaseModel):
token: str
token_type: str
class TokenData(BaseModel):
email: str
token_type: str
exp: datetime
class User(R2RSerializable):
id: UUID
email: str
is_active: bool = True
is_superuser: bool = False
created_at: datetime = datetime.now()
updated_at: datetime = datetime.now()
is_verified: bool = False
collection_ids: list[UUID] = []
graph_ids: list[UUID] = []
document_ids: list[UUID] = []
# Optional fields (to update or set at creation)
limits_overrides: Optional[dict] = None
metadata: Optional[dict] = None
verification_code_expiry: Optional[datetime] = None
name: Optional[str] = None
bio: Optional[str] = None
profile_picture: Optional[str] = None
total_size_in_bytes: Optional[int] = None
num_files: Optional[int] = None
account_type: str = "password"
hashed_password: Optional[str] = None
google_id: Optional[str] = None
github_id: Optional[str] = None
|