Use logger with debug level set

tylertreat-wf · tylertreat-wf · commit a66cf27dc364 · 2014-05-07T14:24:29.000-05:00
diff --git a/bigquery/__init__.py b/bigquery/__init__.py
@@ -0,0 +1,6 @@
+import logging
+logging.basicConfig()
+
+logger = logging.getLogger('bigquery')
+logger.setLevel(logging.DEBUG)
+
diff --git a/bigquery/client.py b/bigquery/client.py
@@ -1,12 +1,13 @@
 import calendar
 from collections import defaultdict
 from datetime import datetime
-import logging
 
 from apiclient.discovery import build
 import httplib2
 from oauth2client.client import SignedJwtAssertionCredentials
 
+from bigquery import logger
+
 
 BIGQUERY_SCOPE = 'https://www.googleapis.com/auth/bigquery'
 BIGQUERY_SCOPE_READ_ONLY = 'https://www.googleapis.com/auth/bigquery.readonly'
@@ -81,7 +82,7 @@ def query(self, query, max_results=None, timeout=10):
             job id and query rows if query completed.
         """
 
-        logging.debug('Executing query: %s' % query)
+        logger.debug('Executing query: %s' % query)
 
         job_collection = self.bigquery.jobs()
         query_data = {'query': query, 'timeoutMs': timeout * 1000}
@@ -112,7 +113,7 @@ def get_query_schema(self, job_id):
             job_collection, self.project_id, job_id, offset=0, limit=0)
 
         if not query_reply['jobComplete']:
-            logging.warning('BigQuery job %s not complete' % job_id)
+            logger.warning('BigQuery job %s not complete' % job_id)
             return []
 
         return query_reply['schema']['fields']
@@ -153,7 +154,7 @@ def get_query_rows(self, job_id, offset=None, limit=None):
             limit=limit)
 
         if not query_reply['jobComplete']:
-            logging.warning('BigQuery job %s not complete' % job_id)
+            logger.warning('BigQuery job %s not complete' % job_id)
             return []
 
         schema = query_reply['schema']['fields']
@@ -211,7 +212,7 @@ def create_table(self, dataset, table, schema):
             return True
 
         except:
-            logging.error('Cannot create table %s.%s' % (dataset, table))
+            logger.error('Cannot create table %s.%s' % (dataset, table))
             return False
 
     def delete_table(self, dataset, table):
@@ -234,7 +235,7 @@ def delete_table(self, dataset, table):
             return True
 
         except:
-            logging.error('Cannot delete table %s.%s' % (dataset, table))
+            logger.error('Cannot delete table %s.%s' % (dataset, table))
             return False
 
     def get_tables(self, dataset_id, app_id, start_time, end_time):
@@ -287,13 +288,13 @@ def push_rows(self, rows, insert_id_key, dataset, table):
             ).execute()
 
             if response.get('insertErrors'):
-                logging.error('BigQuery insert errors: %s' % response)
+                logger.error('BigQuery insert errors: %s' % response)
                 return False
 
             return True
 
         except:
-            logging.error('Problem with BigQuery insertAll')
+            logger.error('Problem with BigQuery insertAll')
             return False
 
     def _get_all_tables(self, dataset_id):
diff --git a/bigquery/query_builder.py b/bigquery/query_builder.py
@@ -1,4 +1,4 @@
-import logging
+from bigquery import logger
 
 
 def render_query(dataset, tables, select=None, conditions=None,
@@ -162,7 +162,7 @@ def _render_conditions(conditions):
         comparators = condition.get('comparators')
 
         if None in (field, field_type, comparators) or not comparators:
-            logging.warn('Invalid condition passed in: %s' % condition)
+            logger.warn('Invalid condition passed in: %s' % condition)
             continue
 
         rendered_conditions.append(
