equal
deleted
inserted
replaced
696 |
696 |
697 |
697 |
698 |
698 |
699 HANDLERS = {} |
699 HANDLERS = {} |
700 |
700 |
701 def register_handler(handlerkls): |
701 def register_handler(handlerkls, overwrite=False): |
702 assert handlerkls is not None |
702 assert handlerkls is not None |
703 HANDLERS[handlerkls.DRIVER] = handlerkls |
703 if overwrite or handlerkls.DRIVER not in HANDLERS: |
|
704 HANDLERS[handlerkls.DRIVER] = handlerkls |
|
705 else: |
|
706 msg = "%s: Handler already exists use overwrite if it's intended\n" |
|
707 "(existing handler class is %r)" |
|
708 raise ValueError(msg % (handlerkls.DRIVER, HANDLERS[handlerkls.DRIVER])) |
704 |
709 |
705 register_handler(PostgresTestDataBaseHandler) |
710 register_handler(PostgresTestDataBaseHandler) |
706 register_handler(SQLiteTestDataBaseHandler) |
711 register_handler(SQLiteTestDataBaseHandler) |
707 register_handler(SQLServerTestDataBaseHandler) |
712 register_handler(SQLServerTestDataBaseHandler) |
708 |
713 |