| Total Complexity | 47 |
| Total Lines | 112 |
| 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 |
||
| 129 | class ResultSets(object): |
||
| 130 | |||
| 131 | def __init__(self): |
||
| 132 | self.foo = '' |
||
| 133 | |||
| 134 | def selector(self, output): |
||
| 135 | if isinstance(output, boto.ec2.instance.Reservation): |
||
| 136 | return self.parseReservation(output) |
||
| 137 | elif isinstance(output, boto.ec2.instance.Instance): |
||
| 138 | return self.parseInstance(output) |
||
| 139 | elif isinstance(output, boto.ec2.volume.Volume): |
||
| 140 | return self.parseVolume(output) |
||
| 141 | elif isinstance(output, boto.ec2.blockdevicemapping.BlockDeviceType): |
||
| 142 | return self.parseBlockDeviceType(output) |
||
| 143 | elif isinstance(output, boto.ec2.zone.Zone): |
||
| 144 | return self.parseEC2Zone(output) |
||
| 145 | elif isinstance(output, boto.ec2.address.Address): |
||
| 146 | return self.parseAddress(output) |
||
| 147 | elif isinstance(output, boto.route53.record.Record): |
||
| 148 | return self.parseRecord(output) |
||
| 149 | elif isinstance(output, boto.route53.zone.Zone): |
||
| 150 | return self.parseR53Zone(output) |
||
| 151 | elif isinstance(output, boto.route53.status.Status): |
||
| 152 | return self.parseR53Status(output) |
||
| 153 | elif isinstance(output, boto.ec2.tag.Tag): |
||
| 154 | return self.parseTag(output) |
||
| 155 | elif isinstance(output, boto.ec2.ec2object.EC2Object): |
||
| 156 | return self.parseEC2Object(output) |
||
| 157 | else: |
||
| 158 | return output |
||
| 159 | |||
| 160 | def formatter(self, output): |
||
| 161 | if isinstance(output, list): |
||
| 162 | return [self.formatter(item) for item in output] |
||
| 163 | elif isinstance(output, dict): |
||
| 164 | return {key: self.formatter(value) for key, value in six.iteritems(output)} |
||
| 165 | else: |
||
| 166 | return self.selector(output) |
||
| 167 | |||
| 168 | def parseReservation(self, output): |
||
| 169 | instance_list = [] |
||
| 170 | for instance in output.instances: |
||
| 171 | instance_data = self.parseInstance(instance) |
||
| 172 | instance_data['owner_id'] = output.owner_id |
||
| 173 | instance_list.append(instance_data) |
||
| 174 | return instance_list |
||
| 175 | |||
| 176 | def parseAddress(self, output): |
||
| 177 | instance_data = {field: getattr(output, field) for field in FieldLists.ADDRESS} |
||
| 178 | return instance_data |
||
| 179 | |||
| 180 | def parseInstance(self, output): |
||
| 181 | instance_data = {field: getattr(output, field) for field in FieldLists.INSTANCE} |
||
| 182 | return instance_data |
||
| 183 | |||
| 184 | def parseVolume(self, output): |
||
| 185 | volume_data = {field: getattr(output, field) for field in FieldLists.VOLUME} |
||
| 186 | return volume_data |
||
| 187 | |||
| 188 | def parseBlockDeviceType(self, output): |
||
| 189 | data = {field: getattr(output, field) for field in FieldLists.BLOCK_DEVICE_TYPE} |
||
| 190 | return data |
||
| 191 | |||
| 192 | def parseEC2Zone(self, output): |
||
| 193 | zone_data = {field: getattr(output, field) for field in FieldLists.EC2ZONE} |
||
| 194 | return zone_data |
||
| 195 | |||
| 196 | def parseRecord(self, output): |
||
| 197 | record_data = {field: getattr(output, field) for field in FieldLists.RECORD} |
||
| 198 | return record_data |
||
| 199 | |||
| 200 | def parseR53Zone(self, output): |
||
| 201 | zone_data = {field: getattr(output, field) for field in FieldLists.R53ZONE} |
||
| 202 | return zone_data |
||
| 203 | |||
| 204 | def parseR53Status(self, output): |
||
| 205 | status_data = {field: getattr(output, field) for field in FieldLists.R53STATUS} |
||
| 206 | return status_data |
||
| 207 | |||
| 208 | def parseBucket(self, output): |
||
| 209 | bucket_data = {field: getattr(output, field) for field in FieldLists.BUCKET} |
||
| 210 | return bucket_data |
||
| 211 | |||
| 212 | def parseTag(self, output): |
||
| 213 | tag_data = {field: getattr(output, field) for field in FieldLists.TAG} |
||
| 214 | return tag_data |
||
| 215 | |||
| 216 | def parseEC2Object(self, output): |
||
| 217 | # Looks like everything that is an EC2Object pretty much only has these extra |
||
| 218 | # 'unparseable' properties so handle region and connection specially. |
||
| 219 | output = vars(output) |
||
| 220 | del output['connection'] |
||
| 221 | # special handling for region since name here is better than id. |
||
| 222 | region = output.get('region', None) |
||
| 223 | output['region'] = region.name if region else '' |
||
| 224 | # now anything that is an EC2Object get some special marshalling care. |
||
| 225 | for k, v in six.iteritems(output): |
||
| 226 | if isinstance(v, boto.ec2.ec2object.EC2Object): |
||
| 227 | # Better not to assume each EC2Object has an id. If not found |
||
| 228 | # resort to the str of the object which should have something meaningful. |
||
| 229 | output[k] = getattr(v, 'id', str(v)) |
||
| 230 | # Generally unmarshallable object might be hiding in list so better to |
||
| 231 | if isinstance(v, list): |
||
| 232 | v_list = [] |
||
| 233 | for item in v: |
||
| 234 | # avoid touching the basic types. |
||
| 235 | if isinstance(item, (basestring, bool, int, long, float)): |
||
| 236 | v_list.append(v) |
||
| 237 | else: |
||
| 238 | v_list.append(str(item)) |
||
| 239 | output[k] = v_list |
||
| 240 | return output |
||
| 241 |