ceph-volume: add type annotations to devices.lvm.create

This commit adds the Python type annotations to `devices.lvm.create`.

Signed-off-by: Guillaume Abrioux <[email protected]>

Author: guits

src/ceph-volume/ceph_volume/devices/lvm/create.py

@@ -1,10 +1,12 @@
 from __future__ import print_function
 from textwrap import dedent
 import logging
+import argparse
 from ceph_volume.util import system
 from ceph_volume.util.arg_validators import exclude_group_options
 from ceph_volume import decorators, terminal, objectstore
 from .common import create_parser, rollback_osd
+from typing import List, Optional
 
 logger = logging.getLogger(__name__)
 
@@ -13,31 +15,34 @@ class Create(object):
 
     help = 'Create a new OSD from an LVM device'
 
-    def __init__(self, argv, args=None):
-        self.objectstore = None
+    def __init__(self, argv: Optional[List[str]] = None,
+                 args: Optional[argparse.Namespace] = None) -> None:
+        self.objectstore: Optional[objectstore.baseobjectstore.BaseObjectStore] = None
         self.argv = argv
         self.args = args
 
     @decorators.needs_root
-    def create(self):
-        if not self.args.osd_fsid:
-            self.args.osd_fsid = system.generate_uuid()
-        self.objectstore = objectstore.mapping['LVM'][self.args.objectstore](args=self.args)
-        self.objectstore.safe_prepare()
-        osd_id = self.objectstore.osd_id
-        try:
-            # we try this for activate only when 'creating' an OSD, because a rollback should not
-            # happen when doing normal activation. For example when starting an OSD, systemd will call
-            # activate, which would never need to be rolled back.
-            self.objectstore.activate()
-        except Exception:
-            logger.exception('lvm activate was unable to complete, while creating the OSD')
-            logger.info('will rollback OSD ID creation')
-            rollback_osd(self.args, osd_id)
-            raise
-        terminal.success("ceph-volume lvm create successful for: %s" % self.args.data)
+    def create(self) -> None:
+        if self.args is not None:
+            if not self.args.osd_fsid:
+                self.args.osd_fsid = system.generate_uuid()
+            self.objectstore = objectstore.mapping['LVM'][self.args.objectstore](args=self.args)
+            if self.objectstore is not None:
+                self.objectstore.safe_prepare()
+                osd_id = self.objectstore.osd_id
+                try:
+                    # we try this for activate only when 'creating' an OSD, because a rollback should not
+                    # happen when doing normal activation. For example when starting an OSD, systemd will call
+                    # activate, which would never need to be rolled back.
+                    self.objectstore.activate()
+                except Exception:
+                    logger.exception('lvm activate was unable to complete, while creating the OSD')
+                    logger.info('will rollback OSD ID creation')
+                    rollback_osd(self.args, osd_id)
+                    raise
+                terminal.success("ceph-volume lvm create successful for: %s" % self.args.data)
 
-    def main(self):
+    def main(self) -> None:
         sub_command_help = dedent("""
         Create an OSD by assigning an ID and FSID, registering them with the
         cluster with an ID and FSID, formatting and mounting the volume, adding
@@ -65,6 +70,8 @@ def main(self):
             prog='ceph-volume lvm create',
             description=sub_command_help,
         )
+        if self.argv is None:
+            self.argv = []
         if len(self.argv) == 0:
             print(sub_command_help)
             return