20 lines
619 B
Python
20 lines
619 B
Python
|
import sqlite3
|
||
|
|
from twisted.internet import task, reactor
|
||
|
|
import logging
|
||
|
|
|
||
|
|
|
||
|
|
def rerun_if_locked(f):
|
||
|
|
|
||
|
|
def rerun(err, *args, **kwargs):
|
||
|
|
if err.check(sqlite3.OperationalError) and err.value.message == "database is locked":
|
||
|
|
logging.warning("database was locked. rerunning %s with args %s, kwargs %s",
|
||
|
|
str(f), str(args), str(kwargs))
|
||
|
|
return task.deferLater(reactor, 0, wrapper, *args, **kwargs)
|
||
|
|
return err
|
||
|
|
|
||
|
|
def wrapper(*args, **kwargs):
|
||
|
|
d = f(*args, **kwargs)
|
||
|
|
d.addErrback(rerun, *args, **kwargs)
|
||
|
|
return d
|
||
|
|
|
||
|
|
return wrapper
|