Ali2206 commited on
Commit
2ba25dc
·
verified ·
1 Parent(s): 004efaa

Update models/schemas.py

Browse files
Files changed (1) hide show
  1. models/schemas.py +12 -1
models/schemas.py CHANGED
@@ -1,20 +1,31 @@
1
  from pydantic import BaseModel, EmailStr
2
- from typing import Optional, List
3
  from datetime import date
4
 
 
5
  class SignupForm(BaseModel):
6
  email: EmailStr
7
  password: str
8
  full_name: str
 
9
 
10
  class TokenResponse(BaseModel):
11
  access_token: str
12
  token_type: str
13
 
 
 
 
 
 
 
 
 
14
  class ContactInfo(BaseModel):
15
  email: Optional[EmailStr] = None
16
  phone: Optional[str] = None
17
 
 
18
  class PatientCreate(BaseModel):
19
  full_name: str
20
  date_of_birth: date
 
1
  from pydantic import BaseModel, EmailStr
2
+ from typing import Optional, List, Literal
3
  from datetime import date
4
 
5
+ # --- USER SCHEMAS ---
6
  class SignupForm(BaseModel):
7
  email: EmailStr
8
  password: str
9
  full_name: str
10
+ role: Literal['admin', 'doctor', 'patient']
11
 
12
  class TokenResponse(BaseModel):
13
  access_token: str
14
  token_type: str
15
 
16
+ # --- DOCTOR CREATION SCHEMA (Admin use only) ---
17
+ class DoctorCreate(BaseModel):
18
+ matricule: str
19
+ email: EmailStr
20
+ password: str
21
+ full_name: str
22
+
23
+ # --- CONTACT INFO ---
24
  class ContactInfo(BaseModel):
25
  email: Optional[EmailStr] = None
26
  phone: Optional[str] = None
27
 
28
+ # --- PATIENT SCHEMA ---
29
  class PatientCreate(BaseModel):
30
  full_name: str
31
  date_of_birth: date