| @@ 21-81 (lines=61) @@ | ||
| 18 | ] |
|
| 19 | ||
| 20 | ||
| 21 | class UserRuleService(Service): |
|
| 22 | __key__ = 'session.user.rule' |
|
| 23 | ||
| 24 | @expose |
|
| 25 | def list(self, full=False): |
|
| 26 | return [ |
|
| 27 | rule.to_json(full=full) |
|
| 28 | for rule in UserRuleManager.get.all() |
|
| 29 | ] |
|
| 30 | ||
| 31 | @expose |
|
| 32 | def update(self, current, full=False): |
|
| 33 | result = [] |
|
| 34 | ||
| 35 | # Build array of current ids |
|
| 36 | current_ids = [ |
|
| 37 | r.get('id') |
|
| 38 | for r in current |
|
| 39 | if r.get('id') is not None |
|
| 40 | ] |
|
| 41 | ||
| 42 | # Delete rules |
|
| 43 | deleted_rules = [ |
|
| 44 | rule |
|
| 45 | for rule in UserRuleManager.get.all() |
|
| 46 | if rule.id not in current_ids |
|
| 47 | ] |
|
| 48 | ||
| 49 | for rule in deleted_rules: |
|
| 50 | rule.delete_instance() |
|
| 51 | ||
| 52 | log.debug('Deleted %r', rule) |
|
| 53 | ||
| 54 | # Create/Update client rules |
|
| 55 | for r in current: |
|
| 56 | id = r.pop('id', None) |
|
| 57 | ||
| 58 | if id is None: |
|
| 59 | # Create new rule |
|
| 60 | rule = UserRuleManager.create(**r) |
|
| 61 | ||
| 62 | log.debug('Created %r', rule) |
|
| 63 | result.append(rule) |
|
| 64 | continue |
|
| 65 | ||
| 66 | # Retrieve existing rule |
|
| 67 | rule = UserRuleManager.get(id=id) |
|
| 68 | ||
| 69 | # Update rule |
|
| 70 | UserRuleManager.update(rule, r) |
|
| 71 | ||
| 72 | log.debug('Updated %r', rule) |
|
| 73 | result.append(rule) |
|
| 74 | ||
| 75 | # Ensure result is sorted by priority |
|
| 76 | result = sorted(result, key=lambda item: item.priority) |
|
| 77 | ||
| 78 | # Convert rules to serializable dictionaries |
|
| 79 | return [ |
|
| 80 | r.to_json(full=full) |
|
| 81 | for r in result |
|
| 82 | ] |
|
| 83 | ||
| @@ 21-81 (lines=61) @@ | ||
| 18 | ] |
|
| 19 | ||
| 20 | ||
| 21 | class ClientRuleService(Service): |
|
| 22 | __key__ = 'session.client.rule' |
|
| 23 | ||
| 24 | @expose |
|
| 25 | def list(self, full=False): |
|
| 26 | return [ |
|
| 27 | rule.to_json(full=full) |
|
| 28 | for rule in ClientRuleManager.get.all() |
|
| 29 | ] |
|
| 30 | ||
| 31 | @expose |
|
| 32 | def update(self, current, full=False): |
|
| 33 | result = [] |
|
| 34 | ||
| 35 | # Build array of current ids |
|
| 36 | current_ids = [ |
|
| 37 | r.get('id') |
|
| 38 | for r in current |
|
| 39 | if r.get('id') is not None |
|
| 40 | ] |
|
| 41 | ||
| 42 | # Delete rules |
|
| 43 | deleted_rules = [ |
|
| 44 | rule |
|
| 45 | for rule in ClientRuleManager.get.all() |
|
| 46 | if rule.id not in current_ids |
|
| 47 | ] |
|
| 48 | ||
| 49 | for rule in deleted_rules: |
|
| 50 | rule.delete_instance() |
|
| 51 | ||
| 52 | log.debug('Deleted %r', rule) |
|
| 53 | ||
| 54 | # Create/Update client rules |
|
| 55 | for r in current: |
|
| 56 | id = r.pop('id', None) |
|
| 57 | ||
| 58 | if id is None: |
|
| 59 | # Create new rule |
|
| 60 | rule = ClientRuleManager.create(**r) |
|
| 61 | ||
| 62 | log.debug('Created %r', rule) |
|
| 63 | result.append(rule) |
|
| 64 | continue |
|
| 65 | ||
| 66 | # Retrieve existing rule |
|
| 67 | rule = ClientRuleManager.get(id=id) |
|
| 68 | ||
| 69 | # Update rule |
|
| 70 | ClientRuleManager.update(rule, r) |
|
| 71 | ||
| 72 | log.debug('Updated %r', rule) |
|
| 73 | result.append(rule) |
|
| 74 | ||
| 75 | # Ensure result is sorted by priority |
|
| 76 | result = sorted(result, key=lambda item: item.priority) |
|
| 77 | ||
| 78 | # Convert rules to serializable dictionaries |
|
| 79 | return [ |
|
| 80 | r.to_json(full=full) |
|
| 81 | for r in result |
|
| 82 | ] |
|
| 83 | ||