from odoo import models


class ResUsers(models.Model):
    _inherit = 'res.users'
    
    def api_single_response(self, res_user, with_detail=''):
        data = {
            'id': res_user.id,
            'partner_id': res_user.partner_id.id,
            'name': res_user.name,
            'email': res_user.login,
            'phone': res_user.phone or '',
            'mobile': res_user.mobile or '',
            'external': res_user.share,
            'company': res_user.company_type == 'company'
        }
        if res_user.parent_id:
            data.update({ 'company': res_user.parent_id.company_type == 'company' })
        
        return data
    
    def api_address_response(self, user):
        data = {
            'id': user.id,
            'type': user.type or '',
            'name': user.name or '',
            'mobile': user.mobile or '',
            'email': user.email or '',
            'street': user.street or '',
            'street2': user.street2 or '',
            'city': None,
            'district': None,
            'sub_district': None,
            'zip': user.zip or ''
        }
        
        if user.kota_id:
            data['city'] = {
                'id': user.kota_id.id,
                'name': user.kota_id.name
            } or None
            
        if user.kecamatan_id:
            data['district'] = {
                'id': user.kecamatan_id.id,
                'name': user.kecamatan_id.name
            }
            
        if user.kelurahan_id:
            data['sub_district'] = {
                'id': user.kelurahan_id.id, 
                'name': user.kelurahan_id.name
            }
            
        return data