add parser

dbt/adapters/postgres.py

@@ -291,7 +291,7 @@ def rename(cls, profile, from_name, to_name, model_name=None):
 
  @classmethod
  def execute_model(cls, profile, model):
- parts = re.split(r'-- (DBT_OPERATION .*)', model.compiled_contents)
+ parts = re.split(r'-- (DBT_OPERATION .*)', model.get('wrapped_sql'))
  connection = cls.get_connection(profile)
 
  if flags.STRICT_MODE:
@@ -317,7 +317,7 @@ def call_expand_target_column_types(kwargs):
  func_map[function](kwargs)
  else:
  handle, cursor = cls.add_query_to_transaction(
- part, connection, model.name)
+ part, connection, model.get('name'))
 
  handle.commit()
 
@@ -504,6 +504,16 @@ def commit(cls, profile):
  handle = connection.get('handle')
  handle.commit()
 
+ @classmethod
+ def rollback(cls, profile):
+ connection = cls.get_connection(profile)
+
+ if flags.STRICT_MODE:
+ validate_connection(connection)
+
+ handle = connection.get('handle')
+ handle.rollback()
+
  @classmethod
  def get_status(cls, cursor):
  return cursor.statusmessage

dbt/adapters/snowflake.py

@@ -182,7 +182,7 @@ def rename(cls, profile, from_name, to_name, model_name=None):
 
  @classmethod
  def execute_model(cls, profile, model):
- parts = re.split(r'-- (DBT_OPERATION .*)', model.compiled_contents)
+ parts = re.split(r'-- (DBT_OPERATION .*)', model.get('wrapped_sql'))
  connection = cls.get_connection(profile)
 
  if flags.STRICT_MODE:
@@ -216,7 +216,7 @@ def call_expand_target_column_types(kwargs):
  func_map[function](kwargs)
  else:
  handle, cursor = cls.add_query_to_transaction(
- part, connection, model.name)
+ part, connection, model.get('name'))
 
  handle.commit()
 

dbt/compat.py

@@ -0,0 +1,45 @@
+import codecs
+
+WHICH_PYTHON = None
+
+try:
+ basestring
+ WHICH_PYTHON = 2
+except NameError:
+ WHICH_PYTHON = 3
+
+if WHICH_PYTHON == 2:
+ basestring = basestring
+else:
+ basestring = str
+
+
+def to_unicode(s):
+ if WHICH_PYTHON == 2:
+ return unicode(s)
+ else:
+ return str(s)
+
+
+def to_string(s):
+ if WHICH_PYTHON == 2:
+ if isinstance(s, unicode):
+ return s
+ elif isinstance(s, basestring):
+ return to_unicode(s)
+ else:
+ return to_unicode(str(s))
+ else:
+ if isinstance(s, basestring):
+ return s
+ else:
+ return str(s)
+
+
+def write_file(path, s):
+ if WHICH_PYTHON == 2:
+ with codecs.open(path, 'w', encoding='utf-8') as f:
+ return f.write(to_string(s))
+ else:
+ with open(path, 'w') as f:
+ return f.write(to_string(s))