Added multi-tenancy support

Change-Id: I89e75d4dd36410ab7e24f81261ad8703dab11297

cloudkitty/cli/writer.py

@@ -15,6 +15,8 @@
 #
 # @author: Stéphane Albert
 #
+from __future__ import print_function
+
 from oslo.config import cfg
 from stevedore import driver
 
@@ -44,14 +46,75 @@ def load_output_backend():
     return backend
 
 
+class DBCommand(object):
+
+    def __init__(self):
+        self._storage = None
+        self._output = None
+        self._load_storage_backend()
+        self._load_output_backend()
+
+    def _load_storage_backend(self):
+        storage_args = {'period': CONF.collect.period}
+        CONF.import_opt('backend', 'cloudkitty.storage', 'storage')
+        backend = driver.DriverManager(
+            STORAGES_NAMESPACE,
+            CONF.storage.backend,
+            invoke_on_load=True,
+            invoke_kwds=storage_args).driver
+        self._storage = backend
+
+    def _load_output_backend(self):
+        CONF.import_opt('backend', 'cloudkitty.config', 'output')
+        backend = i_utils.import_class(CONF.output.backend)
+        self._output = backend
+
+    def generate(self):
+        if not CONF.command.tenant:
+            tenants = self._storage.get_tenants(CONF.command.begin,
+                                                CONF.command.end)
+        else:
+            tenants = [CONF.command.tenant]
+        for tenant in tenants:
+            wo = write_orchestrator.WriteOrchestrator(self._output,
+                                                      tenant,
+                                                      self._storage)
+            wo.init_writing_pipeline()
+            if not CONF.command.begin:
+                wo.restart_month()
+            wo.process()
+
+    def tenants_list(self):
+        tenants = self._storage.get_tenants(CONF.command.begin,
+                                            CONF.command.end)
+        print('Tenant list:')
+        for tenant in tenants:
+            print(tenant)
+
+
+def add_command_parsers(subparsers):
+    command_object = DBCommand()
+
+    parser = subparsers.add_parser('generate')
+    parser.set_defaults(func=command_object.generate)
+    parser.add_argument('--tenant', nargs='?')
+    parser.add_argument('--begin', nargs='?')
+    parser.add_argument('--end', nargs='?')
+
+    parser = subparsers.add_parser('tenants_list')
+    parser.set_defaults(func=command_object.tenants_list)
+    parser.add_argument('--begin', nargs='?')
+    parser.add_argument('--end', nargs='?')
+
+
+command_opt = cfg.SubCommandOpt('command',
+                                title='Command',
+                                help='Available commands',
+                                handler=add_command_parsers)
+
+CONF.register_cli_opt(command_opt)
+
+
 def main():
     service.prepare_service()
-    output_backend = load_output_backend()
-    storage_backend = load_storage_backend()
-
-    wo = write_orchestrator.WriteOrchestrator(output_backend,
-                                              'writer',
-                                              storage_backend)
-    wo.init_writing_pipeline()
-    wo.restart_month()
-    wo.process()
+    CONF.command.func()