| Total Complexity | 44 | 
| Total Lines | 106 | 
| Duplicated Lines | 0 % | 
Complex classes like ResultSets often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
| 1 | import boto | ||
| 121 | class ResultSets(object): | ||
| 122 | |||
| 123 | def __init__(self): | ||
| 124 | self.foo = '' | ||
| 125 | |||
| 126 | def selector(self, output): | ||
| 127 | if isinstance(output, boto.ec2.instance.Reservation): | ||
| 128 | return self.parseReservation(output) | ||
| 129 | elif isinstance(output, boto.ec2.instance.Instance): | ||
| 130 | return self.parseInstance(output) | ||
| 131 | elif isinstance(output, boto.ec2.volume.Volume): | ||
| 132 | return self.parseVolume(output) | ||
| 133 | elif isinstance(output, boto.ec2.blockdevicemapping.BlockDeviceType): | ||
| 134 | return self.parseBlockDeviceType(output) | ||
| 135 | elif isinstance(output, boto.ec2.zone.Zone): | ||
| 136 | return self.parseEC2Zone(output) | ||
| 137 | elif isinstance(output, boto.ec2.address.Address): | ||
| 138 | return self.parseAddress(output) | ||
| 139 | elif isinstance(output, boto.route53.record.Record): | ||
| 140 | return self.parseRecord(output) | ||
| 141 | elif isinstance(output, boto.route53.zone.Zone): | ||
| 142 | return self.parseR53Zone(output) | ||
| 143 | elif isinstance(output, boto.route53.status.Status): | ||
| 144 | return self.parseR53Status(output) | ||
| 145 | elif isinstance(output, boto.ec2.ec2object.EC2Object): | ||
| 146 | return self.parseEC2Object(output) | ||
| 147 | else: | ||
| 148 | return output | ||
| 149 | |||
| 150 | def formatter(self, output): | ||
| 151 | if isinstance(output, list): | ||
| 152 | return [self.formatter(item) for item in output] | ||
| 153 | elif isinstance(output, dict): | ||
| 154 |             return {key: self.formatter(value) for key, value in six.iteritems(output)} | ||
| 155 | else: | ||
| 156 | return self.selector(output) | ||
| 157 | |||
| 158 | def parseReservation(self, output): | ||
| 159 | instance_list = [] | ||
| 160 | for instance in output.instances: | ||
| 161 | instance_data = self.parseInstance(instance) | ||
| 162 | instance_data['owner_id'] = output.owner_id | ||
| 163 | instance_list.append(instance_data) | ||
| 164 | return instance_list | ||
| 165 | |||
| 166 | def parseAddress(self, output): | ||
| 167 |         instance_data = {field: getattr(output, field) for field in FieldLists.ADDRESS} | ||
| 168 | return instance_data | ||
| 169 | |||
| 170 | def parseInstance(self, output): | ||
| 171 |         instance_data = {field: getattr(output, field) for field in FieldLists.INSTANCE} | ||
| 172 | return instance_data | ||
| 173 | |||
| 174 | def parseVolume(self, output): | ||
| 175 |         volume_data = {field: getattr(output, field) for field in FieldLists.VOLUME} | ||
| 176 | return volume_data | ||
| 177 | |||
| 178 | def parseBlockDeviceType(self, output): | ||
| 179 |         data = {field: getattr(output, field) for field in FieldLists.BLOCK_DEVICE_TYPE} | ||
| 180 | return data | ||
| 181 | |||
| 182 | def parseEC2Zone(self, output): | ||
| 183 |         zone_data = {field: getattr(output, field) for field in FieldLists.EC2ZONE} | ||
| 184 | return zone_data | ||
| 185 | |||
| 186 | def parseRecord(self, output): | ||
| 187 |         record_data = {field: getattr(output, field) for field in FieldLists.RECORD} | ||
| 188 | return record_data | ||
| 189 | |||
| 190 | def parseR53Zone(self, output): | ||
| 191 |         zone_data = {field: getattr(output, field) for field in FieldLists.R53ZONE} | ||
| 192 | return zone_data | ||
| 193 | |||
| 194 | def parseR53Status(self, output): | ||
| 195 |         status_data = {field: getattr(output, field) for field in FieldLists.R53STATUS} | ||
| 196 | return status_data | ||
| 197 | |||
| 198 | def parseBucket(self, output): | ||
| 199 |         bucket_data = {field: getattr(output, field) for field in FieldLists.BUCKET} | ||
| 200 | return bucket_data | ||
| 201 | |||
| 202 | def parseEC2Object(self, output): | ||
| 203 | # Looks like everything that is an EC2Object pretty much only has these extra | ||
| 204 | # 'unparseable' properties so handle region and connection specially. | ||
| 205 | output = vars(output) | ||
| 206 | del output['connection'] | ||
| 207 | # special handling for region since name here is better than id. | ||
| 208 |         region = output.get('region', None) | ||
| 209 | output['region'] = region.name if region else '' | ||
| 210 | # now anything that is an EC2Object get some special marshalling care. | ||
| 211 | for k, v in six.iteritems(output): | ||
| 212 | if isinstance(v, boto.ec2.ec2object.EC2Object): | ||
| 213 | # Better not to assume each EC2Object has an id. If not found | ||
| 214 | # resort to the str of the object which should have something meaningful. | ||
| 215 | output[k] = getattr(v, 'id', str(v)) | ||
| 216 | # Generally unmarshallable object might be hiding in list so better to | ||
| 217 | if isinstance(v, list): | ||
| 218 | v_list = [] | ||
| 219 | for item in v: | ||
| 220 | # avoid touching the basic types. | ||
| 221 | if isinstance(item, (basestring, bool, int, long, float)): | ||
| 222 | v_list.append(v) | ||
| 223 | else: | ||
| 224 | v_list.append(str(item)) | ||
| 225 | output[k] = v_list | ||
| 226 | return output | ||
| 227 |