commands/mkinitramfs.py: Add make_parser

Following the same pattern of run.py.

Signed-off-by: Marcos Paulo de Souza <marcos.souza.org@gmail.com>

1 files changed, 13 insertions, 6 deletions

diff --git a/virtme/commands/mkinitramfs.py b/virtme/commands/mkinitramfs.py
index b7c602e..d2874ab 100755
--- a/virtme/commands/mkinitramfs.py
+++ b/virtme/commands/mkinitramfs.py
@@ -10,16 +10,23 @@ from .. import modfinder
 from .. import virtmods
 from .. import mkinitramfs
 
-_ARGPARSER = argparse.ArgumentParser(
-    description='Generate initramfs image for virtme',
-)
-_ARGPARSER.add_argument('--mod-kversion', action='store', default=None)
-_ARGPARSER.add_argument('--rw', action='store_true', default=False)
+def make_parser():
+    parser = argparse.ArgumentParser(
+        description='Generate an initramfs image for virtme',
+    )
+
+    parser.add_argument('--mod-kversion', action='store', default=None,
+                        help='Find kernel modules related to kernel version set')
+
+    parser.add_argument('--rw', action='store_true', default=False,
+                        help='Mount initramfs as rw. Default is ro')
+
+    return parser
 
 def main():
     import sys
 
-    args = _ARGPARSER.parse_args()
+    args = make_parser().parse_args()
 
     config = mkinitramfs.Config()