From 70342c02960f9a5196728d0909762163541e1ac3 Mon Sep 17 00:00:00 2001 From: emmet Date: Fri, 26 Jan 2018 00:59:47 -0800 Subject: fix SnowflakeBackend --- yoyo/backends.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/yoyo/backends.py b/yoyo/backends.py index e8614be..4a8f14b 100644 --- a/yoyo/backends.py +++ b/yoyo/backends.py @@ -459,17 +459,21 @@ class PostgresqlBackend(DatabaseBackend): yield self.connection.autocommit = saved + class SnowflakeBackend(DatabaseBackend): driver_module = 'snowflake.connector' def connect(self, dburi): - connargs = [] + database, schema = dburi.database.split('/') + return self.driver.connect(user=dburi.username, password=dburi.password, account=dburi.hostname, + database=database, schema=schema, warehouse=dburi.args['warehouse']) - if dburi.username is not None: - connargs.append('user=%s' % dburi.username) - if dburi.password is not None: - connargs.append('password=%s' % dburi.password) - connargs.append('account=%s' % dburi.database) - return self.driver.connect(' '.join(connargs)) + def savepoint(self, id): + pass + def savepoint_release(self, id): + pass + + def savepoint_rollback(self, id): + pass \ No newline at end of file -- cgit v1.2.1