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`."""
