sqlalchemy.event.listen(sqlalchemy.engine.Engine, "after_cursor_execute", self.listener) return self def __exit__(self, *_): sqlalchemy.event.remove(sqlalchemy.engine.Engine, "after_cursor_execute", self.listener) @property def get(self): return len(self.queries) benchmark = benchmarks.get_benchmark() def convert_date_format(date, format_from, format_to): """Convert string date format from one to another.""" return datetime.datetime.strptime(date, format_from).strftime(format_to) def iso_to_us_date(date_string): """Convert date string from ISO format to US format.""" return convert_date_format(date_string, DATE_FORMAT_ISO, DATE_FORMAT_US) def generate_query_chunks(query, chunk_size=CHUNK_SIZE, needs_ordering=True): """Make a generator splitting `query` into chunks of size `chunk_size`.""" count = query.count()
sqlalchemy.event.listen(sqlalchemy.engine.Engine, "after_cursor_execute", self.listener) return self def __exit__(self, *_): sqlalchemy.event.remove(sqlalchemy.engine.Engine, "after_cursor_execute", self.listener) @property def get(self): return len(self.queries) benchmark = benchmarks.get_benchmark() with_nop = benchmarks.WithNop def convert_date_format(date, format_from, format_to): """Convert string date format from one to another.""" return datetime.datetime.strptime(date, format_from).strftime(format_to) def iso_to_us_date(date_string): """Convert date string from ISO format to US format.""" return convert_date_format(date_string, DATE_FORMAT_ISO, DATE_FORMAT_US) def generate_query_chunks(query, chunk_size=1000): """Make a generator splitting `query` into chunks of size `chunk_size`."""