Fireant 1.0 Rewrite

.travis.yml

@@ -1,7 +1,6 @@
 language: python
 
 python:
-    - "2.7"
     - "3.3"
     - "3.4"
     - "3.5"

fireant/__init__.py

@@ -1,2 +1,8 @@
-# coding: utf-8
-__version__ = '{major}.{minor}.{patch}'.format(major=0, minor=19, patch=3)
+# noinspection PyUnresolvedReferences
+from .database import *
+# noinspection PyUnresolvedReferences
+from .slicer import *
+# noinspection PyUnresolvedReferences
+from .slicer.widgets import *
+
+__version__ = '1.0.0.dev13'

fireant/database/__init__.py

@@ -1,8 +1,5 @@
-# coding: utf-8
-
-from .database import Database
+from .base import Database
 from .mysql import MySQLDatabase
-from .redshift import RedshiftDatabase
 from .postgresql import PostgreSQLDatabase
-from .vertica import (Vertica,
-                      VerticaDatabase)
+from .redshift import RedshiftDatabase
+from .vertica import VerticaDatabase

fireant/database/database.py → fireant/database/base.py

@@ -1,11 +1,15 @@
-# coding: utf-8
-
 import pandas as pd
 
-from pypika import Query
+from pypika import (
+    Query,
+    terms,
+)
 
 
 class Database(object):
+    """
+    WRITEME
+    """
     # The pypika query class to use for constructing queries
     query_cls = Query
 
@@ -15,7 +19,7 @@ def connect(self):
     def trunc_date(self, field, interval):
         raise NotImplementedError
 
-    def date_add(self, date_part, interval, field):
+    def date_add(self, field: terms.Term, date_part: str, interval: int):
         """ Database specific function for adding or subtracting dates """
         raise NotImplementedError
 
@@ -25,6 +29,9 @@ def fetch(self, query):
             cursor.execute(query)
             return cursor.fetchall()
 
-    def fetch_dataframe(self, query):
+    def fetch_data(self, query):
         with self.connect() as connection:
-            return pd.read_sql(query, connection)
+            return pd.read_sql(query, connection, coerce_float=True, parse_dates=True)
+
+    def totals(self, query, terms):
+        raise NotImplementedError

fireant/database/mysql.py

@@ -1,12 +1,11 @@
-# coding: utf-8
 import pandas as pd
+
 from pypika import (
     Dialects,
     MySQLQuery,
     terms,
 )
-
-from fireant.database import Database
+from .base import Database
 
 
 class Trunc(terms.Function):
@@ -51,25 +50,26 @@ def __init__(self, database=None, host='localhost', port=3306,
     def connect(self):
         import pymysql
 
-        return pymysql.connect(
-            host=self.host, port=self.port, db=self.database,
-            user=self.user, password=self.password,
-            charset=self.charset,
-            cursorclass=pymysql.cursors.Cursor,
-        )
+        return pymysql.connect(host=self.host, port=self.port, db=self.database,
+                               user=self.user, password=self.password,
+                               charset=self.charset,
+                               cursorclass=pymysql.cursors.Cursor)
 
     def fetch(self, query):
         with self.connect().cursor() as cursor:
             cursor.execute(query)
             return cursor.fetchall()
 
-    def fetch_dataframe(self, query):
+    def fetch_data(self, query):
         return pd.read_sql(query, self.connect())
 
     def trunc_date(self, field, interval):
-        return Trunc(field, interval)
+        return Trunc(field, str(interval))
 
-    def date_add(self, date_part, interval, field):
+    def date_add(self, field, date_part, interval):
         # adding an extra 's' as MySQL's interval doesn't work with 'year', 'week' etc, it expects a plural
         interval_term = terms.Interval(**{'{}s'.format(str(date_part)): interval, 'dialect': Dialects.MYSQL})
         return DateAdd(field, interval_term)
+
+    def totals(self, query, terms):
+        raise NotImplementedError

fireant/database/postgresql.py

@@ -1,21 +1,20 @@
-# coding: utf-8
 import pandas as pd
+
 from pypika import (
     PostgreSQLQuery,
     functions as fn,
     terms,
 )
-
-from fireant.database import Database
+from .base import Database
 
 
-class Trunc(terms.Function):
+class DateTrunc(terms.Function):
     """
     Wrapper for the PostgreSQL date_trunc function
     """
 
     def __init__(self, field, date_format, alias=None):
-        super(Trunc, self).__init__('date_trunc', date_format, field, alias=alias)
+        super(DateTrunc, self).__init__('DATE_TRUNC', date_format, field, alias=alias)
         # Setting the fields here means we can access the TRUNC args by name.
         self.field = field
         self.date_format = date_format
@@ -26,6 +25,7 @@ class PostgreSQLDatabase(Database):
     """
     PostgreSQL client that uses the psycopg module.
     """
+
     # The pypika query class to use for constructing queries
     query_cls = PostgreSQLQuery
 
@@ -40,21 +40,22 @@ def __init__(self, database=None, host='localhost', port=5432,
     def connect(self):
         import psycopg2
 
-        return psycopg2.connect(
-            host=self.host, port=self.port, dbname=self.database,
-            user=self.user, password=self.password,
-        )
+        return psycopg2.connect(host=self.host, port=self.port, dbname=self.database,
+                                user=self.user, password=self.password)
 
     def fetch(self, query):
         with self.connect().cursor() as cursor:
             cursor.execute(query)
             return cursor.fetchall()
 
-    def fetch_dataframe(self, query):
+    def fetch_data(self, query):
         return pd.read_sql(query, self.connect())
 
     def trunc_date(self, field, interval):
-        return Trunc(field, interval)
+        return DateTrunc(field, str(interval))
+
+    def date_add(self, field, date_part, interval):
+        return fn.DateAdd(str(date_part), interval, field)
 
-    def date_add(self, date_part, interval, field):
-        return fn.DateAdd(date_part, interval, field)
+    def totals(self, query, terms):
+        raise NotImplementedError