def flush_context(): ThreadLocalODMSession.close_all() try: yield ThreadLocalODMSession.flush_all() finally: ThreadLocalODMSession.close_all()
def main(args): test = TestController() setup(test) url = generate_wiki_thread(test) ThreadLocalODMSession.close_all() # make sure ODM sessions won't get re-used counts = count_page(test, url, verbose=args.verbose, debug_html=args.debug_html) print json.dumps(counts) write_csv(counts, args.id, args.data_file) test.tearDown()
def main(args): test = TestController() setup(test) url = generate_wiki_thread(test) # make sure ODM sessions won't get re-used ThreadLocalODMSession.close_all() counts = count_page(test, url, verbose=args.verbose, debug_html=args.debug_html) print(json.dumps(counts)) write_csv(counts, args.id, args.data_file) test.tearDown()
def __call__(self, environ, start_response): try: result = self.app(environ, start_response) if isinstance(result, list): self._cleanup_request() return result else: return self._cleanup_iterator(result) except self.flush_on_errors as exc: self._cleanup_request() raise except: ThreadLocalODMSession.close_all() ContextualODMSession.close_all(self._context_func()) raise
def __call__(self, environ, start_response): try: result = self.app(environ, start_response) if isinstance(result, list): self._cleanup_request() return result else: return self._cleanup_iterator(result) except self.flush_on_errors as exc: self._cleanup_request() raise except: ThreadLocalODMSession.close_all() ContextualODMSession.close_all(self._context_func()) raise
def _cleanup_request(self): context = self._context_func() ThreadLocalODMSession.flush_all() ContextualODMSession.flush_all(context) ThreadLocalODMSession.close_all() ContextualODMSession.close_all(context)
def _cleanup_request(self): context = self._context_func() ThreadLocalODMSession.flush_all() ContextualODMSession.flush_all(context) ThreadLocalODMSession.close_all() ContextualODMSession.close_all(context)
self.flush_on_errors = flush_on_errors self._context_func = context_func or (lambda: None) def __call__(self, environ, start_response): try: result = self.app(environ, start_response) if isinstance(result, list): self._cleanup_request() return result else: return self._cleanup_iterator(result) except self.flush_on_errors, exc: self._cleanup_request() raise except: ThreadLocalODMSession.close_all() ContextualODMSession.close_all(self._context_func()) raise def _cleanup_request(self): context = self._context_func() ThreadLocalODMSession.flush_all() ContextualODMSession.flush_all(context) ThreadLocalODMSession.close_all() ContextualODMSession.close_all(context) def _cleanup_iterator(self, result): for x in result: yield x self._cleanup_request()
self.flush_on_errors = flush_on_errors self._context_func = context_func or (lambda:None) def __call__(self, environ, start_response): try: result = self.app(environ, start_response) if isinstance(result, list): self._cleanup_request() return result else: return self._cleanup_iterator(result) except self.flush_on_errors, exc: self._cleanup_request() raise except: ThreadLocalODMSession.close_all() ContextualODMSession.close_all(self._context_func()) raise def _cleanup_request(self): context = self._context_func() ThreadLocalODMSession.flush_all() ContextualODMSession.flush_all(context) ThreadLocalODMSession.close_all() ContextualODMSession.close_all(context) def _cleanup_iterator(self, result): for x in result: yield x self._cleanup_request()
def _cleanup_request(): ThreadLocalODMSession.flush_all() ThreadLocalODMSession.close_all()