How to use _cleanup_labels method in autotest

Best Python code snippet using autotest_python

host.py

Source:host.py Github

copy

Full Screen

...48 self.messages.append('Locked host')49 elif options.unlock:50 self.data['locked'] = False51 self.messages.append('Unlocked host')52 def _cleanup_labels(self, labels, platform=None):53 """Removes the platform label from the overall labels"""54 if platform:55 return [label for label in labels56 if label != platform]57 else:58 try:59 return [label for label in labels60 if not label['platform']]61 except TypeError:62 # This is a hack - the server will soon63 # do this, so all this code should be removed.64 return labels65 def get_items(self):66 return self.hosts67class host_help(host):68 """Just here to get the atest logic working.69 Usage is set by its parent"""70 pass71class host_list(action_common.atest_list, host):72 """atest host list [--mlist <file>|<hosts>] [--label <label>]73 [--status <status1,status2>] [--acl <ACL>] [--user <user>]"""74 def __init__(self):75 super(host_list, self).__init__()76 self.parser.add_option('-b', '--label',77 default='',78 help='Only list hosts with all these labels '79 '(comma separated)')80 self.parser.add_option('-s', '--status',81 default='',82 help='Only list hosts with any of these '83 'statuses (comma separated)')84 self.parser.add_option('-a', '--acl',85 default='',86 help='Only list hosts within this ACL')87 self.parser.add_option('-u', '--user',88 default='',89 help='Only list hosts available to this user')90 self.parser.add_option('-N', '--hostnames-only', help='Only return '91 'hostnames for the machines queried.',92 action='store_true')93 self.parser.add_option('--locked',94 default=False,95 help='Only list locked hosts',96 action='store_true')97 self.parser.add_option('--unlocked',98 default=False,99 help='Only list unlocked hosts',100 action='store_true')101 def parse(self):102 """Consume the specific options"""103 label_info = topic_common.item_parse_info(attribute_name='labels',104 inline_option='label')105 (options, leftover) = super(host_list, self).parse([label_info])106 self.status = options.status107 self.acl = options.acl108 self.user = options.user109 self.hostnames_only = options.hostnames_only110 if options.locked and options.unlocked:111 self.invalid_syntax('--locked and --unlocked are '112 'mutually exclusive')113 self.locked = options.locked114 self.unlocked = options.unlocked115 return (options, leftover)116 def execute(self):117 filters = {}118 check_results = {}119 if self.hosts:120 filters['hostname__in'] = self.hosts121 check_results['hostname__in'] = 'hostname'122 if self.labels:123 if len(self.labels) == 1:124 # This is needed for labels with wildcards (x86*)125 filters['labels__name__in'] = self.labels126 check_results['labels__name__in'] = None127 else:128 filters['multiple_labels'] = self.labels129 check_results['multiple_labels'] = None130 if self.status:131 statuses = self.status.split(',')132 statuses = [status.strip() for status in statuses133 if status.strip()]134 filters['status__in'] = statuses135 check_results['status__in'] = None136 if self.acl:137 filters['aclgroup__name'] = self.acl138 check_results['aclgroup__name'] = None139 if self.user:140 filters['aclgroup__users__login'] = self.user141 check_results['aclgroup__users__login'] = None142 if self.locked or self.unlocked:143 filters['locked'] = self.locked144 check_results['locked'] = None145 return super(host_list, self).execute(op='get_hosts',146 filters=filters,147 check_results=check_results)148 def output(self, results):149 if results:150 # Remove the platform from the labels.151 for result in results:152 result['labels'] = self._cleanup_labels(result['labels'],153 result['platform'])154 if self.hostnames_only:155 self.print_list(results, key='hostname')156 else:157 super(host_list, self).output(results, keys=['hostname', 'status',158 'locked', 'platform', 'labels'])159class host_stat(host):160 """atest host stat --mlist <file>|<hosts>"""161 usage_action = 'stat'162 def execute(self):163 results = []164 # Convert wildcards into real host stats.165 existing_hosts = []166 for host in self.hosts:167 if host.endswith('*'):168 stats = self.execute_rpc('get_hosts',169 hostname__startswith=host.rstrip('*'))170 if len(stats) == 0:171 self.failure('No hosts matching %s' % host, item=host,172 what_failed='Failed to stat')173 continue174 else:175 stats = self.execute_rpc('get_hosts', hostname=host)176 if len(stats) == 0:177 self.failure('Unknown host %s' % host, item=host,178 what_failed='Failed to stat')179 continue180 existing_hosts.extend(stats)181 for stat in existing_hosts:182 host = stat['hostname']183 # The host exists, these should succeed184 acls = self.execute_rpc('get_acl_groups', hosts__hostname=host)185 labels = self.execute_rpc('get_labels', host__hostname=host)186 results.append([[stat], acls, labels])187 return results188 def output(self, results):189 for stats, acls, labels in results:190 print '-' * 5191 self.print_fields(stats,192 keys=['hostname', 'platform',193 'status', 'locked', 'locked_by',194 'lock_time', 'protection', ])195 self.print_by_ids(acls, 'ACLs', line_before=True)196 labels = self._cleanup_labels(labels)197 self.print_by_ids(labels, 'Labels', line_before=True)198class host_jobs(host):199 """atest host jobs [--max-query] --mlist <file>|<hosts>"""200 usage_action = 'jobs'201 def __init__(self):202 super(host_jobs, self).__init__()203 self.parser.add_option('-q', '--max-query',204 help='Limits the number of results '205 '(20 by default)',206 type='int', default=20)207 def parse(self):208 """Consume the specific options"""209 (options, leftover) = super(host_jobs, self).parse()210 self.max_queries = options.max_query...

Full Screen

Full Screen

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.

LambdaTest Learning Hubs:

YouTube

You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Run autotest automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

Next-Gen App & Browser Testing Cloud

Was this article helpful?

Helpful

NotHelpful