Respect flake8/pep8 rules

This change updates code to respect flake8/pep8 rules. It changes:

 * correct KeystoneManager.set_client signature
 * correct code indentation
 * remove useless variables
 * support py3X
   * use netaddr instead of ipaddr (not supported by py34)
   * use dict.values instead of dict.itervalues.
 * correct tests
   * "fake.networks[0]['router:external'] = True" update had side
     effects on other tests, fake managers attributes are now builded
     at instantiation to avoid such trouble.

Change-Id: I56943e198c3dc0e28be5f5b799e62f4a6427292c
This commit is contained in:
Cedric Brandily 2014-11-07 00:42:15 +01:00
parent de8f156835
commit ca3f5b8dcc
5 changed files with 76 additions and 76 deletions

View File

@ -27,7 +27,7 @@ from __future__ import print_function
import argparse import argparse
import os import os
from flame import TemplateGenerator from flameclient import flame
def main(args=None): def main(args=None):
@ -67,8 +67,9 @@ def main(args=None):
"stack data file.") "stack data file.")
args = parser.parse_args() args = parser.parse_args()
template = TemplateGenerator(args.username, args.password, args.project, template = flame.TemplateGenerator(args.username, args.password,
args.auth_url, args.insecure) args.project, args.auth_url,
args.insecure)
template.extract_vm_details(args.exclude_servers, args.exclude_volumes, template.extract_vm_details(args.exclude_servers, args.exclude_volumes,
args.generate_stack_data) args.generate_stack_data)
template.extract_data() template.extract_data()

View File

@ -24,7 +24,7 @@
import logging import logging
import ipaddr import netaddr
import yaml import yaml
from flameclient import managers from flameclient import managers
@ -193,8 +193,10 @@ class TemplateGenerator(object):
port_resource_name: { port_resource_name: {
'type': resource_type, 'type': resource_type,
'properties': { 'properties': {
'subnet_id': {'get_resource': subnet_resource_name}, 'subnet_id': {
'router_id': {'get_resource': router_resource_name} 'get_resource': subnet_resource_name},
'router_id': {
'get_resource': router_resource_name}
} }
} }
} }
@ -229,7 +231,7 @@ class TemplateGenerator(object):
router) router)
def _extract_networks(self): def _extract_networks(self):
for n, network in self.networks.itervalues(): for n, network in self.networks.values():
if network['router:external']: if network['router:external']:
self.external_networks.append(network['id']) self.external_networks.append(network['id'])
continue continue
@ -261,7 +263,7 @@ class TemplateGenerator(object):
return "subnet_%d" % self.subnets[subnet_id][0] return "subnet_%d" % self.subnets[subnet_id][0]
def _extract_subnets(self): def _extract_subnets(self):
for n, subnet in self.subnets.itervalues(): for n, subnet in self.subnets.values():
if subnet['network_id'] in self.external_networks: if subnet['network_id'] in self.external_networks:
continue continue
subnet_resource_name = "subnet_%d" % n subnet_resource_name = "subnet_%d" % n
@ -306,12 +308,12 @@ class TemplateGenerator(object):
del rule['tenant_id'] del rule['tenant_id']
del rule['id'] del rule['id']
del rule['security_group_id'] del rule['security_group_id']
rule = dict((k, v) for k, v in rule.iteritems() if v is not None) rule = dict((k, v) for k, v in rule.items() if v is not None)
brules.append(rule) brules.append(rule)
return brules return brules
def _extract_secgroups(self): def _extract_secgroups(self):
for n, secgroup in self.secgroups.itervalues(): for n, secgroup in self.secgroups.values():
resource_name = "security_group_%d" % n resource_name = "security_group_%d" % n
resource_type = 'OS::Neutron::SecurityGroup' resource_type = 'OS::Neutron::SecurityGroup'
@ -340,7 +342,7 @@ class TemplateGenerator(object):
self.template['resources'].update(resource) self.template['resources'].update(resource)
def _extract_keys(self): def _extract_keys(self):
for n, key in self.keys.itervalues(): for n, key in self.keys.values():
key_resource_name = "key_%d" % n key_resource_name = "key_%d" % n
resource_type = 'OS::Nova::KeyPair' resource_type = 'OS::Nova::KeyPair'
@ -389,9 +391,9 @@ class TemplateGenerator(object):
networks = [] networks = []
for net_name in addresses: for net_name in addresses:
ip = addresses[net_name][0]['addr'] ip = addresses[net_name][0]['addr']
for s, subnet in self.subnets.itervalues(): for s, subnet in self.subnets.values():
if ipaddr.IPAddress(ip) in ipaddr.IPNetwork(subnet['cidr']): if netaddr.IPAddress(ip) in netaddr.IPNetwork(subnet['cidr']):
for n, network in self.networks.itervalues(): for n, network in self.networks.values():
if (network['name'] == net_name and if (network['name'] == net_name and
network['id'] == subnet['network_id']): network['id'] == subnet['network_id']):
net = self.get_network_resource_name( net = self.get_network_resource_name(
@ -400,7 +402,7 @@ class TemplateGenerator(object):
return networks return networks
def _extract_servers(self): def _extract_servers(self):
for n, server in self.servers.itervalues(): for n, server in self.servers.values():
resource_name = "server_%d" % n resource_name = "server_%d" % n
resource_type = 'OS::Nova::Server' resource_type = 'OS::Nova::Server'
@ -533,7 +535,7 @@ class TemplateGenerator(object):
self.template['resources'].update(floating_resource) self.template['resources'].update(floating_resource)
def _extract_volumes(self): def _extract_volumes(self):
for n, volume in self.volumes.itervalues(): for n, volume in self.volumes.values():
resource_name = "volume_%d" % n resource_name = "volume_%d" % n
resource_type = 'OS::Cinder::Volume' resource_type = 'OS::Cinder::Volume'

View File

@ -48,7 +48,7 @@ class KeystoneManager(object):
insecure=self.insecure) insecure=self.insecure)
return self._client return self._client
def set_client(client): def set_client(self, client):
self._client = client self._client = client
def get_token(self): def get_token(self):
@ -138,8 +138,8 @@ class NovaManager(object):
def client(self): def client(self):
if not self._client: if not self._client:
self._client = nova_client.Client(self.username, self.password, self._client = nova_client.Client(self.username, self.password,
self.project, self.auth_url, self.project, self.auth_url,
insecure=self.insecure) insecure=self.insecure)
return self._client return self._client
def set_client(self, client): def set_client(self, client):

View File

@ -26,7 +26,6 @@ import datetime
import mock import mock
import flameclient
from flameclient.flame import TemplateGenerator # noqa from flameclient.flame import TemplateGenerator # noqa
from flameclient.tests import base from flameclient.tests import base
@ -85,21 +84,23 @@ class FakeSecurityGroup(FakeBase):
class FakeNeutronManager(object): class FakeNeutronManager(object):
groups = []
routers = [{'name': 'myrouter', def __init__(self):
'id': '1234', self.groups = []
'admin_state_up': 'true', self.routers = [{'name': 'myrouter',
'external_gateway_info': None}, ] 'id': '1234',
ports = [] 'admin_state_up': 'true',
subnets = [] 'external_gateway_info': None}, ]
networks = [{'status': 'ACTIVE', self.ports = []
'subnets': ['1111'], self.subnets = []
'name': 'mynetwork', self.networks = [{'status': 'ACTIVE',
'router:external': False, 'subnets': ['1111'],
'admin_state_up': True, 'name': 'mynetwork',
'shared': False, 'router:external': False,
'id': '2222'}, ] 'admin_state_up': True,
floatingips = [] 'shared': False,
'id': '2222'}, ]
self.floatingips = []
def subnet_list(self): def subnet_list(self):
return self.subnets return self.subnets
@ -125,10 +126,12 @@ class FakeNeutronManager(object):
class FakeNovaManager(object): class FakeNovaManager(object):
servers = [FakeServer()] def __init__(self):
flavors = [FakeFlavor(id='2', name='m1.small')] self.servers = [FakeServer()]
groups = {} self.flavors = [FakeFlavor(id='2', name='m1.small')]
keypairs = [FakeKeypair(name='testkey', public_key='ssh-rsa XXXX')] self.groups = {}
self.keypairs = [FakeKeypair(name='testkey',
public_key='ssh-rsa XXXX')]
def keypair_list(self): def keypair_list(self):
return self.keypairs return self.keypairs
@ -145,7 +148,8 @@ class FakeNovaManager(object):
class FakeCinderManager(object): class FakeCinderManager(object):
volumes = [FakeVolume(), ] def __init__(self):
self.volumes = [FakeVolume(), ]
def volume_list(self): def volume_list(self):
return self.volumes return self.volumes
@ -266,9 +270,9 @@ class StackDataTests(base.TestCase):
'mac_address': 'fa:16:3e:4b:8c:98', 'mac_address': 'fa:16:3e:4b:8c:98',
'fixed_ips': [{'subnet_id': '1111', 'fixed_ips': [{'subnet_id': '1111',
'ip_address': '10.123.2.3'}], 'ip_address': '10.123.2.3'}],
'id': '1234567', 'id': '1234567',
'security_groups': [], 'security_groups': [],
'device_id': '1234'}, ] 'device_id': '1234'}, ]
fake.subnets = [{'name': 'subnet_1111', fake.subnets = [{'name': 'subnet_1111',
'enable_dhcp': True, 'enable_dhcp': True,
'network_id': '1234', 'network_id': '1234',
@ -313,7 +317,6 @@ class StackDataTests(base.TestCase):
self.assertEqual(expected, generator.stack_data) self.assertEqual(expected, generator.stack_data)
def test_network(self): def test_network(self):
fake = FakeNeutronManager()
self.mock_neutron.return_value = FakeNeutronManager() self.mock_neutron.return_value = FakeNeutronManager()
generator = TemplateGenerator('x', 'x', 'x', 'x', True) generator = TemplateGenerator('x', 'x', 'x', 'x', True)
generator.extract_vm_details(False, False, True) generator.extract_vm_details(False, False, True)
@ -422,18 +425,16 @@ class StackDataTests(base.TestCase):
self.assertEqual(expected, generator.stack_data) self.assertEqual(expected, generator.stack_data)
def test_security_group(self): def test_security_group(self):
rules = [{ rules = [{'remote_group_id': None,
'remote_group_id': None, 'direction': 'ingress',
'direction': 'ingress', 'remote_ip_prefix': '0.0.0.0/0',
'remote_ip_prefix': '0.0.0.0/0', 'protocol': 'tcp',
'protocol': 'tcp', 'ethertype': 'IPv4',
'ethertype': 'IPv4', 'tenant_id': '7777',
'tenant_id': '7777', 'port_range_max': 22,
'port_range_max': 22, 'port_range_min': 22,
'port_range_min': 22, 'id': '8901',
'id': '8901', 'security_group_id': '1234'}, ]
'security_group_id': '1234'
}, ]
fake = FakeNeutronManager() fake = FakeNeutronManager()
fake.groups = [{'tenant_id': '7777', fake.groups = [{'tenant_id': '7777',
'name': 'somename', 'name': 'somename',
@ -464,18 +465,16 @@ class StackDataTests(base.TestCase):
self.assertEqual(expected, generator.stack_data) self.assertEqual(expected, generator.stack_data)
def test_default_security_group(self): def test_default_security_group(self):
rules = [{ rules = [{'remote_group_id': None,
'remote_group_id': None, 'direction': 'ingress',
'direction': 'ingress', 'remote_ip_prefix': '0.0.0.0/0',
'remote_ip_prefix': '0.0.0.0/0', 'protocol': 'tcp',
'protocol': 'tcp', 'ethertype': 'IPv4',
'ethertype': 'IPv4', 'tenant_id': '7777',
'tenant_id': '7777', 'port_range_max': 22,
'port_range_max': 22, 'port_range_min': 22,
'port_range_min': 22, 'id': '8901',
'id': '8901', 'security_group_id': '1234'}, ]
'security_group_id': '1234'
}, ]
fake = FakeNeutronManager() fake = FakeNeutronManager()
fake.groups = [{'tenant_id': '7777', fake.groups = [{'tenant_id': '7777',
'name': 'default', 'name': 'default',
@ -735,9 +734,9 @@ class NetworkTests(base.TestCase):
'mac_address': 'fa:16:3e:4b:8c:98', 'mac_address': 'fa:16:3e:4b:8c:98',
'fixed_ips': [{'subnet_id': '1111', 'fixed_ips': [{'subnet_id': '1111',
'ip_address': '10.123.2.3'}], 'ip_address': '10.123.2.3'}],
'id': '1234567', 'id': '1234567',
'security_groups': [], 'security_groups': [],
'device_id': '1234'}, ] 'device_id': '1234'}, ]
fake.subnets = [{'name': 'subnet_1111', fake.subnets = [{'name': 'subnet_1111',
'enable_dhcp': True, 'enable_dhcp': True,
'network_id': '1234', 'network_id': '1234',
@ -779,7 +778,6 @@ class NetworkTests(base.TestCase):
self.assertEqual(expected, generator.template) self.assertEqual(expected, generator.template)
def test_network(self): def test_network(self):
fake = FakeNeutronManager()
self.mock_neutron.return_value = FakeNeutronManager() self.mock_neutron.return_value = FakeNeutronManager()
generator = TemplateGenerator('x', 'x', 'x', 'x', True) generator = TemplateGenerator('x', 'x', 'x', 'x', True)
generator.extract_vm_details(False, False, True) generator.extract_vm_details(False, False, True)
@ -1231,12 +1229,11 @@ class NetworkTests(base.TestCase):
'description': 'security_group_1', 'description': 'security_group_1',
'security_group_rules': rules1, 'security_group_rules': rules1,
'id': '1111'}, 'id': '1111'},
{'tenant_id': '7777', {'tenant_id': '7777',
'name': 'security_group_2', 'name': 'security_group_2',
'description': 'security_group_2', 'description': 'security_group_2',
'security_group_rules': rules2, 'security_group_rules': rules2,
'id': '2222'}, 'id': '2222'}, ]
]
self.mock_neutron.return_value = fake self.mock_neutron.return_value = fake
generator = TemplateGenerator('x', 'x', 'x', 'x', True) generator = TemplateGenerator('x', 'x', 'x', 'x', True)

View File

@ -2,7 +2,7 @@ pbr>=0.6,!=0.7,<1.0
Babel>=1.3 Babel>=1.3
argparse argparse
ipaddr netaddr>=0.7.12
python-keystoneclient python-keystoneclient
python-neutronclient python-neutronclient
python-novaclient python-novaclient