<iman> add CR voucher type account

2 files changed, 12 insertions, 1 deletions

diff --git a/indoteknik_custom/models/res_users.py b/indoteknik_custom/models/res_users.py
index 31b84ae3..63942bb5 100755
--- a/indoteknik_custom/models/res_users.py
+++ b/indoteknik_custom/models/res_users.py
@@ -71,6 +71,13 @@ class ResUsers(models.Model):
             if not vouchers: return None
             return ', '.join(x.code for x in vouchers)
         return None
+
+    def get_voucher_code_switch_account(self, type):
+        if type == 'switch_account':
+            vouchers = self.env['voucher'].get_active_voucher([('account_type', 'in', ['all','company'])])
+            if not vouchers: return None
+            return ', '.join(x.code for x in vouchers)
+        return None
     
     def check_access(self, model, mode):
         assert mode in ('read', 'write', 'create', 'unlink', 'import', 'export'), 'Invalid access mode'
diff --git a/indoteknik_custom/models/voucher.py b/indoteknik_custom/models/voucher.py
index 37c97338..101d4bcf 100644
--- a/indoteknik_custom/models/voucher.py
+++ b/indoteknik_custom/models/voucher.py
@@ -59,7 +59,11 @@ class Voucher(models.Model):
     show_on_email = fields.Selection([
         ('user_activation', 'User Activation')
     ], 'Show on Email')
-
+    account_type = fields.Selection(string='Account Type', default="all", selection=[
+        ('all', "All Account"),
+        ('person', "Account Individu"),
+        ('company', "Account Company"),
+    ])
     @api.constrains('description')
     def _check_description_length(self):
         for record in self: