lint

Author: gsheni

download_analytics/bq.py

@@ -6,7 +6,6 @@
 import logging
 import os
 import pathlib
-import pandas as pd
 
 from google.cloud import bigquery
 from google.oauth2 import service_account
@@ -51,7 +50,8 @@ def run_query(query, dry_run=False, credentials_file=None):
 
     job_config = bigquery.QueryJobConfig(dry_run=True, use_query_cache=False)
     dry_run_job = client.query(query, job_config=job_config)
-    LOGGER.info('Estimated data processed in query (GBs): %.2f', dry_run_job.total_bytes_processed / 1024**3)
+    data_processed_gbs = dry_run_job.total_bytes_processed / 1024**3
+    LOGGER.info('Estimated data processed in query (GBs): %.2f', data_processed_gbs)
     # https://cloud.google.com/bigquery/pricing#on_demand_pricing
     # assuming have hit 1 terabyte processed in month
     cost_per_terabyte = 6.15
@@ -70,11 +70,17 @@ def run_query(query, dry_run=False, credentials_file=None):
     LOGGER.info('Total cost for query: $%.2f', cost)
     return data
 
+
 def bytes_to_megabytes(bytes):
+    """Convert bytes to megabytes."""
     return bytes / 1024 / 1024
 
+
 def bytes_to_gigabytes(bytes):
+    """Convert bytes to gigabytes."""
     return bytes_to_megabytes(bytes) / 1024
 
+
 def bytes_to_terabytes(bytes):
+    """Convert bytes to terabytes."""
     return bytes_to_gigabytes(bytes) / 1024