def upload_to_rockset(collection: str, docs: List[Any]) -> None:
print(f"Writing {len(docs)} documents to Rockset")
client = rockset.Client(
api_server="api.rs2.usw2.rockset.com", api_key=os.environ["ROCKSET_API_KEY"]
)
client.Collection.retrieve(collection).add_docs(docs)
print("Done!")
def main() -> None:
rs = rockset.Client(api_server="api.rs2.usw2.rockset.com",
api_key=os.environ["ROCKSET_API_KEY"])
qlambda = rs.QueryLambda.retrieve('commit_jobs_batch_query',
version='15aba20837ae9d75',
workspace='commons')
commits = get_latest_commits()
results = query_commits(commits, qlambda)
latest_viable_commit = get_latest_green_commit(commits, results)
print(latest_viable_commit)
MIMEText(
body,
'html'
)
)
context = ssl.create_default_context()
with smtplib.SMTP_SSL('smtp.gmail.com', 465, context=context) as server:
server.login(sender_email, password)
server.sendmail(
sender_email, receiver_email, message.as_string()
)
app = flask.Flask(__name__)
rs = rockset.Client(
api_key=os.getenv('RS2_TOKEN'),
api_server='api.rs2.usw2.rockset.com'
)
collection = rs.Collection.retrieve(
'emails',
workspace='harker_hackers'
)
@app.route('/')
def login():
return(
flask.render_template(
'index.jinja'
)
)
@app.route('/authorized')
print("rm: ", TEMP_DIR)
shutil.rmtree(TEMP_DIR)
print("mkdir: ", TEMP_DIR)
TEMP_DIR.mkdir()
print("cd to ", TEMP_DIR)
os.chdir(TEMP_DIR)
# Download and extract all the reports (both GHA and S3)
download_and_extract_s3_reports(args.workflow_run_id, args.workflow_run_attempt)
artifact_urls = get_artifact_urls(args.workflow_run_id)
for name, url in artifact_urls.items():
download_and_extract_artifact(Path(name), url, args.workflow_run_attempt)
# Parse the reports and transform them to JSON
test_cases = []
for xml_report in Path(".").glob("**/*.xml"):
test_cases.extend(
parse_xml_report(
xml_report, int(args.workflow_run_id), int(args.workflow_run_attempt)
)
)
# Write the JSON to rockset
print(f"Writing {len(test_cases)} test cases to Rockset")
client = rockset.Client(
api_server="api.rs2.usw2.rockset.com", api_key=os.environ["ROCKSET_API_KEY"]
)
client.Collection.retrieve("test_run").add_docs(test_cases)
print("Done!")
