in configurations/redshift-performance-test.py [0:0]
def get_sql(engine, number_of_parallel_sessions):
sql_script = ""
pattern = re.compile(r'limit[\s|\t|\n]+[\d]+[\s]*$', re.IGNORECASE)
for query in get_sql_scripts_from_s3():
if not re.search(pattern, query):
query += " limit " + DEFAULT_OUTPUT_LIMIT
sql_script+=query + ";\n"
if DISABLE_RESULT_CACHE=='true':
sql_script = "set enable_result_cache_for_session to false;\n" + sql_script
sql_script = "set query_group to '" + QUERY_LABEL_PREFIX + str(number_of_parallel_sessions) + "';\n" + sql_script
df = pandas.read_sql(text(sql_script), engine)
return df