mirror of
https://github.com/ankitects/anki.git
synced 2025-09-18 14:02:21 -04:00
82a54c780f
- tweak sync code so that a failure in loading the collection won't leave the app with an unopen collection - don't show corrupt collection message when the error is not a db error - catch the temp folder issue when loading the collection. i suspect this was the issue that was causing some people to end up with an open anki instance with no collection loaded
106 lines
2.7 KiB
Python
106 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright: Damien Elmes <anki@ichi2.net>
|
|
# License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
import os
|
|
import time
|
|
|
|
try:
|
|
from pysqlite2 import dbapi2 as sqlite
|
|
except ImportError:
|
|
from sqlite3 import dbapi2 as sqlite
|
|
|
|
Error = sqlite.Error
|
|
|
|
class DB(object):
|
|
def __init__(self, path, text=None, timeout=0):
|
|
encpath = path
|
|
if isinstance(encpath, unicode):
|
|
encpath = path.encode("utf-8")
|
|
self._db = sqlite.connect(encpath, timeout=timeout)
|
|
if text:
|
|
self._db.text_factory = text
|
|
self._path = path
|
|
self.echo = os.environ.get("DBECHO")
|
|
self.mod = False
|
|
|
|
def execute(self, sql, *a, **ka):
|
|
s = sql.strip().lower()
|
|
# mark modified?
|
|
for stmt in "insert", "update", "delete":
|
|
if s.startswith(stmt):
|
|
self.mod = True
|
|
t = time.time()
|
|
if ka:
|
|
# execute("...where id = :id", id=5)
|
|
res = self._db.execute(sql, ka)
|
|
else:
|
|
# execute("...where id = ?", 5)
|
|
res = self._db.execute(sql, a)
|
|
if self.echo:
|
|
#print a, ka
|
|
print sql, "%0.3fms" % ((time.time() - t)*1000)
|
|
if self.echo == "2":
|
|
print a, ka
|
|
return res
|
|
|
|
def executemany(self, sql, l):
|
|
self.mod = True
|
|
t = time.time()
|
|
self._db.executemany(sql, l)
|
|
if self.echo:
|
|
print sql, "%0.3fms" % ((time.time() - t)*1000)
|
|
if self.echo == "2":
|
|
print l
|
|
|
|
def commit(self):
|
|
t = time.time()
|
|
self._db.commit()
|
|
if self.echo:
|
|
print "commit %0.3fms" % ((time.time() - t)*1000)
|
|
|
|
def executescript(self, sql):
|
|
self.mod = True
|
|
if self.echo:
|
|
print sql
|
|
self._db.executescript(sql)
|
|
|
|
def rollback(self):
|
|
self._db.rollback()
|
|
|
|
def scalar(self, *a, **kw):
|
|
res = self.execute(*a, **kw).fetchone()
|
|
if res:
|
|
return res[0]
|
|
return None
|
|
|
|
def all(self, *a, **kw):
|
|
return self.execute(*a, **kw).fetchall()
|
|
|
|
def first(self, *a, **kw):
|
|
c = self.execute(*a, **kw)
|
|
res = c.fetchone()
|
|
c.close()
|
|
return res
|
|
|
|
def list(self, *a, **kw):
|
|
return [x[0] for x in self.execute(*a, **kw)]
|
|
|
|
def close(self):
|
|
self._db.close()
|
|
|
|
def set_progress_handler(self, *args):
|
|
self._db.set_progress_handler(*args)
|
|
|
|
def __enter__(self):
|
|
self._db.execute("begin")
|
|
return self
|
|
|
|
def __exit__(self, exc_type, *args):
|
|
self._db.close()
|
|
|
|
def totalChanges(self):
|
|
return self._db.total_changes
|
|
|
|
def interrupt(self):
|
|
self._db.interrupt()
|