diff --git a/initdb.py b/initdb.py index 55eb76c71..d00c0eeb9 100644 --- a/initdb.py +++ b/initdb.py @@ -4,7 +4,8 @@ import hashlib import random from datetime import datetime, timedelta -from peewee import SqliteDatabase, create_model_tables, drop_model_tables +from peewee import (SqliteDatabase, create_model_tables, drop_model_tables, + savepoint_sqlite) from data.database import * from data import model @@ -111,7 +112,6 @@ def __generate_repository(user, name, description, is_public, permissions, return repo -from util.dbutil import savepoint_sqlite db_initialized_for_testing = False testcases = {} diff --git a/requirements.txt b/requirements.txt index ce9a22c54..98697d7b6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -23,7 +23,7 @@ lockfile==0.9.1 marisa-trie==0.5.1 mixpanel-py==3.0.0 paramiko==1.12.0 -peewee==2.1.7 +peewee==2.2.0 py-bcrypt==0.4 pyPdf==1.13 pycrypto==2.6.1 diff --git a/util/dbutil.py b/util/dbutil.py deleted file mode 100644 index 4e565abca..000000000 --- a/util/dbutil.py +++ /dev/null @@ -1,60 +0,0 @@ -import uuid - -# Note: These savepoint classes are implemented in peewee, but not the version we have. Copied here from https://github.com/coleifer/peewee/blob/b657d08a14e4cdafee417111ccba62ede9344222/peewee.py - -class savepoint(object): - def __init__(self, db, sid=None): - self.db = db - _compiler = db.compiler() - self.sid = sid or 's' + uuid.uuid4().get_hex() - self.quoted_sid = _compiler.quote(self.sid) - - def _execute(self, query): - self.db.execute_sql(query, require_commit=False) - - def commit(self): - self._execute('RELEASE SAVEPOINT %s;' % self.quoted_sid) - - def rollback(self): - self._execute('ROLLBACK TO SAVEPOINT %s;' % self.quoted_sid) - - def __enter__(self): - self._orig_autocommit = self.db.get_autocommit() - self.db.set_autocommit(False) - self._execute('SAVEPOINT %s;' % self.quoted_sid) - return self - - def __exit__(self, exc_type, exc_val, exc_tb): - try: - if exc_type: - self.rollback() - else: - try: - self.commit() - except: - self.rollback() - raise - finally: - self.db.set_autocommit(self._orig_autocommit) - - -class savepoint_sqlite(savepoint): - def __enter__(self): - conn = self.db.get_conn() - # For sqlite, the connection's isolation_level *must* be set to None. - # The act of setting it, though, will break any existing savepoints, - # so only write to it if necessary. - if conn.isolation_level is not None: - self._orig_isolation_level = conn.isolation_level - conn.isolation_level = None - else: - self._orig_isolation_level = None - super(savepoint_sqlite, self).__enter__() - - def __exit__(self, exc_type, exc_val, exc_tb): - try: - return super(savepoint_sqlite, self).__exit__( - exc_type, exc_val, exc_tb) - finally: - if self._orig_isolation_level is not None: - self.db.get_conn().isolation_level = self._orig_isolation_level