Upgrade google.cloud.monitoring

google/datalab/stackdriver/monitoring/__init__.py

@@ -14,7 +14,11 @@
 
 from __future__ import absolute_import
 
-from google.cloud.monitoring import Aligner, Reducer
+from google.cloud.monitoring import enums
+
+Aligner = enums.Aggregation.Aligner
+Reducer = enums.Aggregation.Reducer
+
 from ._group import Groups
 from ._metric import MetricDescriptors
 from ._query import Query

google/datalab/stackdriver/monitoring/_query.py

@@ -14,17 +14,17 @@
 
 from __future__ import absolute_import
 
-import google.cloud.monitoring
+import google.cloud.monitoring_v3.query
 
 from . import _query_metadata
 from . import _utils
 
 
-class Query(google.cloud.monitoring.Query):
+class Query(google.cloud.monitoring_v3.query.Query):
   """Query object for retrieving metric data."""
 
   def __init__(self,
-               metric_type=google.cloud.monitoring.Query.DEFAULT_METRIC_TYPE,
+               metric_type=google.cloud.monitoring_v3.query.Query.DEFAULT_METRIC_TYPE,
                end_time=None, days=0, hours=0, minutes=0, context=None):
     """Initializes the core query parameters.
 
@@ -33,13 +33,13 @@ def __init__(self,
     the resulting duration from the end time.
 
     It is also allowed to omit the end time and duration here,
-    in which case :meth:`~google.cloud.monitoring.query.Query.select_interval`
+    in which case :meth:`~google.cloud.monitoring_v3.query.Query.select_interval`
     must be called before the query is executed.
 
     Args:
       metric_type: The metric type name. The default value is
           :data:`Query.DEFAULT_METRIC_TYPE
-          <google.cloud.monitoring.query.Query.DEFAULT_METRIC_TYPE>`, but
+          <google.cloud.monitoring_v3.query.Query.DEFAULT_METRIC_TYPE>`, but
           please note that this default value is provided only for
           demonstration purposes and is subject to change.
       end_time: The end time (inclusive) of the time interval for which
@@ -54,7 +54,7 @@ def __init__(self,
         ValueError: ``end_time`` was specified but ``days``, ``hours``, and
             ``minutes`` are all zero. If you really want to specify a point in
             time, use
-            :meth:`~google.cloud.monitoring.query.Query.select_interval`.
+            :meth:`~google.cloud.monitoring_v3.query.Query.select_interval`.
     """
     client = _utils.make_client(context)
     super(Query, self).__init__(client, metric_type,

google/datalab/stackdriver/monitoring/_query_metadata.py

@@ -17,7 +17,7 @@
 from __future__ import unicode_literals
 from builtins import object
 
-import google.cloud.monitoring
+from google.cloud.monitoring_v3 import _dataframe
 import pandas
 
 
@@ -78,7 +78,7 @@ def as_dataframe(self, max_rows=None):
            for col in dataframe.columns])
 
       # Re-order the columns.
-      resource_keys = google.cloud.monitoring._dataframe._sorted_resource_labels(
+      resource_keys = _dataframe._sorted_resource_labels(
           dataframe['resource.labels'].columns)
       sorted_columns = [('resource.type', '')]
       sorted_columns += [('resource.labels', key) for key in resource_keys]

google/datalab/stackdriver/monitoring/_utils.py

@@ -14,14 +14,14 @@
 
 from __future__ import absolute_import
 
-import google.cloud.monitoring
+from google.cloud.logging.client import Client
 
 import google.datalab
 
 
 def make_client(context=None):
   context = context or google.datalab.Context.default()
-  client = google.cloud.monitoring.Client(
+  client = Client(
       project=context.project_id,
       credentials=context.credentials,
   )

setup.py

@@ -93,7 +93,8 @@
     'configparser>=3.5.0',
     'mock>=2.0.0',
     'future>=0.16.0',
-    'google-cloud>=0.30.0',
+    'google-cloud-logging>=1.6.0',
+    'google-cloud-monitoring>=0.30.1',
     'google-api-python-client>=1.6.2',
     'seaborn>=0.7.0',
     'plotly>=1.12.5',

tests/stackdriver/monitoring/metric_tests.py

@@ -14,7 +14,7 @@
 import mock
 import unittest
 
-import google.cloud.monitoring
+import google.cloud.logging.client
 
 import google.auth
 import google.datalab
@@ -70,7 +70,7 @@ def test_constructor_maximal(self):
     self.assertEqual(descriptors._type_prefix, TYPE_PREFIX)
     self.assertIsNone(descriptors._descriptors)
 
-  @mock.patch('google.cloud.monitoring.Client.list_metric_descriptors')
+  @mock.patch('google.cloud.logging.client.Client.list_metric_descriptors')
   def test_list(self, mock_gcloud_list_descriptors):
     mock_gcloud_list_descriptors.return_value = self._list_metrics_get_result(
         context=self.context)
@@ -83,7 +83,7 @@ def test_list(self, mock_gcloud_list_descriptors):
     self.assertEqual(metric_descriptor_list[0].type, METRIC_TYPES[0])
     self.assertEqual(metric_descriptor_list[1].type, METRIC_TYPES[1])
 
-  @mock.patch('google.cloud.monitoring.Client.list_metric_descriptors')
+  @mock.patch('google.cloud.logging.client.Client.list_metric_descriptors')
   def test_list_w_api_filter(self, mock_gcloud_list_descriptors):
     mock_gcloud_list_descriptors.return_value = self._list_metrics_get_result(
         context=self.context)
@@ -99,7 +99,7 @@ def test_list_w_api_filter(self, mock_gcloud_list_descriptors):
     self.assertEqual(metric_descriptor_list[0].type, METRIC_TYPES[0])
     self.assertEqual(metric_descriptor_list[1].type, METRIC_TYPES[1])
 
-  @mock.patch('google.cloud.monitoring.Client.list_metric_descriptors')
+  @mock.patch('google.cloud.logging.client.Client.list_metric_descriptors')
   def test_list_w_pattern_match(self, mock_gcloud_list_descriptors):
     mock_gcloud_list_descriptors.return_value = self._list_metrics_get_result(
         context=self.context)
@@ -111,7 +111,7 @@ def test_list_w_pattern_match(self, mock_gcloud_list_descriptors):
     self.assertEqual(len(metric_descriptor_list), 1)
     self.assertEqual(metric_descriptor_list[0].type, METRIC_TYPES[1])
 
-  @mock.patch('google.cloud.monitoring.Client.list_metric_descriptors')
+  @mock.patch('google.cloud.logging.client.Client.list_metric_descriptors')
   def test_list_caching(self, mock_gcloud_list_descriptors):
     mock_gcloud_list_descriptors.return_value = self._list_metrics_get_result(
         context=self.context)

tests/stackdriver/monitoring/query_metadata_tests.py

@@ -14,9 +14,9 @@
 import mock
 import unittest
 
-from google.cloud.monitoring import Resource
-from google.cloud.monitoring import Metric
-from google.cloud.monitoring import TimeSeries
+from google.cloud.logging.resource import Resource
+from google.cloud.monitoring_v3.types import Metric
+from google.cloud.monitoring_v3.types import TimeSeries
 
 import google.auth
 import google.datalab

tests/stackdriver/monitoring/query_tests.py

@@ -15,7 +15,7 @@
 import mock
 import unittest
 
-from google.cloud.monitoring import Query as BaseQuery
+from google.cloud.monitoring_v3.query import Query as BaseQuery
 
 import google.auth
 import google.datalab