<iman> update new switch account api

1 files changed, 14 insertions, 5 deletions

diff --git a/indoteknik_api/controllers/api_v1/user.py b/indoteknik_api/controllers/api_v1/user.py
index 7166bd79..4b29c820 100644
--- a/indoteknik_api/controllers/api_v1/user.py
+++ b/indoteknik_api/controllers/api_v1/user.py
@@ -409,8 +409,8 @@ class User(controller.Controller):
     @http.route(prefix + 'user/<id>/switch', auth='public', methods=['PUT', 'OPTIONS'], csrf=False)
     @controller.Controller.must_authorized()
     def switch_account(self, **kw):
-        id = kw.get('id')
-        user = request.env['res.users'].search([('id', '=', id)], limit=1)
+        id = int(kw.get('id'))
+        user = request.env['res.partner'].search([('id', '=', id)], limit=1)
 
         response = {
             'switch': False,
@@ -482,7 +482,7 @@ class User(controller.Controller):
                 }
                 new_company = request.env['res.partner'].create(new_company_data)
                 request.env['user.company.request'].create({
-                    'user_id': user.partner_id.id,
+                    'user_id': user.id,
                     'user_company_id': new_company.id,
                     'user_input': business_name
                 })
@@ -520,6 +520,9 @@ class User(controller.Controller):
                         'mimetype': sppkp_mimetype
                     })
                     new_company.message_post(body="SPPKP Uploaded", attachment_ids=[sppkp_attachment.id])
+        request_company = self.get_user_by_email(user.email)
+        request_company.parent_name = business_name
+        request_company.send_company_request_mail()
 
         response['switch'] = 'Pending'
         return self.response(response)
@@ -528,7 +531,7 @@ class User(controller.Controller):
     # @controller.Controller.must_authorized()
     def switch_account_progres(self, **kw):
         id = int(kw.get('id'))
-        user = request.env['res.users'].search([('id', '=', id)], limit=1)
+        # user = request.env['res.partner'].search([('id', '=', id)], limit=1)
         response = {
             'status': ''
         }
@@ -545,4 +548,10 @@ class User(controller.Controller):
                 response['status'] = new_company_request.is_approve
         else:
             response['status'] = 'unknown'
-        return self.response(response)
\ No newline at end of file
+        return self.response(response)
+
+    def get_user_by_email(self, email):
+        return request.env['res.users'].search([
+            ('login', '=', email),
+            ('active', 'in', [True, False])
+        ])
\ No newline at end of file