Split 'sources' module into files

Improve readability by spliting the 'sources' module into separate
files. Each file contains only one class.

In addition update the mock statements in the unit tests to match these
changes.

Add recursive-include in MANIFEST.in to include virtBootstrap.sources
module.

Update the unit tests to match these changes.

3 files changed, 108 insertions, 50 deletions

diff --git a/src/virtBootstrap/sources/__init__.py b/src/virtBootstrap/sources/__init__.py
new file mode 100644
index 0000000..e891e9b
--- /dev/null
+++ b/src/virtBootstrap/sources/__init__.py
@@ -0,0 +1,26 @@
+"""
+sources - Class definitions which process container image or
+          tarball to extract the root file system in destination
+          directory or qcow2 image.
+
+ Authors:
+    Radostin Stoyanov <rstoyanov1@gmail.com>
+
+ Copyright (c) 2017 Radostin Stoyanov
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+"""
+
+from virtBootstrap.sources.file_source import FileSource
+from virtBootstrap.sources.docker_source import DockerSource
diff --git a/src/virtBootstrap/sources.py b/src/virtBootstrap/sources/docker_source.py
index 40b66f9..54d8903 100644
--- a/src/virtBootstrap/sources.py
+++ b/src/virtBootstrap/sources/docker_source.py
@@ -16,8 +16,9 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
 """
-Class definitions which process container image or
-archive from source and unpack them in destination directory.
+DockerSource aim is to download container image from Docker registry and
+extract the layers of root file system into destination directory or qcow2
+image with backing chains.
 """
 
 import select
@@ -35,54 +36,6 @@ from virtBootstrap import utils
 logger = logging.getLogger(__name__)
 
 
-class FileSource(object):
-    """
-    Extract root filesystem from file.
-    """
-    def __init__(self, **kwargs):
-        """
-        Bootstrap root filesystem from tarball
-
-        @param uri: Path to tar archive file.
-        @param fmt: Format used to store image [dir, qcow2]
-        @param progress: Instance of the progress module
-        """
-        self.path = kwargs['uri'].path
-        self.output_format = kwargs.get('fmt', utils.DEFAULT_OUTPUT_FORMAT)
-        self.progress = kwargs['progress'].update_progress
-
-    def unpack(self, dest):
-        """
-        Safely extract root filesystem from tarball
-
-        @param dest: Directory path where the files to be extraced
-        """
-
-        if not os.path.isfile(self.path):
-            raise Exception('Invalid file source "%s"' % self.path)
-
-        if self.output_format == 'dir':
-            self.progress("Extracting files into destination directory",
-                          value=0, logger=logger)
-            utils.safe_untar(self.path, dest)
-
-        elif self.output_format == 'qcow2':
-            # Remove the old path
-            file_name = os.path.basename(self.path)
-            qcow2_file = os.path.realpath('{}/{}.qcow2'.format(dest,
-                                                               file_name))
-
-            self.progress("Extracting files into qcow2 image", value=0,
-                          logger=logger)
-            utils.create_qcow2(self.path, qcow2_file)
-        else:
-            raise Exception("Unknown format:" + self.output_format)
-
-        self.progress("Extraction completed successfully!", value=100,
-                      logger=logger)
-        logger.info("Files are stored in: " + dest)
-
-
 class DockerSource(object):
     """
     Extract files from Docker image
diff --git a/src/virtBootstrap/sources/file_source.py b/src/virtBootstrap/sources/file_source.py
new file mode 100644
index 0000000..c02f735
--- /dev/null
+++ b/src/virtBootstrap/sources/file_source.py
@@ -0,0 +1,79 @@
+# Authors: Cedric Bosdonnat <cbosdonnat@suse.com>
+#
+# Copyright (C) 2017 SUSE, Inc.
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+"""
+FileSource aim is to extract root filesystem from tar archive to destination
+directory or qcow2 image.
+"""
+
+import os
+import logging
+
+from virtBootstrap import utils
+
+
+# pylint: disable=invalid-name
+# Create logger
+logger = logging.getLogger(__name__)
+
+
+class FileSource(object):
+    """
+    Extract root filesystem from file.
+    """
+    def __init__(self, **kwargs):
+        """
+        Bootstrap root filesystem from tarball
+
+        @param uri: Path to tar archive file.
+        @param fmt: Format used to store image [dir, qcow2]
+        @param progress: Instance of the progress module
+        """
+        self.path = kwargs['uri'].path
+        self.output_format = kwargs.get('fmt', utils.DEFAULT_OUTPUT_FORMAT)
+        self.progress = kwargs['progress'].update_progress
+
+    def unpack(self, dest):
+        """
+        Safely extract root filesystem from tarball
+
+        @param dest: Directory path where the files to be extraced
+        """
+
+        if not os.path.isfile(self.path):
+            raise Exception('Invalid file source "%s"' % self.path)
+
+        if self.output_format == 'dir':
+            self.progress("Extracting files into destination directory",
+                          value=0, logger=logger)
+            utils.safe_untar(self.path, dest)
+
+        elif self.output_format == 'qcow2':
+            # Remove the old path
+            file_name = os.path.basename(self.path)
+            qcow2_file = os.path.realpath('{}/{}.qcow2'.format(dest,
+                                                               file_name))
+
+            self.progress("Extracting files into qcow2 image", value=0,
+                          logger=logger)
+            utils.create_qcow2(self.path, qcow2_file)
+        else:
+            raise Exception("Unknown format:" + self.output_format)
+
+        self.progress("Extraction completed successfully!", value=100,
+                      logger=logger)
+        logger.info("Files are stored in: " + dest)