FIX company_code: records with company_id is False

company_code/README.rst

@@ -35,11 +35,13 @@ write this code in your custom code according your model
 
 
 ```python
+
 class ResPartner(models.Model):
     _inherit = 'res.partner'
 
     def name_get(self):
         return self.env['res.company']._add_company_code(super())
+
 ```
 
 Bug Tracker

company_code/models/company.py

@@ -24,9 +24,10 @@ class ResCompany(models.Model):
         else:
             codes = {x.id: x['company_id']['code'] for x in records
                      if getattr(x, 'company_id')}
-        res = [(elm[0], '%s (%s)' % (elm[1], codes[elm[0]] or ''))
+        if not codes:
+            return super_object.name_get()
+        return [(elm[0], '%s (%s)' % (elm[1], codes[elm[0]] or ''))
                 for elm in super_object.name_get()]
-        return res
 
     def name_get(self):
         return self.env['res.company']._add_company_code(super())