# Python script to update companies
old_clickstocart = env['res.company'].search([('name', '=', 'Clickstocart Inc')], limit=1)
if old_clickstocart:
    old_clickstocart.write({'name': 'Clickstocart Inc US'})
    print("Renamed Clickstocart Inc to Clickstocart Inc US")

ca_company = env['res.company'].search([('name', '=', 'Clickstocart Inc CA')], limit=1)
if not ca_company:
    ca_company = env['res.company'].create({'name': 'Clickstocart Inc CA'})
    print("Created new company: Clickstocart Inc CA")

# Ensure warehouses exist for both
all_companies = env['res.company'].search([])
for comp in all_companies:
    if comp.name == 'My Company': continue
    warehouse = env['stock.warehouse'].search([('company_id', '=', comp.id)], limit=1)
    if not warehouse:
        env['stock.warehouse'].create({
            'name': comp.name,
            'code': comp.name[-3:].upper().strip(),
            'company_id': comp.id
        })
        print(f"Created warehouse for {comp.name}")

# Ensure admin has access to all companies
admin = env.ref('base.user_admin')
admin.write({
    'company_ids': [(6, 0, all_companies.ids)]
})

# Ensure the specified user has access to all companies
alaguraj = env['res.users'].search([('login', '=', 'alaguraj0361@gmail.com')], limit=1)
if alaguraj:
    alaguraj.write({
        'company_ids': [(6, 0, all_companies.ids)]
    })

env.cr.commit()
print("Done configuring 4 companies.")