From 925e9a0fd3d337eb050c1eca5d4a4b7475394b4a Mon Sep 17 00:00:00 2001
From: Tatiana <43054051+tacoelho@users.noreply.github.com>
Date: Fri, 16 Oct 2020 20:06:33 -0300
Subject: [PATCH] Update dataproc.py

Replace googleapiclient with dataproc python package #15
---
 gcloud_utils/dataproc.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/gcloud_utils/dataproc.py b/gcloud_utils/dataproc.py
index 24fe84b..2d2f93c 100644
--- a/gcloud_utils/dataproc.py
+++ b/gcloud_utils/dataproc.py
@@ -7,7 +7,7 @@
 import datetime
 import os
 import re
-from googleapiclient import discovery
+from google.cloud import dataproc
 
 logging.basicConfig(level=logging.INFO,
                     format='%(asctime)s - %(levelname)s - %(name)s - %(message)s')
@@ -17,9 +17,9 @@ class Dataproc(object):
     """Module to handle with Dataproc cluster"""
 
     def __init__(self, project, region, http=None):
-        self.__project = project
+        self.__project = project_id
         self.__region = region
-        self.__logger = logging.getLogger(name=self.__class__.__name__)
+        self.__logger = logging.getLogger(cluster_name=self.__class__.__name__)
         self.__client = discovery.build('dataproc', 'v1', http=http)
 
         self.__pattern = re.compile(r'[\W_]+')