Added support for timescaledb

agnostic/__init__.py

@@ -117,6 +117,22 @@ def askpass(user, db):
                 raise
         return PostgresBackend(host, port, user, password, database, schema)
 
+    elif db_type == 'timescale':
+        if user is None or database is None:
+            raise RuntimeError('TimescaleDB requires user and database '
+                'arguments.')
+        host = host or 'localhost'
+        password = password or askpass(user, database)
+        try:
+            from agnostic.timescale import TimescaleBackend
+        except ImportError as ie: # pragma: no cover
+            if ie.name == 'pg8000':
+                raise RuntimeError('The `pg8000` module is required for '
+                    ' TimescaleDB.')
+            else:
+                raise
+        return TimescaleBackend(host, port, user, password, database, schema)
+
     elif db_type == 'sqlite':
         if (host is not None or port is not None or user is not None or
             password is not None or schema is not None):

agnostic/cli.py

@@ -28,7 +28,7 @@ def __init__(self):
     envvar='AGNOSTIC_TYPE',
     metavar='<type>',
     required=True,
-    type=click.Choice(['mysql', 'postgres', 'sqlite']),
+    type=click.Choice(['mysql', 'postgres', 'sqlite', 'timescale']),
     help='Type of database.'
 )
 @click.option(

agnostic/timescale.py

@@ -0,0 +1,20 @@
+import os
+import subprocess
+
+import pg8000
+
+from agnostic.postgres import PostgresBackend
+
+
+class TimescaleBackend(PostgresBackend):
+
+    def clear_db(self, cursor):
+
+        ''' Remove timescale extension. '''
+
+        # Drop extension
+        cursor.execute('''
+            DROP EXTENSION IF EXISTS timescaledb CASCADE;
+        ''')
+
+        return super().clear_db(cursor)