CLI to update a Bluesky Resource document spec field

update_resource_spec.py

import argparse
import pymongo
import sys


def update_resource_spec(mongo_uri, resource_uid, new_spec, wet_run):
    client = MongoClient(mongo_uri)
    collection = client.get_database()["resource"]
    print(f"Updating resource spec {'(dry_run)' if not wet_run else ''}")

    if wet_run:
        collection.update_one({"uid": resource_uid}, {"$set": {"spec": new_spec}})

    print(
        "update operation: update_one({'uid': "
        + resource_uid
        + "}, {'$set': {'spec': "
        + new_spec
        + "}})"
    )

def main(args=sys.argv):
    parser = argparse.ArgumentParser(
        description="Updates a bluesky resource document spec"
    )
    parser.add_argument("mongo_uri")
    parser.add_argument("resource_uid")
    parser.add_argument("new_spec")
    parser.add_argument(
        "-wet",
        "--wet_run",
        action="store_true",
        help="this flag needs to be set to modify the resource document",
    )
    args = parser.parse_args()
    update_resource_spec(args.mongo_uri, args.resource_uid, args.new_spec, args.wet_run)
    
    
if __name__ == "__main__":
    main()