|
9 | 9 | PhoneNumber, |
10 | 10 | SATRegimeCode, |
11 | 11 | TOSAgreement, |
12 | | - TOSRequest, |
13 | 12 | UserQuery, |
14 | 13 | UserRequest, |
15 | 14 | UserStatus, |
16 | 15 | UserUpdateRequest, |
17 | 16 | ) |
18 | 17 | from cuenca_validations.types.enums import Country, Gender, State |
19 | 18 | from cuenca_validations.types.general import SerializableHttpUrl |
20 | | -from cuenca_validations.types.identities import Curp |
| 19 | +from cuenca_validations.types.identities import AddressRequest, Curp |
21 | 20 | from pydantic import ConfigDict, EmailStr, Field |
22 | 21 |
|
23 | 22 | from ..http import Session, session as global_session |
@@ -107,70 +106,54 @@ def balance(self) -> int: |
107 | 106 | def create( |
108 | 107 | cls, |
109 | 108 | curp: Curp, |
110 | | - phone_number: Optional[PhoneNumber] = None, |
111 | | - email_address: Optional[EmailStr] = None, |
112 | 109 | profession: Optional[str] = None, |
113 | | - address: Optional[Address] = None, |
| 110 | + address: Optional[AddressRequest] = None, |
114 | 111 | email_verification_id: Optional[str] = None, |
115 | 112 | phone_verification_id: Optional[str] = None, |
116 | | - status: Optional[UserStatus] = None, |
117 | | - required_level: Optional[int] = None, |
118 | | - terms_of_service: Optional[TOSRequest] = None, |
119 | 113 | *, |
120 | 114 | session: Session = global_session, |
121 | 115 | ) -> 'User': |
122 | 116 | req = UserRequest( |
123 | 117 | curp=curp, |
124 | | - phone_number=phone_number, |
125 | | - email_address=email_address, |
126 | 118 | profession=profession, |
127 | 119 | address=address, |
128 | 120 | email_verification_id=email_verification_id, |
129 | 121 | phone_verification_id=phone_verification_id, |
130 | | - required_level=required_level, |
131 | | - status=status, |
132 | | - terms_of_service=terms_of_service, |
133 | 122 | ) |
134 | 123 | return cls._create(session=session, **req.model_dump()) |
135 | 124 |
|
136 | 125 | @classmethod |
137 | 126 | def update( |
138 | 127 | cls, |
139 | 128 | user_id: str, |
140 | | - phone_number: Optional[PhoneNumber] = None, |
141 | | - email_address: Optional[str] = None, |
142 | 129 | profession: Optional[str] = None, |
143 | | - address: Optional[Address] = None, |
| 130 | + address: Optional[AddressRequest] = None, |
144 | 131 | beneficiaries: Optional[list[Beneficiary]] = None, |
145 | 132 | govt_id: Optional[KYCFile] = None, |
146 | 133 | proof_of_address: Optional[KYCFile] = None, |
147 | 134 | proof_of_life: Optional[KYCFile] = None, |
148 | | - terms_of_service: Optional[TOSRequest] = None, |
149 | | - verification_id: Optional[str] = None, |
150 | 135 | status: Optional[UserStatus] = None, |
151 | 136 | email_verification_id: Optional[str] = None, |
152 | 137 | phone_verification_id: Optional[str] = None, |
153 | 138 | curp_document: Optional[SerializableHttpUrl] = None, |
154 | 139 | fiscal_regime_code: Optional[SATRegimeCode] = None, |
| 140 | + pronouns: Optional[str] = None, |
155 | 141 | *, |
156 | 142 | session: Session = global_session, |
157 | 143 | ) -> 'User': |
158 | 144 | request = UserUpdateRequest( |
159 | | - phone_number=phone_number, |
160 | | - email_address=email_address, |
161 | 145 | profession=profession, |
162 | 146 | address=address, |
163 | 147 | beneficiaries=beneficiaries, |
164 | 148 | govt_id=govt_id, |
165 | 149 | proof_of_address=proof_of_address, |
166 | 150 | proof_of_life=proof_of_life, |
167 | | - terms_of_service=terms_of_service, |
168 | | - verification_id=verification_id, |
| 151 | + status=status, |
169 | 152 | email_verification_id=email_verification_id, |
170 | 153 | phone_verification_id=phone_verification_id, |
171 | 154 | curp_document=curp_document, |
172 | 155 | fiscal_regime_code=fiscal_regime_code, |
173 | | - status=status, |
| 156 | + pronouns=pronouns, |
174 | 157 | ) |
175 | 158 | return cls._update(id=user_id, **request.model_dump(), session=session) |
176 | 159 |
|
|
0 commit comments