diff options
Diffstat (limited to 'lib/sqlalchemy/orm/uowdumper.py')
| -rw-r--r-- | lib/sqlalchemy/orm/uowdumper.py | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/lib/sqlalchemy/orm/uowdumper.py b/lib/sqlalchemy/orm/uowdumper.py index 5d3bfb85f..a46f90563 100644 --- a/lib/sqlalchemy/orm/uowdumper.py +++ b/lib/sqlalchemy/orm/uowdumper.py @@ -8,6 +8,7 @@ from sqlalchemy.orm import unitofwork from sqlalchemy.orm import util as mapperutil +import StringIO class UOWDumper(unitofwork.UOWExecutor): def __init__(self, tasks, buf): @@ -16,6 +17,12 @@ class UOWDumper(unitofwork.UOWExecutor): self.buf = buf self.execute(None, tasks) + @classmethod + def dump(cls, tasks): + buf = StringIO.StringIO() + UOWDumper(tasks, buf) + return buf.getvalue() + def execute(self, trans, tasks, isdelete=None): if isdelete is not True: for task in tasks: @@ -100,11 +107,7 @@ class UOWDumper(unitofwork.UOWExecutor): name = repr(task.mapper) else: name = '(none)' - sd = getattr(task, '_superduper', False) - if sd: - return ("SD UOWTask(%s, %s)" % (hex(id(task)), name)) - else: - return ("UOWTask(%s, %s)" % (hex(id(task)), name)) + return ("UOWTask(%s, %s)" % (hex(id(task)), name)) def _repr_task_class(self, task): if task.mapper is not None and task.mapper.__class__.__name__ == 'Mapper': |