ENH: query price presentation (uses pyfiglet fonts)

Author: Piotr Chromiec

pandas_gbq/gbq.py

@@ -7,6 +7,9 @@
 import sys
 
 import numpy as np
+from pyfiglet import Figlet
+from collections import OrderedDict
+
 
 from distutils.version import StrictVersion
 from pandas import compat, DataFrame, concat
@@ -201,6 +204,21 @@ def __init__(self, project_id, reauth=False, verbose=False,
         self.dialect = dialect
         self.credentials = self.get_credentials()
         self.service = self.get_service()
+        # BQ Queries costs $5 per TB. First 1 TB per month is free
+        # see here for more: https://cloud.google.com/bigquery/pricing
+        self.query_price_for_TB = 5. # USD
+
+        # pyfiglet stuff for price output
+        self.__figlets = OrderedDict([
+            (0.1, Figlet(font='straight')),
+            (0.5, Figlet(font='starwars')),
+            (1.0, Figlet(font='blocks')),
+            (2.0, Figlet(font='fraktur')),
+            (5.0, Figlet(font='doh'))])
+
+
+        for k in self.__figlets:
+            self.__figlets[k].width=120
 
     def get_credentials(self):
         if self.private_key:
@@ -421,6 +439,17 @@ def sizeof_fmt(num, suffix='B'):
             num /= 1024.0
         return fmt % (num, 'Y', suffix)
 
+    def query_price_for(self, bytes_num):
+        figlet = None
+        price = bytes_num * self.query_price_for_TB / 2**40
+
+        for (k,v) in self.__figlets.items():
+            if price > k:
+                figlet = v
+
+        text_price = '$ {0:.2f}'.format(price)
+        return figlet and '\n' + figlet.renderText(text_price) or text_price
+
     def get_service(self):
         import httplib2
         from google_auth_httplib2 import AuthorizedHttp
@@ -543,8 +572,9 @@ def run_query(self, query, **kwargs):
             else:
                 bytes_processed = int(query_reply.get(
                     'totalBytesProcessed', '0'))
-                self._print('Query done.\nProcessed: {}\n'.format(
-                    self.sizeof_fmt(bytes_processed)))
+                self._print('Query done.\nProcessed: {}\nPrice: {}'.format(
+                    self.sizeof_fmt(bytes_processed),
+                    self.query_price_for(bytes_processed)))
 
             self._print('Retrieving results...')