Refactor code

- Fix pep8 error H903: [1]
- Add method _update_task_state to reduce code lines.
- Change single quotes to double quotes in docstring - PEP8 docstring
convention [2].

[1] http://paste.openstack.org/show/614990/
[2] https://www.python.org/dev/peps/pep-0257/

Partials-Bug: #1702587

Change-Id: Idd125665a2a5f0a73337d36f0259a9f9107946ad

zun/api/controllers/v1/containers.py

@@ -172,13 +172,13 @@ class ContainersController(base.Controller):
         return view.format_container(pecan.request.host_url, container)
 
     def _generate_name_for_container(self):
-        '''Generate a random name like: zeta-22-container.'''
+        """Generate a random name like: zeta-22-container."""
         name_gen = name_generator.NameGenerator()
         name = name_gen.generate()
         return name + '-container'
 
     def _check_for_restart_policy(self, container_dict):
-        '''Check for restart policy input'''
+        """Check for restart policy input"""
         restart_policy = container_dict.get('restart_policy')
         if not restart_policy:
             return

zun/api/controllers/v1/images.py

@@ -52,7 +52,7 @@ class ImageCollection(collection.Collection):
 
 
 class ImagesController(base.Controller):
-    '''Controller for Images'''
+    """Controller for Images"""
 
     _custom_actions = {
         'search': ['GET']
@@ -61,7 +61,7 @@ class ImagesController(base.Controller):
     @pecan.expose('json')
     @exception.wrap_pecan_controller_exception
     def get_all(self, **kwargs):
-        '''Retrieve a list of images.'''
+        """Retrieve a list of images."""
         context = pecan.request.context
         policy.enforce(context, "image:get_all",
                        action="image:get_all")

zun/common/name_generator.py

@@ -23,10 +23,10 @@ class NameGenerator(object):
         self.random = Random()
 
     def generate(self):
-        '''Generate a random name compose of a Greek leter and
+        """Generate a random name compose of a Greek leter and
 
         a number, like: beta_2.
-        '''
+        """
 
         letter = self.random.choice(self.letters)
         number = self.random.randint(1, 24)

zun/common/utils.py

@@ -170,7 +170,7 @@ def translate_exception(function):
 
 
 def check_container_id(function):
-    '''Check container_id property of given container instance.'''
+    """Check container_id property of given container instance."""
 
     @functools.wraps(function)
     def decorated_function(*args, **kwargs):

zun/compute/manager.py

@@ -33,7 +33,7 @@ LOG = logging.getLogger(__name__)
 
 
 class Manager(object):
-    '''Manages the running containers.'''
+    """Manages the running containers."""
 
     def __init__(self, container_driver=None):
         super(Manager, self).__init__()
@@ -73,6 +73,10 @@ class Manager(object):
             LOG.error("Error occurred while deleting sandbox: %s",
                       six.text_type(e))
 
+    def _update_task_state(self, context, container, task_state):
+        container.task_state = task_state
+        container.save(context)
+
     def _do_container_create(self, context, container, requested_networks,
                              limits=None, reraise=False):
         LOG.debug('Creating container: %s', container.uuid)
@@ -86,9 +90,7 @@ class Manager(object):
             self._fail_container(self, context, container, msg)
             return
 
-        container.task_state = consts.SANDBOX_CREATING
+        self._update_task_state(context, container, consts.SANDBOX_CREATING)
-        container.save(context)
-        sandbox_id = None
         sandbox_image = CONF.sandbox_image
         sandbox_image_driver = CONF.sandbox_image_driver
         sandbox_image_pull_policy = CONF.sandbox_image_pull_policy
@@ -109,8 +111,7 @@ class Manager(object):
                 self._fail_container(context, container, six.text_type(e))
             return
 
-        container.task_state = consts.IMAGE_PULLING
+        self._update_task_state(context, container, consts.IMAGE_PULLING)
-        container.save(context)
         repo, tag = utils.parse_image_name(container.image)
         image_pull_policy = utils.get_image_pull_policy(
             container.image_pull_policy, tag)
@@ -151,8 +152,7 @@ class Manager(object):
                                     limits):
                 container = self.driver.create(context, container,
                                                sandbox_id, image)
-                container.task_state = None
+                self._update_task_state(context, container, None)
-                container.save(context)
                 return container
         except exception.DockerError as e:
             with excutils.save_and_reraise_exception(reraise=reraise):
@@ -173,12 +173,10 @@ class Manager(object):
 
     def _do_container_start(self, context, container, reraise=False):
         LOG.debug('Starting container: %s', container.uuid)
-        container.task_state = consts.CONTAINER_STARTING
+        self._update_task_state(context, container, consts.CONTAINER_STARTING)
-        container.save(context)
         try:
             container = self.driver.start(context, container)
-            container.task_state = None
+            self._update_task_state(context, container, None)
-            container.save(context)
             return container
         except exception.DockerError as e:
             with excutils.save_and_reraise_exception(reraise=reraise):
@@ -194,8 +192,7 @@ class Manager(object):
     @translate_exception
     def container_delete(self, context, container, force):
         LOG.debug('Deleting container: %s', container.uuid)
-        container.task_state = consts.CONTAINER_DELETING
+        self._update_task_state(context, container, consts.CONTAINER_DELETING)
-        container.save(context)
         reraise = not force
         try:
             self.driver.delete(container, force)
@@ -211,8 +208,8 @@ class Manager(object):
 
         sandbox_id = self.driver.get_sandbox_id(container)
         if sandbox_id:
-            container.task_state = consts.SANDBOX_DELETING
+            self._update_task_state(context, container,
-            container.save(context)
+                                    consts.SANDBOX_DELETING)
             try:
                 self.driver.delete_sandbox(context, container, sandbox_id)
             except Exception as e:
@@ -220,8 +217,7 @@ class Manager(object):
                     LOG.exception("Unexpected exception: %s",
                                   six.text_type(e))
                     self._fail_container(context, container, six.text_type(e))
-        container.task_state = None
+        self._update_task_state(context, container, None)
-        container.save(context)
         container.destroy(context)
         self._get_resource_tracker()
 
@@ -250,7 +246,7 @@ class Manager(object):
     def container_list(self, context):
         LOG.debug('Listing container...')
         try:
-            return self.driver.list()
+            return self.driver.list(context)
         except exception.DockerError as e:
             LOG.error("Error occurred while calling Docker list API: %s",
                       six.text_type(e))
@@ -277,12 +273,10 @@ class Manager(object):
 
     def _do_container_reboot(self, context, container, timeout, reraise=False):
         LOG.debug('Rebooting container: %s', container.uuid)
-        container.task_state = consts.CONTAINER_REBOOTING
+        self._update_task_state(context, container, consts.CONTAINER_REBOOTING)
-        container.save(context)
         try:
             container = self.driver.reboot(context, container, timeout)
-            container.task_state = None
+            self._update_task_state(context, container, None)
-            container.save(context)
             return container
         except exception.DockerError as e:
             with excutils.save_and_reraise_exception(reraise=reraise):
@@ -300,12 +294,10 @@ class Manager(object):
 
     def _do_container_stop(self, context, container, timeout, reraise=False):
         LOG.debug('Stopping container: %s', container.uuid)
-        container.task_state = consts.CONTAINER_STOPPING
+        self._update_task_state(context, container, consts.CONTAINER_STOPPING)
-        container.save(context)
         try:
             container = self.driver.stop(context, container, timeout)
-            container.task_state = None
+            self._update_task_state(context, container, None)
-            container.save(context)
             return container
         except exception.DockerError as e:
             with excutils.save_and_reraise_exception(reraise=reraise):
@@ -561,8 +553,6 @@ class Manager(object):
     def _do_container_commit(self, context, snapshot_image, container,
                              repository, tag=None):
         LOG.debug('Creating image...')
-        container_image = None
-        container_image_id = None
         if tag is None:
             tag = 'latest'
 

zun/compute/rpcapi.py

@@ -39,13 +39,13 @@ def check_container_host(func):
 
 @profiler.trace_cls("rpc")
 class API(rpc_service.API):
-    '''Client side of the container compute rpc API.
+    """Client side of the container compute rpc API.
 
     API version history:
 
         * 1.0 - Initial version.
         * 1.1 - Add image endpoints.
-    '''
+    """
 
     def __init__(self, transport=None, context=None, topic=None):
         if topic is None:

zun/container/docker/driver.py

@@ -71,7 +71,7 @@ def wrap_docker_error(function):
 
 
 class DockerDriver(driver.ContainerDriver):
-    '''Implementation of container drivers for Docker.'''
+    """Implementation of container drivers for Docker."""
 
     def __init__(self):
         super(DockerDriver, self).__init__()
@@ -758,7 +758,7 @@ class NovaDockerDriver(DockerDriver):
         return sandbox_id
 
     def _ensure_active(self, novaclient, server, timeout=300):
-        '''Wait until the Nova instance to become active.'''
+        """Wait until the Nova instance to become active."""
         def _check_active():
             return novaclient.check_active(server)
 
@@ -791,7 +791,7 @@ class NovaDockerDriver(DockerDriver):
         novaclient.stop_server(server_name)
 
     def _ensure_deleted(self, novaclient, server_id, timeout=300):
-        '''Wait until the Nova instance to be deleted.'''
+        """Wait until the Nova instance to be deleted."""
         def _check_delete_complete():
             return novaclient.check_delete_server_complete(server_id)
 

zun/container/driver.py

@@ -58,7 +58,7 @@ def load_container_driver(container_driver=None):
 
 
 class ContainerDriver(object):
-    '''Base class for container drivers.'''
+    """Base class for container drivers."""
 
     def create(self, context, container, sandbox_name=None):
         """Create a container."""
@@ -203,6 +203,9 @@ class ContainerDriver(object):
     def add_security_group(self, context, container, security_group, **kwargs):
         raise NotImplementedError()
 
+    def get_available_nodes(self):
+        pass
+
     def get_available_resources(self, node):
         numa_topo_obj = self.get_host_numa_topology()
         node.numa_topology = numa_topo_obj

zun/image/driver.py

@@ -136,7 +136,7 @@ def upload_image_data(context, image, image_tag, image_data,
 
 
 class ContainerImageDriver(object):
-    '''Base class for container image driver.'''
+    """Base class for container image driver."""
 
     def pull_image(self, context, repo, tag):
         """Pull an image."""

zun/servicegroup/zun_service_periodic.py

@@ -23,10 +23,10 @@ LOG = log.getLogger(__name__)
 
 
 class ZunServicePeriodicTasks(periodic_task.PeriodicTasks):
-    '''Zun periodic Task class
+    """Zun periodic Task class
 
     Any periodic task job need to be added into this class
-    '''
+    """
 
     def __init__(self, conf, binary):
         self.zun_service_ref = None

zun/tests/unit/compute/test_compute_manager.py

@@ -291,7 +291,7 @@ class TestManager(base.TestCase):
     @mock.patch.object(fake_driver, 'list')
     def test_container_list(self, mock_list):
         self.compute_manager.container_list(self.context)
-        mock_list.assert_called_once_with()
+        mock_list.assert_called_once_with(self.context)
 
     @mock.patch.object(fake_driver, 'list')
     def test_container_list_failed(self, mock_list):

zun/tests/unit/container/fake_driver.py

@@ -17,7 +17,7 @@ from zun.container import driver
 
 
 class FakeDriver(driver.ContainerDriver):
-    '''Fake driver for testing.'''
+    """Fake driver for testing."""
 
     def __init__(self):
         super(FakeDriver, self).__init__()

zun/tests/unit/db/utils.py

@@ -117,7 +117,7 @@ def create_test_image(**kwargs):
 
 
 def _generate_repo_for_image():
-    '''Generate a random name like: zeta-22-image.'''
+    """Generate a random name like: zeta-22-image."""
     name_gen = name_generator.NameGenerator()
     name = name_gen.generate()
     return name + '-image'

zun/websocket/websocketproxy.py

@@ -12,10 +12,10 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
-'''
+"""
 Websocket proxy that is compatible with OpenStack Zun.
 Leverages websockify.py by Joel Martin
-'''
+"""
 
 import errno
 import select
@@ -82,10 +82,10 @@ class ZunProxyRequestHandlerBase(object):
         return None
 
     def _handle_ins_outs(self, target, ins, outs):
-        '''Handle the select file ins and outs
+        """Handle the select file ins and outs
 
-            handle the operation ins and outs from select
+        Handle the operation ins and outs from select
-        '''
+        """
         if self.request in outs:
             # Send queued target data to the client
             self.c_pend = self.send_frames(self.cqueue)
@@ -122,10 +122,10 @@ class ZunProxyRequestHandlerBase(object):
             self.cqueue.append(buf)
 
     def do_proxy(self, target):
-        '''Proxy websocket link
+        """Proxy websocket link
 
         Proxy client WebSocket to normal target socket.
-        '''
+        """
         self.cqueue = []
         self.tqueue = []
         self.c_pend = 0