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!")